def test_failed_task_delete_by_storage(self, mock_session):
fake_failed_task_storage_id = [models.FailedTask().storage_id]
mock_session.return_value.__enter__.return_value.query.return_value \
= fake_failed_task_storage_id
result = db_api \
.task_delete_by_storage(context, fake_failed_task_storage_id)
assert result is None
def test_failed_task_delete(self, mock_session):
fake_failed_task = [models.FailedTask().id]
mock_session.return_value.__enter__.return_value.query.return_value \
= fake_failed_task
result = db_api.task_delete(context, fake_failed_task)
assert result is None
def test_failed_task_create(self, mock_session):
fake_failed_task = models.FailedTask()
mock_session.return_value.__enter__.return_value.query.return_value \
= fake_failed_task
result = db_api.failed_task_create(context, fake_failed_task)
assert len(result) == 0