Function TestAwfVersionInAwInfo

pkg/workflow/aw_info_versions_test.go:162–244

(t *testing.T)

}

func TestAwfVersionInAwInfo(t *testing.T) {
	tests := []struct {
		name               string
		firewallEnabled    bool
		firewallVersion    string
		agentVersion       string
		expectedAwfVersion string
		description        string
	}{
		{
			name:               "Firewall enabled with explicit version",
			firewallEnabled:    true,
			firewallVersion:    "v1.0.0",
			expectedAwfVersion: "v1.0.0",
			description:        "Should use explicit firewall version",
		},
		{
			name:               "Firewall enabled with default version",
			firewallEnabled:    true,
			firewallVersion:    "",
			expectedAwfVersion: string(constants.DefaultFirewallVersion),
			description:        "Should use default firewall version when not specified",
		},
		{
			name:               "Firewall disabled",
			firewallEnabled:    false,
			firewallVersion:    "",
			agentVersion:       "",
			expectedAwfVersion: "",
			description:        "Should have empty awf_version when firewall is disabled",
		},
		{
			name:               "sandbox.agent.version overrides firewall version",
			firewallEnabled:    true,
			firewallVersion:    "",
			agentVersion:       "v0.30.1",
			expectedAwfVersion: "v0.30.1",
			description:        "Should prefer sandbox.agent.version override",
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			compiler := NewCompiler(WithVersion("1.0.0"))
			registry := GetGlobalEngineRegistry()
			engine, err := registry.GetEngine("copilot")
			if err != nil {
				t.Fatalf("Failed to get copilot engine: %v", err)
			}

			workflowData := &WorkflowData{
				Name: "Test Workflow",
			}

			if tt.firewallEnabled {
				workflowData.NetworkPermissions = &NetworkPermissions{
					Firewall: &FirewallConfig{
						Enabled: true,