Function TestConclusionJobWithGeneratedAssets

pkg/workflow/notify_comment_test.go:567–625

TestConclusionJobWithGeneratedAssets tests that the conclusion job includes environment variables for safe output job URLs when safe output jobs are present

(t *testing.T)

// TestConclusionJobWithGeneratedAssets tests that the conclusion job includes environment variables
// for safe output job URLs when safe output jobs are present
func TestConclusionJobWithGeneratedAssets(t *testing.T) {
	compiler := NewCompiler()

	statusCommentTrue := true
	// Create workflow data with safe outputs configuration
	workflowData := &WorkflowData{
		Name:          "Test Workflow",
		StatusComment: &statusCommentTrue, // Explicitly enable status comments
		SafeOutputs: &SafeOutputsConfig{
			AddComments: &AddCommentsConfig{
				BaseSafeOutputConfig: BaseSafeOutputConfig{
					Max: strPtr("1"),
				},
			},
			NoOp: &NoOpConfig{},
		},
		AIReaction: "eyes",
	}

	// Define safe output jobs that should have URL outputs
	safeOutputJobNames := []string{
		"create_issue",
		"add_comment",
		"create_pull_request",
		"create_discussion",
	}

	// Build the conclusion job
	job, err := compiler.buildConclusionJob(workflowData, string(constants.AgentJobName), safeOutputJobNames)
	if err != nil {
		t.Fatalf("Failed to build conclusion job: %v", err)
	}

	if job == nil {
		t.Fatal("Expected conclusion job to be created")
	}

	// Convert job to YAML string for checking
	jobYAML := strings.Join(job.Steps, "")

	// Check that GH_AW_SAFE_OUTPUT_JOBS environment variable is declared with JSON mapping
	if !strings.Contains(jobYAML, "GH_AW_SAFE_OUTPUT_JOBS:") {
		t.Error("Expected GH_AW_SAFE_OUTPUT_JOBS environment variable to be declared")
	}

	// Check that individual URL output environment variables are declared
	expectedEnvVars := []string{
		"GH_AW_OUTPUT_CREATE_ISSUE_ISSUE_URL: ${{ needs.create_issue.outputs.issue_url }}",
		"GH_AW_OUTPUT_ADD_COMMENT_COMMENT_URL: ${{ needs.add_comment.outputs.comment_url }}",
		"GH_AW_OUTPUT_CREATE_PULL_REQUEST_PULL_REQUEST_URL: ${{ needs.create_pull_request.outputs.pull_request_url }}",
		"GH_AW_OUTPUT_CREATE_DISCUSSION_DISCUSSION_URL: ${{ needs.create_discussion.outputs.discussion_url }}",
	}

	for _, expectedVar := range expectedEnvVars {
		if !strings.Contains(jobYAML, expectedVar) {
			t.Errorf("Expected environment variable not found: %s", expectedVar)
		}
	}