adds missing tests

app/tests/web/endpoints/test_default.py

@@ -1,6 +1,8 @@
 from unittest.mock import MagicMock
 from fastapi.testclient import TestClient
 import pytest
+from app.shared.schemas import Usage, UsageResponse
+from app.shared.user_groups import GroupInfo
 from app.web.config import VERSION
 from app.tests.web.db.test_crud import test_data
 
@@ -13,6 +15,7 @@ def test_endpoint_home(client_with_auth):
     assert "breakingChanges" in j
     assert "groups" not in j
 
+
 def test_endpoint_health(client_with_auth):
     r = client_with_auth.get("/health")
     assert r.status_code == 200
@@ -28,7 +31,7 @@ def test_endpoint_active(app):
 
     from app.web.security import get_user_state
     app.dependency_overrides[get_user_state] = lambda: m_user_state
-    
+
     # inactive user
     m_user_state.active = False
     client = TestClient(app)
@@ -42,7 +45,6 @@ def test_endpoint_active(app):
     r = client.get("/user/active")
     assert r.status_code == 200
     assert r.json() == {"active": True}
-    
 
 
 def test_no_serve_local_archive_by_default(client_with_auth):
@@ -100,3 +102,74 @@ async def test_prometheus_metrics(test_data, client_with_token, get_settings):
     assert 'database_metrics_counter_total{query="count_by_user",user="rick@example.com"} 34.0' in r3.text
     assert 'database_metrics_counter_total{query="count_by_user",user="morty@example.com"} 33.0' in r3.text
     assert 'database_metrics_counter_total{query="count_by_user",user="jerry@example.com"} 33.0' in r3.text
+
+
+def test_endpoint_get_user_permissions_no_user_auth(client, test_no_auth):
+    test_no_auth(client.get, "/user/permissions")
+
+
+def test_endpoint_get_user_permissions(app):
+    from app.web.security import get_user_state
+
+    m_user_state = MagicMock()
+    rv = {
+        "all": GroupInfo(read=True),
+        "group1": GroupInfo(archive_url=True),
+    }
+    from loguru import logger
+    logger.info(rv)
+    m_user_state.permissions = rv
+
+    app.dependency_overrides[get_user_state] = lambda: m_user_state
+
+    client = TestClient(app)
+    r = client.get("/user/permissions")
+    assert r.status_code == 200
+    response = r.json()
+    assert response.keys() == {"all", "group1"}
+    assert response["all"]["read"]
+    assert response["group1"]["read"] == []
+    assert response["group1"]["archive_url"]
+    assert response["all"]["archive_url"] == False
+
+
+def test_endpoint_get_user_usage_no_user_auth(client, test_no_auth):
+    test_no_auth(client.get, "/user/usage")
+
+
+def test_endpoint_get_user_usage_inactive(app):
+    from app.web.security import get_user_state
+
+    m_user_state = MagicMock()
+    m_user_state.active = False
+
+    app.dependency_overrides[get_user_state] = lambda: m_user_state
+
+    client = TestClient(app)
+    r = client.get("/user/usage")
+    assert r.status_code == 403
+    assert r.json() == {"detail": "User is not active."}
+
+
+def test_endpoint_get_user_usage_active(app):
+    from app.web.security import get_user_state
+
+    m_user_state = MagicMock()
+    m_user_state.active = True
+    mock_usage = UsageResponse(
+        monthly_urls=1,
+        monthly_mbs=2,
+        total_sheets=3,
+        groups={
+            "group1": Usage(monthly_urls=4, monthly_mbs=5, total_sheets=6),
+            "group2": Usage(monthly_urls=7, monthly_mbs=8, total_sheets=9)
+        }
+    )
+    m_user_state.usage.return_value = mock_usage
+
+    app.dependency_overrides[get_user_state] = lambda: m_user_state
+
+    client = TestClient(app)
+    r = client.get("/user/usage")
+    assert r.status_code == 200
+    assert UsageResponse(**r.json()) == mock_usage