def test_update_last_row_with_no_data(self):
"""Test the update_last_row function when no data is present."""
txlog = self.obj_factory.make_transaction_log()
worker_name = self.obj_factory.get_unique_unicode()
utils.update_last_row(
worker_name=worker_name, row_id=txlog.id,
timestamp=txlog.timestamp, store_name=self.obj_factory.sstore_name)
result = self.obj_factory.sstore.execute(
u"""SELECT row_id, timestamp
FROM txlog.db_worker_last_row
WHERE worker_id=?""", (worker_name,)).get_one()
self.assertEqual((txlog.id, txlog.timestamp), result)
def test_update_last_row_with_no_data(self):
"""Test the update_last_row function when no data is present."""
txlog = self.obj_factory.make_transaction_log()
worker_name = self.obj_factory.get_unique_unicode()
utils.update_last_row(worker_name=worker_name,
row_id=txlog.id,
timestamp=txlog.timestamp,
store_name=self.obj_factory.sstore_name)
result = self.obj_factory.sstore.execute(
u"""SELECT row_id, timestamp
FROM txlog.db_worker_last_row
WHERE worker_id=?""", (worker_name, )).get_one()
self.assertEqual((txlog.id, txlog.timestamp), result)
def test_update_last_row_with_data(self):
"""Test the update_last_row function when data for this worker is
present.
"""
txlog = self.obj_factory.make_transaction_log()
txlog2 = self.obj_factory.make_transaction_log()
worker_name = self.obj_factory.get_unique_unicode()
self._create_db_worker_last_row_entry(worker_name, txlog)
utils.update_last_row(
worker_name=worker_name, row_id=txlog2.id,
timestamp=txlog2.timestamp)
result = self.obj_factory.store.execute(
"""SELECT row_id, timestamp
FROM txlog_db_worker_last_row
WHERE worker_id=?""", (worker_name,)).get_one()
self.assertEqual((txlog2.id, txlog2.timestamp), result)
