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))