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)