def setUp(self):
self._command = UpdateWait(clock=mock.create_autospec(spec=time))
self._mock_options = mock_verb_options(self._command)
self._mock_options.jobspec = self.TEST_JOBKEY
self._mock_options.id = 'update_id'
self._fake_context = FakeAuroraCommandContext()
self._fake_context.set_options(self._mock_options)
self._mock_api = self._fake_context.get_api('UNUSED')
self._fetch_call = call(update_key=JobUpdateKey(
job=self.TEST_JOBKEY.to_thrift(), id=self._mock_options.id))
