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)
