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)
