Пример #1
0
def test_list():
    datastore_mock = Mock()
    storage = DatastoreStorage(datastore_mock)
    t = Task('1', sum, (), {})
    q_mock = Mock()
    q_mock.fetch.return_value = [t]
    datastore_mock.query.return_value = q_mock

    tasks = storage.list_tasks()

    assert len(tasks) == 1
    assert tasks[0] == t
Пример #2
0
def test_list():
    datastore_mock = Mock()
    storage = DatastoreStorage(datastore_mock)
    t = Task('1', sum, (), {})
    q_mock = Mock()
    q_mock.fetch.return_value = [t]
    datastore_mock.query.return_value = q_mock

    tasks = storage.list_tasks()

    assert len(tasks) == 1
    assert tasks[0] == t