def test_start_passes_in_all_properties(self):
self.http_client.post_session.return_value = self.session_create_json
self.http_client.get_session.return_value = self.ready_sessions_json
self.http_client.get_statement.return_value = self.ready_statement_json
kind = constants.SESSION_KIND_SPARK
properties = {"kind": kind, "extra": 1}
ipython_display = MagicMock()
session = LivySession(self.http_client, properties, ipython_display)
session.start()
self.http_client.post_session.assert_called_with(properties)
def test_start_passes_in_all_properties(self):
self.http_client.post_session.return_value = self.session_create_json
self.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(self.http_client, properties, ipython_display)
session.start(create_sql_context=False)
conf.override_all({})
self.http_client.post_session.assert_called_with(properties)
def test_start_passes_in_all_properties(self):
self.http_client.post_session.return_value = self.session_create_json
self.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(self.http_client, properties, ipython_display)
session.start(create_sql_context=False)
conf.override_all({})
self.http_client.post_session.assert_called_with(properties)
