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)
