def test_finaliser_successful_pipeline(self): data = AssemblerState() assembler = Assembler("Finalizer test", InputValidator(), HelloStage(), Config()) assembler.set_finaliser(TestFinalStage()) assembler.init_assembler() # Run assembler which will succeed assembler.run(data) # Finalizer should be executed self.assertTrue(data.final_ran)
def test_finaliser_fail_pipeline(self): # Ensure pipeline will crash data = AssemblerState() data.text = "" assembler = Assembler("Finalizer test", InputValidator(), HelloStage(), Config()) assembler.set_finaliser(TestFinalStage()) assembler.init_assembler() # Run assembler which will fail assembler.run(data) # Finalizer should still be executed self.assertTrue(data.final_ran)