def test_mass_delete(self):
datastore_mock = Mock()
storage = DatastoreStorage(datastore_mock)
q_mock = Mock()
q_mock.fetch.return_value = [
Mock(key=datastore.Key('task', n, dataset_id='test'))
for n in range(102)]
datastore_mock.query.return_value = q_mock
storage.delete_tasks()
assert datastore_mock.delete_multi.call_count == 2
def test_mass_delete(self):
datastore_mock = Mock()
storage = DatastoreStorage(datastore_mock)
q_mock = Mock()
q_mock.fetch.return_value = [
Mock(key=datastore.Key('task', n, project='test'))
for n in range(102)
]
datastore_mock.query.return_value = q_mock
storage.delete_tasks()
assert datastore_mock.delete_multi.call_count == 2
