def test_create_marker_table(self): """ Is the marker table created as expected for the SQLAlchemyTarget :return: """ target = sqla.SQLAlchemyTarget(CONNECTION_STRING, "test_table", "12312123") target.create_marker_table() self.assertTrue(target.engine.dialect.has_table(target.engine.connect(), target.marker_table))
def test_touch(self): """ Touch takes care of creating a checkpoint for task completion :return: """ target = sqla.SQLAlchemyTarget(CONNECTION_STRING, "test_table", "12312123") target.create_marker_table() self.assertFalse(target.exists()) target.touch() self.assertTrue(target.exists())
def output(self): return sqla.SQLAlchemyTarget(self.connection_string, self.table, self.task_id)