Skip to content

Commit 316b87d

Browse files
author
Jeny Sadadia
committed
test: mock database index creation
Need to mock `Database.create_indexes` function in unit tests used for app startup. Signed-off-by: Jeny Sadadia <[email protected]>
1 parent f7ee582 commit 316b87d

File tree

4 files changed

+20
-6
lines changed

4 files changed

+20
-6
lines changed

test/conftest.py

+9
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,15 @@ def mock_db_create(mocker):
4444
return async_mock
4545

4646

47+
@pytest.fixture
48+
def mock_db_create_indexes(mocker):
49+
"""Mocks async call to Database class method used to create indexes"""
50+
async_mock = AsyncMock()
51+
mocker.patch('api.db.Database.create_indexes',
52+
side_effect=async_mock)
53+
return async_mock
54+
55+
4756
@pytest.fixture
4857
def mock_db_find_by_attributes(mocker):
4958
"""

test/test_subscribe_handler.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616

1717

1818
def test_subscribe_endpoint(mock_get_current_user, mock_init_sub_id,
19-
mock_subscribe):
19+
mock_subscribe, mock_db_create_indexes):
2020
"""
2121
Test Case : Test KernelCI API /subscribe endpoint
2222
Expected Result :

test/test_unsubscribe_handler.py

+4-2
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,8 @@
1515

1616

1717
def test_unsubscribe_endpoint(mock_get_current_user,
18-
mock_init_sub_id, mock_unsubscribe):
18+
mock_init_sub_id, mock_unsubscribe,
19+
mock_db_create_indexes):
1920
"""
2021
Test Case : Test KernelCI API /unsubscribe endpoint positive path
2122
Expected Result :
@@ -32,7 +33,8 @@ def test_unsubscribe_endpoint(mock_get_current_user,
3233

3334

3435
def test_unsubscribe_endpoint_empty_response(mock_get_current_user,
35-
mock_init_sub_id):
36+
mock_init_sub_id,
37+
mock_db_create_indexes):
3638
"""
3739
Test Case : Test KernelCI API /unsubscribe endpoint negative path
3840
Expected Result :

test/test_user_handler.py

+6-3
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,8 @@
1818

1919

2020
def test_create_regular_user(mock_init_sub_id, mock_get_current_admin_user,
21-
mock_db_create, mock_publish_cloudevent):
21+
mock_db_create, mock_publish_cloudevent,
22+
mock_db_create_indexes):
2223
"""
2324
Test Case : Test KernelCI API /user endpoint to create regular user
2425
when requested with admin user's bearer token
@@ -47,7 +48,8 @@ def test_create_regular_user(mock_init_sub_id, mock_get_current_admin_user,
4748

4849

4950
def test_create_admin_user(mock_init_sub_id, mock_get_current_admin_user,
50-
mock_db_create, mock_publish_cloudevent):
51+
mock_db_create, mock_publish_cloudevent,
52+
mock_db_create_indexes):
5153
"""
5254
Test Case : Test KernelCI API /user endpoint to create admin user
5355
when requested with admin user's bearer token
@@ -76,7 +78,8 @@ def test_create_admin_user(mock_init_sub_id, mock_get_current_admin_user,
7678

7779

7880
def test_create_user_endpoint_negative(mock_init_sub_id, mock_get_current_user,
79-
mock_publish_cloudevent):
81+
mock_publish_cloudevent,
82+
mock_db_create_indexes):
8083
"""
8184
Test Case : Test KernelCI API /user endpoint when requested
8285
with regular user's bearer token

0 commit comments

Comments
 (0)