def test_heartbeat_cancel(self): heartbeat = FakeHeartbeatCancel(max_count=10) heartbeater = HeartbeatProcess(heartbeat, interval=0.1) token = uuid4() heartbeater.run(token, FakeTask('test_task')) self.assertEqual(heartbeat._token, token)
def test_heartbeat_cancel(self): heartbeat = FakeHeartbeatCancel(max_count=10) heartbeater = HeartbeatProcess(heartbeat, interval=0.1) token = uuid4() heartbeater.run(token, FakeTask('test_task')) self.assertEquals(heartbeat._token, token)
def test_heartbeat_doesnotexist_error(self): error_type = 'Unknown execution: blah' heartbeat = FakeHeartbeatRaises( swf.exceptions.DoesNotExistError('error', error_type), 10) heartbeater = HeartbeatProcess(heartbeat, interval=0.1) token = uuid4() heartbeater.run(token, FakeTask('test_task')) self.assertEqual(heartbeat._token, token)
def test_heartbeat_ok(self): heartbeat = FakeHeartbeat(max_count=1) heartbeater = HeartbeatProcess(heartbeat, interval=0.1) token = uuid4() try: heartbeater.run(token, FakeTask('test_task')) except StopIteration: pass self.assertEqual(heartbeat._token, token)
def test_heartbeat_ok(self): heartbeat = FakeHeartbeat(max_count=1) heartbeater = HeartbeatProcess(heartbeat, interval=0.1) token = uuid4() try: heartbeater.run(token, FakeTask('test_task')) except StopIteration: pass self.assertEquals(heartbeat._token, token)
def test_heartbeat_doesnotexist_error(self): error_type = 'Unknown execution: blah' heartbeat = FakeHeartbeatRaises( swf.exceptions.DoesNotExistError('error', error_type), 10) heartbeater = HeartbeatProcess(heartbeat, interval=0.1) token = uuid4() heartbeater.run(token, FakeTask('test_task')) self.assertEquals(heartbeat._token, token)
def test_heartbeat_running(self): max_count = 3 heartbeat = FakeHeartbeat(max_count) heartbeater = HeartbeatProcess(heartbeat, interval=0.1) token = uuid4() try: heartbeater.run(token, FakeTask("test_task")) except StopIteration: pass self.assertEqual(heartbeat._max_count, max_count) self.assertEqual(heartbeat._token, token)
def test_heartbeat_process_kill_parent(self): heartbeat = FakeHeartbeat(1000) heartbeater = HeartbeatProcess(heartbeat, interval=0.1) token = uuid4() task = FakeTask('test_task') result_queue = mp.Queue() handler = FakeTaskHandler(heartbeater.run, (token, task), {}, result_queue) handler_process = mp.Process(target=handler) handler_process.start() time.sleep(2) heartbeat_pid = result_queue.get() handler_process.terminate() os.kill(handler_process.pid, signal.SIGQUIT) handler_process.join() time.sleep(2) with self.assertRaises(OSError): os.kill(heartbeat_pid, signal.SIGKILL)
def test_heartbeat_non_int(self): with self.assertRaises(ValueError): HeartbeatProcess(lambda *args: None, interval='NOT_AN_INTEGER')
def test_heartbeat_0(self): with self.assertRaises(ValueError): HeartbeatProcess(lambda *args: None, interval=0)