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_can_build_all_clients():
factory = LivyClientFactory()
for kind in Constants.session_kinds_supported:
session = MagicMock()
session.kind = kind
factory.build_client(session)
def test_build_unknown_language():
session = MagicMock()
session.kind = "unknown"
factory = LivyClientFactory()
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)