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)
