Test

Author: firelizzard18

analyzer_test.go

@@ -185,3 +185,42 @@ See the License for the specific language governing permissions and
 limitations under the License.`))
 	require.Nil(t, issue)
 }
+
+func TestFix(t *testing.T) {
+	const src = `// mycompany.net
+// SPDX-License-Identifier: Foo
+
+// Package foo
+package foo
+
+func Foo() { println("Foo") }
+`
+
+	a := goheader.New(
+		goheader.WithTemplate(`{{ MY COMPANY }}
+SPDX-License-Identifier: Foo`),
+		goheader.WithValues(map[string]goheader.Value{
+			"MY COMPANY": &goheader.ConstValue{
+				RawValue: "mycompany.com",
+			},
+		}))
+
+	fset := token.NewFileSet()
+	file, err := parser.ParseFile(fset, "foo.go", src, parser.ParseComments)
+	require.NoError(t, err)
+
+	issue := a.Analyze(&goheader.Target{
+		File: file,
+		Path: t.TempDir(),
+	})
+	require.NotNil(t, issue)
+	require.NotNil(t, issue.Fix())
+	require.Equal(t, []string{
+		"// mycompany.net",
+		"// SPDX-License-Identifier: Foo",
+	}, issue.Fix().Actual)
+	require.Equal(t, []string{
+		"// mycompany.com",
+		"// SPDX-License-Identifier: Foo",
+	}, issue.Fix().Expected)
+}