Function TestNewFieldRemovalCodemod_TableDriven

pkg/cli/codemod_factory_test.go:208–271

(t *testing.T)

}

func TestNewFieldRemovalCodemod_TableDriven(t *testing.T) {
	tests := []struct {
		name        string
		content     string
		frontmatter map[string]any
		wantApplied bool
		wantContent string // expected substring in result when applied, or full match when not applied
	}{
		{
			name:    "parent key absent",
			content: "---\non: workflow_dispatch\n---\n\n# Test",
			frontmatter: map[string]any{
				"on": "workflow_dispatch",
			},
			wantApplied: false,
		},
		{
			name:    "parent not a map",
			content: "---\non: workflow_dispatch\nparent: scalar\n---\n\n# Test",
			frontmatter: map[string]any{
				"on":     "workflow_dispatch",
				"parent": "scalar",
			},
			wantApplied: false,
		},
		{
			name:    "child field absent",
			content: "---\non: workflow_dispatch\nparent:\n  other: val\n---\n\n# Test",
			frontmatter: map[string]any{
				"on":     "workflow_dispatch",
				"parent": map[string]any{"other": "val"},
			},
			wantApplied: false,
		},
		{
			name:    "child field present",
			content: "---\non: workflow_dispatch\nparent:\n  child: yes\n  other: val\n---\n\n# Test",
			frontmatter: map[string]any{
				"on":     "workflow_dispatch",
				"parent": map[string]any{"child": true, "other": "val"},
			},
			wantApplied: true,
			wantContent: "other: val",
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			codemod := newFieldRemovalCodemod(baseFieldRemovalConfig())

			result, applied, err := codemod.Apply(tt.content, tt.frontmatter)

			require.NoError(t, err, "Apply should not return an error")
			assert.Equal(t, tt.wantApplied, applied, "Applied flag should match expectation")

			if tt.wantApplied {
				assert.Contains(t, result, tt.wantContent, "Result should contain expected content")
				assert.NotContains(t, result, "child:", "Result should not contain removed field")