def test_runs_subaction(self): pipe = Pipeline() pipe.add_action(self.sub0) pipe.add_action(self.sub1) pipe.run_actions(None) self.assertTrue(self.sub0.ran) self.assertTrue(self.sub1.ran) self.assertNotEqual(self.sub0.elapsed_time, 0) self.assertNotEqual(self.sub1.elapsed_time, 0)
def test_keep_connection(self): class KeepConnection(Action): def __init__(self): super(KeepConnection, self).__init__() self.name = "keep-connection" def run(self, connection, args=None): pass pipe = Pipeline() pipe.add_action(KeepConnection()) conn = object() self.assertIs(conn, pipe.run_actions(conn))
def test_change_connection(self): class MakeNewConnection(Action): def __init__(self): super(MakeNewConnection, self).__init__() self.name = "make-new-connection" def run(self, connection, args=None): new_connection = object() return new_connection pipe = Pipeline() pipe.add_action(MakeNewConnection()) conn = object() self.assertIsNot(conn, pipe.run_actions(conn))