Function TestParseAndUpdateGoModFile

rivershared/cmd/update-mod-go/main_test.go:24–118

(t *testing.T)

)`

func TestParseAndUpdateGoModFile(t *testing.T) {
	t.Parallel()

	type testBundle struct{}

	setup := func(t *testing.T) (string, *testBundle) { //nolint:unparam
		t.Helper()

		file, err := os.CreateTemp(t.TempDir(), "go.mod")
		require.NoError(t, err)
		t.Cleanup(func() { _ = os.Remove(file.Name()) })

		_, err = file.WriteString(sampleGoMod)
		require.NoError(t, err)
		require.NoError(t, file.Close())

		return file.Name(), &testBundle{}
	}

	requireDirectives := func(t *testing.T, filename, goVersion, toolchainName string) {
		t.Helper()

		fileData, err := os.ReadFile(filename)
		require.NoError(t, err)

		modFile, err := modfile.Parse(filename, fileData, nil)
		require.NoError(t, err)

		require.Equal(t, goVersion, modFile.Go.Version)
		require.Equal(t, toolchainName, modFile.Toolchain.Name)
	}

	t.Run("WritesChanges", func(t *testing.T) {
		t.Parallel()

		filename, _ := setup(t)

		anyMismatch, err := parseAndUpdateGoModFile(false, filename, "go.work", "1.23", "go1.23.2")
		require.NoError(t, err)
		require.True(t, anyMismatch)

		// Reread the file that the command above just wrote and make sure the right
		// changes were made.
		requireDirectives(t, filename, "1.23", "go1.23.2")

		// Running again is allowed and should be idempotent. This time it'll
		// return that no changes were made.
		anyMismatch, err = parseAndUpdateGoModFile(false, filename, "go.work", "1.23", "go1.23.2")
		require.NoError(t, err)
		require.False(t, anyMismatch)
	})

	t.Run("NoChanges", func(t *testing.T) {
		t.Parallel()

		filename, _ := setup(t)

		anyMismatch, err := parseAndUpdateGoModFile(false, filename, "go.work", "1.22", "go1.23.1")