def mock_runner_with_status_response(self, response): runner = BuildRunner('url', {}, 'mellon') start_response = Response() start_response.json = MagicMock(return_value={"build_id": "1"}) runner._network.post = MagicMock(return_value=start_response) status_response = Response() status_response.json = MagicMock(return_value=response) runner._network.get = MagicMock(return_value=status_response) runner._download_and_extract_results = MagicMock() return runner
def mock_runner_with_status_response(self, response): runner = BuildRunner('url', {}, 'mellon') start_response = Response() start_response.json = MagicMock(return_value={"build_id": "1"}) runner._network.post = MagicMock(return_value=start_response) status_response = Response() status_response.json = MagicMock(return_value=response) runner._network.get = MagicMock(return_value=status_response) runner._download_and_extract_results = MagicMock() return runner
def _create_build_runner_instance(self, response): mock_get_logger = self.patch('app.client.build_runner.get_logger') logger = mock_get_logger.return_value build_id = '1' runner = BuildRunner('url', {}, 'mellon') start_response = Response() start_response.json = MagicMock(return_value={'build_id': build_id}) runner._network.post = MagicMock(return_value=start_response) status_response = Response() status_response.json = MagicMock(return_value=response) runner._network.get = MagicMock(return_value=status_response) runner._download_and_extract_results = MagicMock() return runner, logger, build_id
def _create_build_runner_instance(self, response): mock_get_logger = self.patch('app.client.build_runner.get_logger') logger = mock_get_logger.return_value build_id = '1' runner = BuildRunner('url', {}, 'mellon') start_response = Response() start_response.json = MagicMock(return_value={'build_id': build_id}) runner._network.post = MagicMock(return_value=start_response) status_response = Response() status_response.json = MagicMock(return_value=response) runner._network.get = MagicMock(return_value=status_response) runner._download_and_extract_results = MagicMock() return runner, logger, build_id