def test_build_session_with_defaults(): factory = LivyClientFactory() connection_string = get_connection_string("url", "user", "pass") language = "python" session = factory.create_session(language, connection_string) assert session.language == language assert session.id == "-1" assert session.started_sql_context is False
def test_build_session(): factory = LivyClientFactory() connection_string = get_connection_string("url", "user", "pass") kind = Constants.session_kind_pyspark properties = {"kind": kind} ipython_display = MagicMock() session = factory.create_session(ipython_display, connection_string, properties, "1", True) assert session.kind == kind assert session.id == "1" assert session.started_sql_context assert session.properties == properties
def test_build_spark(): session = MagicMock() session.kind = Constants.session_kind_spark factory = LivyClientFactory() client = factory.build_client(session) assert isinstance(client, PandasScalaLivyClient)
def test_build_unknown_language(): session = MagicMock() session.kind = "unknown" factory = LivyClientFactory() factory.build_client(session)
def test_can_build_all_clients(): factory = LivyClientFactory() for kind in Constants.session_kinds_supported: session = MagicMock() session.kind = kind factory.build_client(session)
def test_can_build_all_clients(): session = MagicMock() factory = LivyClientFactory() for language in Constants.lang_supported: factory.build_client(language, session)
def test_get_headers(): factory = LivyClientFactory() generated_headers = factory._get_headers() assert_equals(headers, generated_headers)