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)
