def test_get_session_id_for_client_not_there():
manager = ClientManager()
manager.get_sessions_list = MagicMock(return_value=[])
id = manager.get_session_id_for_client("name")
assert id is None
def test_get_any_client():
client = MagicMock()
manager = ClientManager()
manager.add_client("name", client)
assert_equals(client, manager.get_any_client())
def test_client_names_returned():
client = MagicMock()
manager = ClientManager()
manager.add_client("name0", client)
manager.add_client("name1", client)
assert_equals({"name0", "name1"}, set(manager.get_sessions_list()))
def test_get_session_id_for_client():
manager = ClientManager()
manager.get_sessions_list = MagicMock(return_value=["name"])
manager._livy_clients["name"] = MagicMock()
id = manager.get_session_id_for_client("name")
assert id is not None
def test_delete_client():
client = MagicMock()
manager = ClientManager()
manager.add_client("name", client)
manager.delete_client("name")
manager.get_client("name")
def test_serialize_periodically():
conf.override_all({conf.serialize_period_seconds.__name__: 0.1,
conf.serialize_periodically.__name__: True})
serializer = MagicMock()
ClientManager(serializer)
time.sleep(0.5)
assert serializer.serialize_state.call_count >= 1
conf.load()
def test_get_any_client_raises_exception_with_two_clients():
client = MagicMock()
manager = ClientManager()
manager.add_client("name0", client)
manager.add_client("name1", client)
manager.get_any_client()
def test_clean_up():
client0 = MagicMock()
client1 = MagicMock()
manager = ClientManager()
manager.add_client("name0", client0)
manager.add_client("name1", client1)
manager.clean_up_all()
client0.close_session.assert_called_once_with()
client1.close_session.assert_called_once_with()
def test_deserialize_on_creation():
serializer = MagicMock()
serializer.deserialize_state.return_value = [("py", None), ("sc", None)]
manager = ClientManager(serializer)
assert "py" in manager.get_sessions_list()
assert "sc" in manager.get_sessions_list()
serializer = MagicMock()
manager = ClientManager(serializer)
assert len(manager.get_sessions_list()) == 0
def test_get_any_client_raises_exception_with_no_client():
manager = ClientManager()
manager.get_any_client()
def test_get_client_throws_when_client_not_exists():
manager = ClientManager()
manager.get_client("name")
def test_add_client_throws_when_client_exists():
client = MagicMock()
manager = ClientManager()
manager.add_client("name", client)
manager.add_client("name", client)
def test_delete_client_throws_when_client_not_exists():
manager = ClientManager()
manager.delete_client("name")
