diff --git a/tests/test_api.py b/tests/test_api.py index 07b31b90..2e49037f 100644 --- a/tests/test_api.py +++ b/tests/test_api.py @@ -19,7 +19,7 @@ class APITestCase(TestCase): "GET", content_type="application/json", ) - self.assertEqual(status, "200 OK") + self.assertEqual(status, 200) study_summaries = json.loads(body)["study_summaries"] self.assertEqual(len(study_summaries), 2) @@ -39,7 +39,7 @@ class APITestCase(TestCase): content_type="application/json", body=json.dumps(request_body), ) - self.assertEqual(status, "201 Created") + self.assertEqual(status, 201) self.assertEqual(len(storage.get_all_study_summaries()), 1) def test_create_study_duplicated(self) -> None: @@ -59,7 +59,7 @@ class APITestCase(TestCase): content_type="application/json", body=json.dumps(request_body), ) - self.assertEqual(status, "400 Bad Request") + self.assertEqual(status, 400) self.assertEqual(len(storage.get_all_study_summaries()), 1) def test_delete_study(self) -> None: @@ -75,7 +75,7 @@ class APITestCase(TestCase): "DELETE", content_type="application/json", ) - self.assertEqual(status, "204 No Content") + self.assertEqual(status, 204) self.assertEqual(len(storage.get_all_study_summaries()), 1) def test_delete_study_not_found(self) -> None: @@ -87,7 +87,7 @@ class APITestCase(TestCase): "DELETE", content_type="application/json", ) - self.assertEqual(status, "404 Not Found") + self.assertEqual(status, 404) class BottleRequestHookTestCase(TestCase): @@ -104,4 +104,4 @@ class BottleRequestHookTestCase(TestCase): "GET", content_type="application/json", ) - self.assertEqual(status, "200 OK") + self.assertEqual(status, 200) diff --git a/tests/wsgi_utils.py b/tests/wsgi_utils.py index 4a88b9c3..d6975951 100644 --- a/tests/wsgi_utils.py +++ b/tests/wsgi_utils.py @@ -47,7 +47,7 @@ def send_request( queries: Optional[Dict[str, str]] = None, headers: Optional[Dict[str, str]] = None, content_type: str = "text/plain; charset=utf-8", -) -> Tuple[str, List[Tuple[str, str]], bytes]: +) -> Tuple[int, List[Tuple[str, str]], bytes]: status: str = "" response_headers: List[Tuple[str, str]] = [] @@ -65,4 +65,5 @@ def send_request( for b in iterable_body: body += b - return status, response_headers, body + status_code = int(status.split()[0]) + return status_code, response_headers, body