class TestWithFakeRemoteAgent(TestRemoteJobManager): def setUp(self): self.agent = FakeRemoteAgent(self._get_port(), self.handle_job_func, self.update_image_aliases_func, self.get_task_directory_hashes_func, self.update_task_directory_func) TestJobManager.setUp(self) wait = 0 while self.job_manager.number_agents_available() != 1 and wait < 3: time.sleep(30) wait += 1 assert self.job_manager.number_agents_available() == 1 def tearDown(self): self.agent.close() TestJobManager.tearDown(self) def handle_job_func(self, job_id, course_id, task_id, inputdata, debug, callback_status): return {"result": "success", "grade": 100.0} def update_image_aliases_func(self, image_aliases): pass def get_task_directory_hashes_func(self): return {} def update_task_directory_func(self, remote_tar_file, to_delete): pass
def setUp(self): self.agent = FakeRemoteAgent(self._get_port(), self.handle_job_func, self.update_image_aliases_func, self.get_task_directory_hashes_func, self.update_task_directory_func) TestJobManager.setUp(self) wait = 0 while self.job_manager.number_agents_available() != 1 and wait < 3: time.sleep(30) wait += 1 assert self.job_manager.number_agents_available() == 1