From 03ebfa3f6e02ee8be7173dafb9a0e246d8960656 Mon Sep 17 00:00:00 2001 From: Aaron Suarez Date: Sun, 5 Jul 2020 19:18:58 -0500 Subject: [PATCH] Raise DBAPIError rather than None --- tests/conftest.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/conftest.py b/tests/conftest.py index fd34e2fb..5549a26a 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -4,6 +4,7 @@ from app import app from app import db as _db from app.utils import standardize_response +from sqlalchemy.exc import DBAPIError TEST_DATABASE_URI = 'sqlite:///:memory:' @@ -190,7 +191,7 @@ def fake_category_query_error(mocker): Mocks an exception being raised during a query to test error handling """ - mocker.patch('app.models.Category.query', new=None) + mocker.patch('app.models.Category.query', side_effect=DBAPIError) @pytest.fixture(scope='function') @@ -199,7 +200,7 @@ def fake_language_query_error(mocker): Mocks an exception being raised during a query to test error handling """ - mocker.patch('app.models.Language.query', new=None) + mocker.patch('app.models.Language.query', side_effect=DBAPIError) @pytest.fixture(scope='function')