Function TestModelAliasesAWFConfigJSON

pkg/workflow/model_aliases_import_test.go:101–143

TestModelAliasesAWFConfigJSON verifies that model alias entries from imported workflows are merged into WorkflowData.ModelMappings during compilation and emitted under apiProxy.models in the AWF config JSON.

(t *testing.T)

// are merged into WorkflowData.ModelMappings during compilation and emitted under
// apiProxy.models in the AWF config JSON.
func TestModelAliasesAWFConfigJSON(t *testing.T) {
	awfConfig := workflow.AWFCommandConfig{
		EngineName:     "copilot",
		AllowedDomains: "github.com",
		WorkflowData: &workflow.WorkflowData{
			EngineConfig: &workflow.EngineConfig{ID: "copilot"},
			NetworkPermissions: &workflow.NetworkPermissions{
				Firewall: &workflow.FirewallConfig{Enabled: true},
			},
			// Simulate: import defines a new alias, main overrides a builtin.
			ModelMappings: workflow.MergeImportedModelAliases(
				[]map[string][]string{
					{"import-alias": {"import/model"}},
				},
				map[string][]string{
					"haiku": {"main/haiku-override"},
				},
			),
		},
	}

	jsonStr, err := workflow.BuildAWFConfigJSON(awfConfig)
	require.NoError(t, err, "BuildAWFConfigJSON should not return an error")

	// models must appear nested under apiProxy
	assert.Contains(t, jsonStr, `"models"`, "models section must be present under apiProxy in AWF config JSON")

	// Verify that the alias map is correctly populated in WorkflowData.
	mappings := awfConfig.WorkflowData.ModelMappings
	require.NotNil(t, mappings, "ModelMappings should be set on WorkflowData")

	// Imported alias is in the model mappings.
	assert.Equal(t, []string{"import/model"}, mappings["import-alias"],
		"import-alias from imported workflow should be in ModelMappings")

	// Main workflow override wins over builtin haiku.
	assert.Equal(t, []string{"main/haiku-override"}, mappings["haiku"],
		"main workflow alias should override builtin haiku in ModelMappings")

	// Other builtins preserved.
	assert.NotEmpty(t, mappings["sonnet"], "builtin sonnet should still be in ModelMappings")
	assert.NotEmpty(t, mappings["agent"], "builtin agent should still be in ModelMappings")
}