3030import java .util .Objects ;
3131import java .util .Set ;
3232import java .util .function .Function ;
33- import java .util .function .LongSupplier ;
3433import java .util .stream .Collectors ;
3534
3635import static org .hamcrest .Matchers .equalTo ;
@@ -170,11 +169,10 @@ protected Reader<LifecyclePolicy> instanceReader() {
170169
171170 public void testFirstAndLastSteps () {
172171 Client client = mock (Client .class );
173- LongSupplier nowSupplier = () -> 0L ;
174172 lifecycleName = randomAlphaOfLengthBetween (1 , 20 );
175173 Map <String , Phase > phases = new LinkedHashMap <>();
176174 LifecyclePolicy policy = new LifecyclePolicy (TestLifecycleType .INSTANCE , lifecycleName , phases );
177- List <Step > steps = policy .toSteps (client , nowSupplier );
175+ List <Step > steps = policy .toSteps (client );
178176 assertThat (steps .size (), equalTo (2 ));
179177 assertThat (steps .get (0 ), instanceOf (InitializePolicyContextStep .class ));
180178 assertThat (steps .get (0 ).getKey (), equalTo (new StepKey ("new" , "init" , "init" )));
@@ -184,7 +182,6 @@ public void testFirstAndLastSteps() {
184182
185183 public void testToStepsWithOneStep () {
186184 Client client = mock (Client .class );
187- LongSupplier nowSupplier = () -> 0L ;
188185 MockStep mockStep = new MockStep (
189186 new Step .StepKey ("test" , "test" , "test" ), TerminalPolicyStep .KEY );
190187
@@ -196,8 +193,8 @@ public void testToStepsWithOneStep() {
196193 phases .put (firstPhase .getName (), firstPhase );
197194 LifecyclePolicy policy = new LifecyclePolicy (TestLifecycleType .INSTANCE , lifecycleName , phases );
198195 StepKey firstStepKey = InitializePolicyContextStep .KEY ;
199- StepKey secondStepKey = new StepKey ("new" , PhaseAfterStep .NAME , PhaseAfterStep .NAME );
200- List <Step > steps = policy .toSteps (client , nowSupplier );
196+ StepKey secondStepKey = new StepKey ("new" , PhaseCompleteStep .NAME , PhaseCompleteStep .NAME );
197+ List <Step > steps = policy .toSteps (client );
201198 assertThat (steps .size (), equalTo (4 ));
202199 assertSame (steps .get (0 ).getKey (), firstStepKey );
203200 assertThat (steps .get (0 ).getNextStepKey (), equalTo (secondStepKey ));
@@ -210,13 +207,12 @@ public void testToStepsWithOneStep() {
210207
211208 public void testToStepsWithTwoPhases () {
212209 Client client = mock (Client .class );
213- LongSupplier nowSupplier = () -> 0L ;
214210 MockStep secondActionStep = new MockStep (new StepKey ("second_phase" , "test2" , "test" ), TerminalPolicyStep .KEY );
215- MockStep secondAfter = new MockStep (new StepKey ("first_phase" , PhaseAfterStep .NAME , PhaseAfterStep .NAME ),
211+ MockStep secondAfter = new MockStep (new StepKey ("first_phase" , PhaseCompleteStep .NAME , PhaseCompleteStep .NAME ),
216212 secondActionStep .getKey ());
217213 MockStep firstActionAnotherStep = new MockStep (new StepKey ("first_phase" , "test" , "bar" ), secondAfter .getKey ());
218214 MockStep firstActionStep = new MockStep (new StepKey ("first_phase" , "test" , "foo" ), firstActionAnotherStep .getKey ());
219- MockStep firstAfter = new MockStep (new StepKey ("new" , PhaseAfterStep .NAME , PhaseAfterStep .NAME ), firstActionStep .getKey ());
215+ MockStep firstAfter = new MockStep (new StepKey ("new" , PhaseCompleteStep .NAME , PhaseCompleteStep .NAME ), firstActionStep .getKey ());
220216 MockStep init = new MockStep (InitializePolicyContextStep .KEY , firstAfter .getKey ());
221217
222218 lifecycleName = randomAlphaOfLengthBetween (1 , 20 );
@@ -231,17 +227,17 @@ public void testToStepsWithTwoPhases() {
231227 phases .put (secondPhase .getName (), secondPhase );
232228 LifecyclePolicy policy = new LifecyclePolicy (TestLifecycleType .INSTANCE , lifecycleName , phases );
233229
234- List <Step > steps = policy .toSteps (client , nowSupplier );
230+ List <Step > steps = policy .toSteps (client );
235231 assertThat (steps .size (), equalTo (7 ));
236232 assertThat (steps .get (0 ).getClass (), equalTo (InitializePolicyContextStep .class ));
237233 assertThat (steps .get (0 ).getKey (), equalTo (init .getKey ()));
238234 assertThat (steps .get (0 ).getNextStepKey (), equalTo (init .getNextStepKey ()));
239- assertThat (steps .get (1 ).getClass (), equalTo (PhaseAfterStep .class ));
235+ assertThat (steps .get (1 ).getClass (), equalTo (PhaseCompleteStep .class ));
240236 assertThat (steps .get (1 ).getKey (), equalTo (firstAfter .getKey ()));
241237 assertThat (steps .get (1 ).getNextStepKey (), equalTo (firstAfter .getNextStepKey ()));
242238 assertThat (steps .get (2 ), equalTo (firstActionStep ));
243239 assertThat (steps .get (3 ), equalTo (firstActionAnotherStep ));
244- assertThat (steps .get (4 ).getClass (), equalTo (PhaseAfterStep .class ));
240+ assertThat (steps .get (4 ).getClass (), equalTo (PhaseCompleteStep .class ));
245241 assertThat (steps .get (4 ).getKey (), equalTo (secondAfter .getKey ()));
246242 assertThat (steps .get (4 ).getNextStepKey (), equalTo (secondAfter .getNextStepKey ()));
247243 assertThat (steps .get (5 ), equalTo (secondActionStep ));
@@ -337,7 +333,7 @@ public void testGetNextValidStep() {
337333 currentStep = new StepKey ("phase_1" , "action_3" , "step_missing" );
338334 nextStep = policy .getNextValidStep (currentStep );
339335 assertNotNull (nextStep );
340- assertEquals (new StepKey ("phase_1" , PhaseAfterStep .NAME , PhaseAfterStep .NAME ), nextStep );
336+ assertEquals (new StepKey ("phase_1" , PhaseCompleteStep .NAME , PhaseCompleteStep .NAME ), nextStep );
341337
342338 // current action exists but step does not and action is last in the
343339 // last phase
@@ -356,7 +352,7 @@ public void testGetNextValidStep() {
356352 currentStep = new StepKey ("phase_1" , "action_4" , "step_2" );
357353 nextStep = policy .getNextValidStep (currentStep );
358354 assertNotNull (nextStep );
359- assertEquals (new StepKey ("phase_1" , PhaseAfterStep .NAME , PhaseAfterStep .NAME ), nextStep );
355+ assertEquals (new StepKey ("phase_1" , PhaseCompleteStep .NAME , PhaseCompleteStep .NAME ), nextStep );
360356
361357 // current action no longer exists and action was last in the last phase
362358 currentStep = new StepKey ("phase_4" , "action_4" , "step_2" );
@@ -368,7 +364,7 @@ public void testGetNextValidStep() {
368364 currentStep = new StepKey ("phase_3" , "action_2" , "step_2" );
369365 nextStep = policy .getNextValidStep (currentStep );
370366 assertNotNull (nextStep );
371- assertEquals (new StepKey ("phase_2" , PhaseAfterStep .NAME , PhaseAfterStep .NAME ), nextStep );
367+ assertEquals (new StepKey ("phase_2" , PhaseCompleteStep .NAME , PhaseCompleteStep .NAME ), nextStep );
372368
373369 // current phase no longer exists and was last phase
374370 currentStep = new StepKey ("phase_5" , "action_2" , "step_2" );
0 commit comments