Improve tests

tests/app/endpoints.py

@@ -34,21 +34,21 @@ def get_my_uid(uid: user_id):
 
 
 @ep.response_model(str)
-def get_snake(name) -> Optional[str]:
-    return None
+def get_snake(name=None) -> Optional[str]:
+    return name
 
 
-def get_snake_legacy(name) -> Optional[str]:
-    return None
+def get_snake_legacy(name=None) -> Optional[str]:
+    return name
 
 
 get_snake_legacy.not_found_on_none = True
 
 
 @ep.login_required
 @ep.response_model(str)
-def get_secure_snake(site_id, name) -> Optional[str]:
-    return None
+def get_secure_snake(site_id, name=None) -> Optional[str]:
+    return name
 
 
 @ep.login_required
@@ -74,13 +74,13 @@ def process_request(request, body):
     return {"body": body, "headers": request.headers}
 
 
-class RawRequest(BaseModel):
+class RawResp(BaseModel):
     raw_body: str
     headers: Dict[str, str]
 
 
-@ep.response_model(RawRequest)
-def process_raw_request(request):
+@ep.response_model(RawResp)
+def process_raw_request(request) -> Optional[dict]:
     return {"raw_body": request.stream.read().decode(), "headers": request.headers}
 
 

tests/test_rest.py

@@ -156,15 +156,24 @@ def test_not_found():
 
     headers = {"Authorization": sid}
 
+    url = urls.base + "snakes/"
+    assert requests.get(url).status_code == 404
+
     url = urls.base + "snakes/viper"
+    assert requests.get(url).status_code == 200
+
+    url = urls.base + "snakes-legacy/"
     assert requests.get(url).status_code == 404
 
     url = urls.base + "snakes-legacy/viper"
-    assert requests.get(url).status_code == 404
+    assert requests.get(url).status_code == 200
 
-    url = urls.base + "sites/1/snakes/viper"
+    url = urls.base + "sites/1/snakes/"
     assert requests.get(url, headers=headers).status_code == 404
 
+    url = urls.base + "sites/1/snakes/viper"
+    assert requests.get(url, headers=headers).status_code == 200
+
 
 def test_site_group_access():
     # 1. No group