def test_wait_for_status_times_out(self): http_client = MagicMock() http_client.post.return_value = DummyResponse(201, self.session_create_json) self.get_responses = [DummyResponse(200, self.busy_sessions_json), DummyResponse(200, self.busy_sessions_json), DummyResponse(200, self.ready_sessions_json)] http_client.get.side_effect = self._next_response_get _t_config_hook({ "status_sleep_seconds": 0.011, "statement_sleep_seconds": 6000 }) session = LivySession(http_client, "scala", "-1", False) _t_config_hook({}) session.start() session.wait_for_status("idle", 0.01)
def test_wait_for_status_returns_when_in_state(self): http_client = MagicMock() http_client.post.return_value = DummyResponse(201, self.session_create_json) self.get_responses = [DummyResponse(200, self.busy_sessions_json), DummyResponse(200, self.ready_sessions_json)] http_client.get.side_effect = self._next_response_get _t_config_hook({ "status_sleep_seconds": 0.01, "statement_sleep_seconds": 0.01 }) session = LivySession(http_client, "scala", "-1", False) _t_config_hook({}) session.start() session.wait_for_status("idle", 30) http_client.get.assert_called_with("/sessions", [200]) assert_equals(2, http_client.get.call_count)