Add a transaction around the extend_processing call

This commit is contained in:
Joseph Schorr 2015-01-29 18:40:41 -05:00
parent 838bfe23b1
commit 3872d29de9
5 changed files with 19 additions and 15 deletions

View file

@ -231,3 +231,6 @@ 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)
if __name__ == '__main__':
unittest.main()