def test_finish_job(self):
job = simcity.start_job()
simcity.finish_job(job)
assert_true(job['start'] > 0)
assert_true(job['done'] > 0)
assert_true(simcity.get_job()['done'] > 0)
job = simcity.queue_job(job, 'ssh')
assert_true(job['archive'] > 0)
def test_queue_job(self):
job = simcity.queue_job(Job({'_id': 'aaaa'}), 'ssh')
assert_true(job['queue'] > 0)
assert_equals(len(self.db.saved), 1)
job = simcity.queue_job(simcity.get_job(), 'ssh')
assert_true(job['queue'] > 0)
assert_true(simcity.get_job()['queue'] > 0)
job = simcity.finish_job(job)
assert_true(job['archive'] > 0)
def cleanup_env(self, *kargs, **kwargs):
self.job["tasks_processed"] = self.tasks_processed
simcity.finish_job(self.job, database=self.job_db)
