def test_task_delete_by_storage(self, mock_session):
fake_task_storage_id = [models.Task().storage_id]
mock_session.return_value.__enter__.return_value.query.return_value \
= fake_task_storage_id
result = db_api \
.task_delete_by_storage(context, fake_task_storage_id)
assert result is None
def test_task_delete(self, mock_session):
fake_task = [models.Task().id]
mock_session.return_value.__enter__.return_value.query.return_value \
= fake_task
result = db_api.task_delete(context, fake_task)
assert result is None
def test_task_create(self, mock_session):
fake_task = models.Task()
mock_session.return_value.__enter__.return_value.query.return_value \
= fake_task
result = db_api.task_create(context, fake_task)
assert len(result) == 0