Function TestMCPNetworkCodemod_PreservesComments

pkg/cli/codemod_mcp_network_test.go:410–446

(t *testing.T)

}

func TestMCPNetworkCodemod_PreservesComments(t *testing.T) {
	codemod := getMCPNetworkMigrationCodemod()

	content := `---
on: workflow_dispatch
# MCP server configuration
mcp-servers:
  my-server:
    container: my-image
    # Network configuration (deprecated)
    network:
      allowed:
        - example.com
---

# Test`

	frontmatter := map[string]any{
		"on": "workflow_dispatch",
		"mcp-servers": map[string]any{
			"my-server": map[string]any{
				"container": "my-image",
				"network": map[string]any{
					"allowed": []any{"example.com"},
				},
			},
		},
	}

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

	require.NoError(t, err)
	assert.True(t, applied)
	assert.Contains(t, result, "# MCP server configuration")
	// Comments at the same level as the removed field are preserved (they're siblings, not children)
	assert.Contains(t, result, "# Network configuration (deprecated)")
}

func TestMCPNetworkCodemod_EmptyNetworkAllowed(t *testing.T) {
	codemod := getMCPNetworkMigrationCodemod()