Exemplo n.º 1
0
    def test_list(self):
        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
Exemplo n.º 2
0
    def test_list(self):
        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