Function TestAwInfoHasMCPServers

pkg/cli/audit_expanded_test.go:673–725

(t *testing.T)

}

func TestAwInfoHasMCPServers(t *testing.T) {
	tests := []struct {
		name           string
		awInfoContent  string
		awInfoSubdir   string
		expectedNames  []string
		expectedHasMCP bool
	}{
		{
			name:           "with MCP servers",
			awInfoContent:  `{"engine_id":"copilot","steps":{"mcp_servers":{"github":{},"filesystem":{}}}}`,
			expectedNames:  []string{"filesystem", "github"},
			expectedHasMCP: true,
		},
		{
			name:           "without MCP servers",
			awInfoContent:  `{"engine_id":"copilot","steps":{"firewall":"squid"}}`,
			expectedHasMCP: false,
		},
		{
			name:           "without steps",
			awInfoContent:  `{"engine_id":"copilot"}`,
			expectedHasMCP: false,
		},
		{
			name:           "with MCP servers in activation subdir",
			awInfoContent:  `{"engine_id":"copilot","steps":{"mcp_servers":{"github":{}}}}`,
			awInfoSubdir:   "activation",
			expectedNames:  []string{"github"},
			expectedHasMCP: true,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			tmpDir := testutil.TempDir(t, "mcp-servers-*")
			targetDir := tmpDir
			if tt.awInfoSubdir != "" {
				targetDir = filepath.Join(tmpDir, tt.awInfoSubdir)
				err := os.MkdirAll(targetDir, 0755)
				require.NoError(t, err, "Should create subdir")
			}
			err := os.WriteFile(filepath.Join(targetDir, "aw_info.json"), []byte(tt.awInfoContent), 0644)
			require.NoError(t, err, "Should write aw_info.json")

			names, hasMCP := extractMCPServerNamesFromAwInfo(tmpDir)
			assert.Equal(t, tt.expectedHasMCP, hasMCP, "Has MCP servers should match")
			if tt.expectedHasMCP {
				assert.Equal(t, tt.expectedNames, names, "MCP server names should match")
			}
		})
	}
}