Skip to content

Commit e159afe

Browse files
committed
adjusted tests
1 parent 34a11af commit e159afe

File tree

1 file changed

+15
-1
lines changed

1 file changed

+15
-1
lines changed

tests/utils.py

Lines changed: 15 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,14 +20,28 @@
2020
ROOT_DIR = pathlib.Path(__file__).parent.parent
2121

2222

23+
class MockRequest:
24+
def __init__(self, headers):
25+
self.headers = headers
26+
27+
2328
# class for mocking responses
2429
class MockResponse:
25-
def __init__(self, json_data, status_code, uri, content=None, headers=None):
30+
def __init__(
31+
self,
32+
json_data,
33+
status_code,
34+
uri,
35+
content=None,
36+
headers=None,
37+
request_headers=None,
38+
):
2639
self.headers = headers
2740
self.json_data = json_data
2841
self.status_code = status_code
2942
self.content = content
3043
self.url = "http://localhost:80" + uri
44+
self.request = MockRequest(request_headers)
3145

3246
def json(self):
3347
return self.json_data

0 commit comments

Comments
 (0)