Update tests/repositories/test_legacy_repository.py

Co-authored-by: Arun Babu Neelicattu <[email protected]>
python-poetry · Feb 10, 2021 · 4c734f1 · 4c734f1
1 parent 6c92d42
commit 4c734f1
Showing 1 changed file with 2 additions and 8 deletions.
diff --git a/tests/repositories/test_legacy_repository.py b/tests/repositories/test_legacy_repository.py
@@ -310,14 +310,8 @@ def test_get_200_returns_page(http):
     assert repo._get("/foo")
 
 
-def test_get_404_returns_none(http):
-    repo = MockHttpRepository({"/foo": 404}, http)
-
-    assert repo._get("/foo") is None
-
-
-def test_get_4xx_and_5xx_raises(http):
-    endpoints = {"/{}".format(code): code for code in {401, 403, 500}}
+def test_get_40x_and_returns_none(http):
+    endpoints = {"/{}".format(code): code for code in {401, 403, 404}}
     repo = MockHttpRepository(endpoints, http)
 
     for endpoint in endpoints: