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_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