Go: Add unit tests for hasInvalidToolchainVersion

Author: mbg

go/extractor/project/project_test.go

@@ -3,6 +3,8 @@ package project
 import (
 	"path/filepath"
 	"testing"
+
+	"golang.org/x/mod/modfile"
 )
 
 func testStartsWithAnyOf(t *testing.T, path string, prefix string, expectation bool) {
@@ -25,3 +27,38 @@ func TestStartsWithAnyOf(t *testing.T) {
 	testStartsWithAnyOf(t, filepath.Join("foo", "bar"), "bar", false)
 	testStartsWithAnyOf(t, filepath.Join("foo", "bar"), filepath.Join("foo", "baz"), false)
 }
+
+func testHasInvalidToolchainVersion(t *testing.T, contents string) bool {
+	modFile, err := modfile.Parse("test.go", []byte(contents), nil)
+
+	if err != nil {
+		t.Errorf("Unable to parse %s: %s.\n", contents, err.Error())
+	}
+
+	return hasInvalidToolchainVersion(modFile)
+}
+
+func TestHasInvalidToolchainVersion(t *testing.T) {
+	invalid := []string{
+		"go 1.21\n",
+		"go 1.22\n",
+	}
+
+	for _, v := range invalid {
+		if !testHasInvalidToolchainVersion(t, v) {
+			t.Errorf("Expected testHasInvalidToolchainVersion(\"%s\") to be true, but got false", v)
+		}
+	}
+
+	valid := []string{
+		"go 1.20\n",
+		"go 1.21.1\n",
+		"go 1.22\n\ntoolchain go1.22.0\n",
+	}
+
+	for _, v := range valid {
+		if testHasInvalidToolchainVersion(t, v) {
+			t.Errorf("Expected testHasInvalidToolchainVersion(\"%s\") to be false, but got true", v)
+		}
+	}
+}