Skip to content

Commit 990c82c

Browse files
committed
Use mocked_session_get by default
One can still define a separate responder function here, but since this is the only response defined for the tests thus far, this seems like a reasonable default.
1 parent 3dfe15f commit 990c82c

File tree

4 files changed

+2
-16
lines changed

4 files changed

+2
-16
lines changed

tests/conftest.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -81,6 +81,7 @@ def mock_client_session_class(request):
8181
"""
8282

8383
httputils.client_session = request.cls.mock_client_session = mock.AsyncMock()
84+
httputils.client_session.get = mocked_session_get
8485
try:
8586
yield
8687
finally:
@@ -94,6 +95,7 @@ async def mock_client_session():
9495
"""
9596

9697
httputils.client_session = mock.AsyncMock()
98+
httputils.client_session.get = mocked_session_get
9799
try:
98100
yield httputils.client_session
99101
finally:

tests/test_csbs.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
11
import pytest
22

33
from app.services.location import csbs
4-
from tests.conftest import mocked_session_get
54

65

76
def mocked_csbs_requests_get(*args, **kwargs):
@@ -28,7 +27,6 @@ def read_file(self):
2827

2928
@pytest.mark.asyncio
3029
async def test_get_locations(mock_client_session):
31-
mock_client_session.get = mocked_session_get
3230
data = await csbs.get_locations()
3331

3432
assert isinstance(data, list)

tests/test_jhu.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,6 @@
44

55
from app import location
66
from app.services.location import jhu
7-
from tests.conftest import mocked_session_get
87
from tests.conftest import mocked_strptime_isoformat
98

109
DATETIME_STRING = "2020-03-17T10:23:22.505550"
@@ -16,7 +15,6 @@ async def test_get_locations(mock_datetime, mock_client_session):
1615
mock_datetime.utcnow.return_value.isoformat.return_value = DATETIME_STRING
1716
mock_datetime.strptime.side_effect = mocked_strptime_isoformat
1817

19-
mock_client_session.get = mocked_session_get
2018
output = await jhu.get_locations()
2119

2220
assert isinstance(output, list)

tests/test_routes.py

Lines changed: 0 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,6 @@
66
import pytest
77
from async_asgi_testclient import TestClient
88

9-
from .conftest import mocked_session_get
109
from .conftest import mocked_strptime_isoformat
1110
from .test_jhu import DATETIME_STRING
1211
from app.main import APP
@@ -34,8 +33,6 @@ def read_file_v1(self, state):
3433
@mock.patch("app.services.location.jhu.datetime")
3534
async def test_root_api(self, mock_datetime):
3635
"""Validate that / returns a 200 and is not a redirect."""
37-
self.mock_client_session.get = mocked_session_get
38-
3936
response = await self.asgi_client.get("/")
4037

4138
assert response.status_code == 200
@@ -45,7 +42,6 @@ async def test_root_api(self, mock_datetime):
4542
async def test_v1_confirmed(self, mock_datetime):
4643
mock_datetime.utcnow.return_value.isoformat.return_value = self.date
4744
mock_datetime.strptime.side_effect = mocked_strptime_isoformat
48-
self.mock_client_session.get = mocked_session_get
4945

5046
state = "confirmed"
5147
expected_json_output = self.read_file_v1(state=state)
@@ -58,7 +54,6 @@ async def test_v1_confirmed(self, mock_datetime):
5854
async def test_v1_deaths(self, mock_datetime):
5955
mock_datetime.utcnow.return_value.isoformat.return_value = self.date
6056
mock_datetime.strptime.side_effect = mocked_strptime_isoformat
61-
self.mock_client_session.get = mocked_session_get
6257

6358
state = "deaths"
6459
expected_json_output = self.read_file_v1(state=state)
@@ -71,7 +66,6 @@ async def test_v1_deaths(self, mock_datetime):
7166
async def test_v1_recovered(self, mock_datetime):
7267
mock_datetime.utcnow.return_value.isoformat.return_value = self.date
7368
mock_datetime.strptime.side_effect = mocked_strptime_isoformat
74-
self.mock_client_session.get = mocked_session_get
7569

7670
state = "recovered"
7771
expected_json_output = self.read_file_v1(state=state)
@@ -84,7 +78,6 @@ async def test_v1_recovered(self, mock_datetime):
8478
async def test_v1_all(self, mock_datetime):
8579
mock_datetime.utcnow.return_value.isoformat.return_value = self.date
8680
mock_datetime.strptime.side_effect = mocked_strptime_isoformat
87-
self.mock_client_session.get = mocked_session_get
8881

8982
state = "all"
9083
expected_json_output = self.read_file_v1(state=state)
@@ -97,7 +90,6 @@ async def test_v1_all(self, mock_datetime):
9790
async def test_v2_latest(self, mock_datetime):
9891
mock_datetime.utcnow.return_value.isoformat.return_value = DATETIME_STRING
9992
mock_datetime.strptime.side_effect = mocked_strptime_isoformat
100-
self.mock_client_session.get = mocked_session_get
10193

10294
state = "latest"
10395
response = await self.asgi_client.get(f"/v2/{state}")
@@ -111,7 +103,6 @@ async def test_v2_latest(self, mock_datetime):
111103
async def test_v2_locations(self, mock_datetime):
112104
mock_datetime.utcnow.return_value.isoformat.return_value = DATETIME_STRING
113105
mock_datetime.strptime.side_effect = mocked_strptime_isoformat
114-
self.mock_client_session.get = mocked_session_get
115106

116107
state = "locations"
117108
response = await self.asgi_client.get("/v2/{}".format(state))
@@ -128,7 +119,6 @@ async def test_v2_locations(self, mock_datetime):
128119
async def test_v2_locations_id(self, mock_datetime):
129120
mock_datetime.utcnow.return_value.isoformat.return_value = DATETIME_STRING
130121
mock_datetime.strptime.side_effect = mocked_strptime_isoformat
131-
self.mock_client_session.get = mocked_session_get
132122

133123
state = "locations"
134124
test_id = 1
@@ -158,7 +148,6 @@ async def test_v2_locations_id(self, mock_datetime):
158148
],
159149
)
160150
async def test_locations_status_code(async_api_client, query_params, expected_status, mock_client_session):
161-
mock_client_session.get = mocked_session_get
162151
response = await async_api_client.get("/v2/locations", query_string=query_params)
163152

164153
print(f"GET {response.url}\n{response}")
@@ -179,7 +168,6 @@ async def test_locations_status_code(async_api_client, query_params, expected_st
179168
],
180169
)
181170
async def test_latest(async_api_client, query_params, mock_client_session):
182-
mock_client_session.get = mocked_session_get
183171
response = await async_api_client.get("/v2/latest", query_string=query_params)
184172

185173
print(f"GET {response.url}\n{response}")

0 commit comments

Comments
 (0)