def test_close_should_mark_as_finished(self):
# given:
dbfile = '%s/dbfile.db' % self.tmp_folder
options = {'database': dbfile}
# when:
persistence = SqlitePersistence(dict(options=options), meta())
persistence.commit_position(dict(read=10000))
persistence.close()
# then:
result = query_db(dbfile, 'SELECT * FROM job')
self.assertTrue(result[0]['job_finished'], "Job should be marked as finished")
def test_create_commit_and_get_position(self):
# given:
dbfile = '%s/dbfile.db' % self.tmp_folder
options = {'database': dbfile}
# when:
persistence = SqlitePersistence(dict(options=options), meta())
persistence.commit_position(dict(read=10000))
persistence.commit_position(dict(read=20000))
# then:
result = query_db(dbfile, 'SELECT * FROM job WHERE id = %s'
% persistence.persistence_state_id)
self.assertEqual(dict(read=20000), persistence.get_last_position())
self.assertFalse(result[0]['job_finished'], "Job should not marked as finished")