vbatts/quay
Archived
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

Handle read timeouts from etcd when watching a key.

Browse source
This commit is contained in:
Jake Moshenko 2014-12-23 12:13:49 -05:00
parent 29bd428817
commit 709e571b78
2 changed files with 19 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

13
test/test_buildman.py
View file

@ -6,6 +6,7 @@ import time
from trollius import coroutine, get_event_loop, From, Future, sleep
from mock import Mock
from threading import Event
from urllib3.exceptions import ReadTimeoutError
from buildman.manager.executor import BuilderExecutor
from buildman.manager.ephemeral import (EphemeralBuilderManager, ETCD_BUILDER_PREFIX,
@ -176,3 +177,15 @@ class TestEphemeral(unittest.TestCase):
self.job_heartbeat_callback.assert_called_once_with(self.mock_job)
self.assertEqual(self.etcd_client_mock.write.call_count, 1)
self.assertEqual(self.etcd_client_mock.write.call_args_list[0][0][0], self.mock_job_key)
@async_test
def test_etcd_read_timeout(self):
# Send a signal to the callback that a worker key has been changed
read_timeout_future = Future()
read_timeout_future.set_exception(ReadTimeoutError(None, None, None))
self.manager._handle_key_expiration(read_timeout_future)
yield From(sleep(.01))
self.assertEquals(self.test_executor.stop_builder.call_count, 0)