Python SessionManager Examples

Example #1

File: test_sessionmanager.py Project: cfregly/sparkmagic

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

Example #2

File: test_sessionmanager.py Project: mindis/sparkmagic

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

Example #3

File: test_sessionmanager.py Project: cfregly/sparkmagic

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()

Example #4

File: test_sessionmanager.py Project: cfregly/sparkmagic

def test_get_any_client():
    client = MagicMock()
    manager = SessionManager()

    manager.add_session("name", client)

    assert_equals(client, manager.get_any_session())

Example #5

File: test_sessionmanager.py Project: mindis/sparkmagic

def test_get_any_client():
    client = MagicMock()
    manager = SessionManager()

    manager.add_session("name", client)

    assert_equals(client, manager.get_any_session())

Example #6

File: test_sessionmanager.py Project: mindis/sparkmagic

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

Example #7

File: test_sessionmanager.py Project: cfregly/sparkmagic

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()))

Example #8

File: test_sessionmanager.py Project: mindis/sparkmagic

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()))

Example #9

File: test_sessionmanager.py Project: cfregly/sparkmagic

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

Example #10

File: test_sessionmanager.py Project: cfregly/sparkmagic

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()

Example #11

File: test_sessionmanager.py Project: mindis/sparkmagic

def test_delete_client():
    client = MagicMock()
    manager = SessionManager()

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

    manager.get_session("name")

Example #12

File: test_sessionmanager.py Project: cfregly/sparkmagic

def test_delete_client():
    client = MagicMock()
    manager = SessionManager()

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

    manager.get_session("name")

Example #13

File: test_sessionmanager.py Project: mindis/sparkmagic

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()

Example #14

File: test_sessionmanager.py Project: mindis/sparkmagic

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()

Example #15

File: test_sessionmanager.py Project: mindis/sparkmagic

def test_delete_client_throws_when_client_not_exists():
    manager = SessionManager()

    manager.delete_client("name")

Example #16

File: test_sessionmanager.py Project: mindis/sparkmagic

def test_get_any_client_raises_exception_with_no_client():
    manager = SessionManager()

    manager.get_any_session()

Example #17

File: test_sessionmanager.py Project: cfregly/sparkmagic

def test_get_client_throws_when_client_not_exists():
    manager = SessionManager()

    manager.get_session("name")

Example #18

File: test_sessionmanager.py Project: cfregly/sparkmagic

def test_get_any_client_raises_exception_with_no_client():
    manager = SessionManager()

    manager.get_any_session()

Example #19

File: test_sessionmanager.py Project: mindis/sparkmagic

def test_get_client_throws_when_client_not_exists():
    manager = SessionManager()

    manager.get_session("name")

Example #20

File: test_sessionmanager.py Project: cfregly/sparkmagic

def test_add_client_throws_when_client_exists():
    client = MagicMock()
    manager = SessionManager()

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

Example #21

File: test_sessionmanager.py Project: cfregly/sparkmagic

def test_delete_client_throws_when_client_not_exists():
    manager = SessionManager()

    manager.delete_client("name")

Example #22

File: test_sessionmanager.py Project: mindis/sparkmagic

def test_add_client_throws_when_client_exists():
    client = MagicMock()
    manager = SessionManager()

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