fix broken scheduler test with new behavior

streamingfast · Mar 29, 2024 · a284f11 · a284f11
1 parent c63881e
commit a284f11
Showing 1 changed file with 9 additions and 6 deletions.
diff --git a/orchestrator/stage/stages_test.go b/orchestrator/stage/stages_test.go
@@ -59,32 +59,35 @@ S:.
 M:N`)
 
 	j1, _ := stages.NextJob()
-	assert.Equal(t, 1, j1.Stage)
+	assert.Equal(t, 0, j1.Stage)
 	assert.Equal(t, 0, j1.Segment)
 
 	segmentStateEquals(t, stages, `
-S:.
 S:S
+S:.
 M:N`)
 
-	stages.forceTransition(0, 1, UnitCompleted)
+	stages.forceTransition(0, 0, UnitCompleted)
 
 	segmentStateEquals(t, stages, `
-S:.
 S:C
+S:.
 M:N`)
 
 	stages.NextJob()
+	assert.Equal(t, 0, j1.Stage)
+	assert.Equal(t, 0, j1.Segment)
 
 	segmentStateEquals(t, stages, `
-S:S
 S:C
+S:S
 M:N`)
 
 	stages.NextJob()
 
+	stages.forceTransition(0, 1, UnitCompleted)
 	segmentStateEquals(t, stages, `
-S:SS
+S:CS
 S:C.
 M:N.`)