Cutlery/calibre
1
0
Fork 0
mirror of https://github.com/kovidgoyal/calibre.git synced 2025-08-05 08:40:13 -04:00
Code Issues Packages Projects Releases Wiki Activity

...

Browse Source
This commit is contained in:
Kovid Goyal 2016-06-23 10:14:33 +05:30
parent c1924546c4
commit 698eef053a
1 changed files with 25 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
src/calibre/srv/tests/loop.py
View File

@ -233,12 +233,16 @@ class LoopTest(BaseTest):
class FakeLog(list): class FakeLog(list):
def error(self, *args): def error(self, *args):
self.append(' '.join(args)) self.append(' '.join(args))
jm = JobsManager(O(1, 5), FakeLog())
job_id = jm.start_job('simple test', 'calibre.srv.jobs', 'sleep_test', args=(1.0,))
job_id2 = jm.start_job('t2', 'calibre.srv.jobs', 'sleep_test', args=(3,))
jid = jm.start_job('err test', 'calibre.srv.jobs', 'error_test')
status = jm.job_status(job_id)[0]
s = ('waiting', 'running') s = ('waiting', 'running')
jm = JobsManager(O(1, 5), FakeLog())
# Start jobs
job_id1 = jm.start_job('simple test', 'calibre.srv.jobs', 'sleep_test', args=(1.0,))
job_id2 = jm.start_job('t2', 'calibre.srv.jobs', 'sleep_test', args=(3,))
job_id3 = jm.start_job('err test', 'calibre.srv.jobs', 'error_test')
# Job 1
job_id = job_id1
status = jm.job_status(job_id)[0]
self.assertIn(status, s) self.assertIn(status, s)
status2 = jm.job_status(job_id2)[0] status2 = jm.job_status(job_id2)[0]
self.assertEqual(status2, 'waiting') self.assertEqual(status2, 'waiting')
@ -249,17 +253,24 @@ class LoopTest(BaseTest):
self.assertFalse(was_aborted) self.assertFalse(was_aborted)
self.assertFalse(tb) self.assertFalse(tb)
self.assertEqual(result, 1.0) self.assertEqual(result, 1.0)
status2 = jm.job_status(job_id2)[0]
time.sleep(0.01) # Job 2
self.assertEqual(status2, 'running') job_id = job_id2
jm.abort_job(job_id2)
self.assertTrue(jm.wait_for_running_job(job_id2))
status, result, tb, was_aborted = jm.job_status(job_id2)
self.assertTrue(was_aborted)
while jm.job_status(job_id)[0] == 'waiting': while jm.job_status(job_id)[0] == 'waiting':
time.sleep(0.01) time.sleep(0.01)
self.assertIn(jm.wait_for_running_job(jid), (True, None)) status2 = jm.job_status(job_id)[0]
status, result, tb, was_aborted = jm.job_status(jid) self.assertEqual(status2, 'running')
jm.abort_job(job_id)
self.assertTrue(jm.wait_for_running_job(job_id))
status, result, tb, was_aborted = jm.job_status(job_id)
self.assertTrue(was_aborted)
# Job 3
job_id = job_id3
while jm.job_status(job_id)[0] == 'waiting':
time.sleep(0.01)
self.assertIn(jm.wait_for_running_job(job_id), (True, None))
status, result, tb, was_aborted = jm.job_status(job_id)
self.assertEqual(status, 'finished') self.assertEqual(status, 'finished')
self.assertFalse(was_aborted) self.assertFalse(was_aborted)
self.assertTrue(tb), self.assertIn('a testing error', tb) self.assertTrue(tb), self.assertIn('a testing error', tb)