Integrate cleanup into pipeline execution
This commit is contained in:
@@ -10,7 +10,7 @@ import (
|
||||
|
||||
func TestBuildFullPlanOrder(t *testing.T) {
|
||||
got := BuildFullPlan()
|
||||
want := []string{"prepare", "transcribe", "merge", "polish", "normalize", "trim", "render", "extract", "analyze", "publish", "notify"}
|
||||
want := []string{"prepare", "transcribe", "merge", "polish", "normalize", "trim", "render", "extract", "analyze", "publish", "notify", "cleanup"}
|
||||
if len(got) != len(want) {
|
||||
t.Fatalf("len(plan) = %d, want %d", len(got), len(want))
|
||||
}
|
||||
@@ -22,15 +22,15 @@ func TestBuildFullPlanOrder(t *testing.T) {
|
||||
}
|
||||
|
||||
func TestBuildSingleStagePlan(t *testing.T) {
|
||||
stages, err := BuildSingleStagePlan("polish")
|
||||
stages, err := BuildSingleStagePlan("cleanup")
|
||||
if err != nil {
|
||||
t.Fatalf("BuildSingleStagePlan() error = %v", err)
|
||||
}
|
||||
if len(stages) != 1 {
|
||||
t.Fatalf("len(plan) = %d, want 1", len(stages))
|
||||
}
|
||||
if stages[0].Name() != "polish" {
|
||||
t.Fatalf("stage name = %q, want %q", stages[0].Name(), "polish")
|
||||
if stages[0].Name() != "cleanup" {
|
||||
t.Fatalf("stage name = %q, want %q", stages[0].Name(), "cleanup")
|
||||
}
|
||||
}
|
||||
|
||||
@@ -92,6 +92,19 @@ func TestBuildBoundedPlanDefaultsToFullCanonicalPlan(t *testing.T) {
|
||||
}
|
||||
}
|
||||
|
||||
func TestBuildBoundedPlanThroughNotifyExcludesCleanup(t *testing.T) {
|
||||
plan, err := BuildBoundedPlan("", "notify")
|
||||
if err != nil {
|
||||
t.Fatalf("BuildBoundedPlan() error = %v", err)
|
||||
}
|
||||
if plan.Contains("cleanup") {
|
||||
t.Fatalf("plan = %#v, cleanup must be outside a notify-bounded run", plan.Names())
|
||||
}
|
||||
if got := plan.Through(); got != "notify" {
|
||||
t.Fatalf("through = %q, want notify", got)
|
||||
}
|
||||
}
|
||||
|
||||
func TestBuildBoundedPlanRejectsInvalidBounds(t *testing.T) {
|
||||
tests := []struct {
|
||||
name string
|
||||
@@ -99,9 +112,9 @@ func TestBuildBoundedPlanRejectsInvalidBounds(t *testing.T) {
|
||||
through string
|
||||
want []string
|
||||
}{
|
||||
{name: "unknown from", from: "missing", through: "analyze", want: []string{"unknown from stage", "missing", "prepare", "notify"}},
|
||||
{name: "unknown through", from: "extract", through: "missing", want: []string{"unknown through stage", "missing", "prepare", "notify"}},
|
||||
{name: "reversed", from: "publish", through: "render", want: []string{"publish", "occurs after", "render", "prepare", "notify"}},
|
||||
{name: "unknown from", from: "missing", through: "analyze", want: []string{"unknown from stage", "missing", "prepare", "cleanup"}},
|
||||
{name: "unknown through", from: "extract", through: "missing", want: []string{"unknown through stage", "missing", "prepare", "cleanup"}},
|
||||
{name: "reversed", from: "publish", through: "render", want: []string{"publish", "occurs after", "render", "prepare", "cleanup"}},
|
||||
}
|
||||
for _, test := range tests {
|
||||
t.Run(test.name, func(t *testing.T) {
|
||||
|
||||
Reference in New Issue
Block a user