def test_directory():
    directory = Directory()
    mock_actor_1 = MockActor(1)
    mock_actor_2 = MockActor(2)

    directory.add_actor("mock-actor-1", mock_actor_1)
    directory.add_actor("mock-actor-2", mock_actor_2)

    assert(mock_actor_1 == directory.get_actor("mock-actor-1"))
    assert(mock_actor_2 == directory.get_actor("mock-actor-2"))