def _expect_get_jobs(self, *envs):
self._api.get_jobs(self.ROLE).AndReturn(
Response(responseCode=ResponseCode.OK,
message='Mock OK',
result=Result(getJobsResult=GetJobsResult(configs=set(
JobConfiguration(key=JobKey(
role=self.ROLE, environment=env, name=self.NAME))
for env in envs)))))
def test_start_maintenance(self, mock_api): mock_api.return_value = Response(responseCode=ResponseCode.OK) maintenance = MesosMaintenance(DEFAULT_CLUSTER, 'quiet') maintenance.start_maintenance(MOCK_TEST_HOSTS)
