def _create_session(self, kind=constants.SESSION_KIND_SPARK, session_id="-1", sql_created=False, http_client=None): if http_client is None: http_client = MagicMock() ipython_display = MagicMock() session = LivySession(http_client, {"kind": kind}, ipython_display, session_id, sql_created) return session
def _create_session(kind=constants.SESSION_KIND_SPARK, session_id=-1, sql_created=False, http_client=None, spark_events=None): if http_client is None: http_client = MagicMock() if spark_events is None: spark_events = MagicMock() ipython_display = MagicMock() session = LivySession(http_client, {"kind": kind}, ipython_display, session_id, sql_created, spark_events) return session
def test_start_passes_in_all_properties(self): http_client = MagicMock() http_client.post.return_value = DummyResponse(201, self.session_create_json) conf.override_all({ "status_sleep_seconds": 0.01, "statement_sleep_seconds": 0.01 }) kind = constants.SESSION_KIND_SPARK properties = {"kind": kind, "extra": 1} ipython_display = MagicMock() session = LivySession(http_client, properties, ipython_display) session.start() conf.load() http_client.post.assert_called_with("/sessions", [201], properties)
def test_start_passes_in_all_properties(self): http_client = MagicMock() http_client.post_session.return_value = self.session_create_json http_client.get_session.return_value = self.ready_sessions_json conf.override_all({ "status_sleep_seconds": 0.01, "statement_sleep_seconds": 0.01 }) kind = constants.SESSION_KIND_SPARK properties = {"kind": kind, "extra": 1} ipython_display = MagicMock() session = LivySession(http_client, properties, ipython_display) session.start(create_sql_context=False) conf.load() http_client.post_session.assert_called_with(properties)