示例#1
0
def test_get_any_client():
    client = MagicMock()
    manager = ClientManager()

    manager.add_client("name", client)

    assert_equals(client, manager.get_any_client())
示例#2
0
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()
示例#3
0
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()))
示例#4
0
def test_delete_client():
    client = MagicMock()
    manager = ClientManager()

    manager.add_client("name", client)
    manager.delete_client("name")

    manager.get_client("name")
示例#5
0
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()
示例#6
0
def test_add_client_throws_when_client_exists():
    client = MagicMock()
    manager = ClientManager()

    manager.add_client("name", client)
    manager.add_client("name", client)