Method VerifyCodeFix

internal/fourslash/fourslash.go:1587–1662

VerifyCodeFix verifies that applying a code fix produces the expected file content.

(t *testing.T, options VerifyCodeFixOptions)

// VerifyCodeFix verifies that applying a code fix produces the expected file content.
func (f *FourslashTest) VerifyCodeFix(t *testing.T, options VerifyCodeFixOptions) {
	t.Helper()

	if options.UserPreferences != nil {
		reset := f.ConfigureWithReset(t, *options.UserPreferences)
		defer reset()
	}

	actions := f.getCodeFixActions(t)

	if len(actions) == 0 {
		t.Fatalf("No code fixes returned.")
	}
	if options.Index >= len(actions) {
		t.Fatalf("Code fix index %d out of range (got %d fixes)", options.Index, len(actions))
	}

	matchingAction := actions[options.Index]
	if matchingAction.Title != options.Description {
		found := false
		for _, action := range actions {
			if action.Title == options.Description {
				matchingAction = action
				found = true
				break
			}
		}
		if !found {
			var titles []string
			for _, a := range actions {
				titles = append(titles, a.Title)
			}
			t.Fatalf("No code fix with description %q at index %d found. Available fixes: %v", options.Description, options.Index, titles)
		}
	}

	originalContent := f.getScriptInfo(f.activeFilename).content
	expectedContent := options.NewFileContent
	if options.NewRangeContent != "" {
		selection := f.getSelection()
		if selection.Pos() == selection.End() {
			ranges := f.getRangesInFile(f.activeFilename)
			if len(ranges) == 0 {
				t.Fatal("Expected a selected range or fourslash range for NewRangeContent verification.")
			}
			selection = ranges[0].Range
		}
		expectedContent = originalContent[:selection.Pos()] + options.NewRangeContent + originalContent[selection.End():]
	}

	if options.ApplyChanges {
		if matchingAction.Edit != nil && matchingAction.Edit.Changes != nil {
			expectedURI := lsconv.FileNameToDocumentURI(f.activeFilename)
			for uri, edits := range *matchingAction.Edit.Changes {
				if uri != expectedURI {
					t.Fatalf("Code fix returned edits for unexpected URI %q (expected %q)", uri, expectedURI)
				}
				f.applyTextEdits(t, edits)