def test_get_any_client(): client = MagicMock() manager = SessionManager() manager.add_session("name", client) assert_equals(client, manager.get_any_session())
def test_get_session_id_for_client_not_there(): manager = SessionManager() 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 = SessionManager() manager.get_sessions_list = MagicMock(return_value=["name"]) manager._sessions["name"] = MagicMock() id = manager.get_session_id_for_client("name") assert id is not None
def test_client_names_returned(): client = MagicMock() manager = SessionManager() manager.add_session("name0", client) manager.add_session("name1", client) assert_equals({"name0", "name1"}, set(manager.get_sessions_list()))
def test_delete_client(): client = MagicMock() manager = SessionManager() manager.add_session("name", client) manager.delete_client("name") manager.get_session("name")
def test_get_any_client_raises_exception_with_two_clients(): client = MagicMock() manager = SessionManager() manager.add_session("name0", client) manager.add_session("name1", client) manager.get_any_session()
def test_clean_up(): client0 = MagicMock() client1 = MagicMock() manager = SessionManager() manager.add_session("name0", client0) manager.add_session("name1", client1) manager.clean_up_all() client0.delete.assert_called_once_with() client1.delete.assert_called_once_with()
def test_get_session_name_by_id_endpoint(): manager = SessionManager() id_to_search = "0" endpoint_to_search = "endpoint" name_to_search = "name" name = manager.get_session_name_by_id_endpoint(id_to_search, endpoint_to_search) assert_equals(None, name) session = MagicMock() type(session).id = PropertyMock(return_value=int(id_to_search)) session.endpoint = endpoint_to_search manager.add_session(name_to_search, session) name = manager.get_session_name_by_id_endpoint(id_to_search, endpoint_to_search) assert_equals(name_to_search, name)
def test_get_client_throws_when_client_not_exists(): manager = SessionManager() manager.get_session("name")
def test_get_any_client_raises_exception_with_no_client(): manager = SessionManager() manager.get_any_session()
def test_add_client_throws_when_client_exists(): client = MagicMock() manager = SessionManager() manager.add_session("name", client) manager.add_session("name", client)
def test_delete_client_throws_when_client_not_exists(): manager = SessionManager() manager.delete_client("name")
def get_session_manager(): ipython_display = MagicMock() return SessionManager(ipython_display)