Function TestRefreshStopTimeBehavior

pkg/workflow/stop_after_test.go:166–273

TestRefreshStopTimeBehavior tests that the refreshStopTime flag controls stop time preservation

(t *testing.T)

// TestRefreshStopTimeBehavior tests that the refreshStopTime flag controls stop time preservation
func TestRefreshStopTimeBehavior(t *testing.T) {
	// Create a temporary directory for test files
	tmpDir, err := os.MkdirTemp("", "refresh-stop-time-test")
	if err != nil {
		t.Fatalf("Failed to create temp dir: %v", err)
	}
	defer os.RemoveAll(tmpDir)

	// Create a markdown workflow file with stop-after
	mdFile := filepath.Join(tmpDir, "test.md")
	lockFile := filepath.Join(tmpDir, "test.lock.yml")

	// Create a lock file with existing stop time
	existingStopTime := "2025-12-31 23:59:59"
	lockContent := fmt.Sprintf(`name: Test Workflow
on:
  workflow_dispatch:
jobs:
  stop_time_check:
    runs-on: ubuntu-latest
    steps:
      - uses: actions/github-script`+"@v8\n"+`        env:
          GH_AW_STOP_TIME: %s
          GH_AW_WORKFLOW_NAME: "Test Workflow"
`, existingStopTime)
	err = os.WriteFile(lockFile, []byte(lockContent), 0644)
	if err != nil {
		t.Fatalf("Failed to create lock file: %v", err)
	}

	// Test 1: Default behavior should preserve existing stop time
	t.Run("default behavior preserves stop time", func(t *testing.T) {
		compiler := NewCompiler()
		compiler.SetRefreshStopTime(false)

		frontmatter := map[string]any{
			"on": map[string]any{
				"workflow_dispatch": nil,
				"stop-after":        "+48h",
			},
		}

		workflowData := &WorkflowData{}
		err := compiler.processStopAfterConfiguration(frontmatter, workflowData, mdFile)
		if err != nil {
			t.Fatalf("processStopAfterConfiguration failed: %v", err)
		}

		if workflowData.StopTime != existingStopTime {
			t.Errorf("Expected stop time to be preserved as %q, got %q", existingStopTime, workflowData.StopTime)
		}
	})

	// Test 2: With refresh flag, should generate new stop time
	t.Run("refresh flag generates new stop time", func(t *testing.T) {
		compiler := NewCompiler()
		compiler.SetRefreshStopTime(true)