Upload test_admin_stats.py

test_admin_stats.py

@@ -1,26 +1,20 @@
 import os
 import sys
 
-# Add app to path
 sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__))))
 
 from app.db.base import SessionLocal
-from app.api.admin import get_dashboard_stats
+from app.api.admin import get_dashboard_stats, get_chart_data
 
 def run():
     db = SessionLocal()
-    # Mock user_id 1 (assume admin for test or bypass check_admin temporarily for script or just pass admin user id)
-    # The check_admin function needs a real admin user ID. 
-    # Let's just catch the exception and see what check_admin complains about if user 1 is not admin.
     try:
-        from app.models.user import User, UserRole
-        admin = db.query(User).filter(User.role == UserRole.ADMIN).first()
-        if not admin:
-            print("No admin found in DB.")
-            return
-            
-        res = get_dashboard_stats(admin.id, db)
-        print("Success:", res)
+        res1 = get_dashboard_stats(1, db)
+        print("Success stats:", res1)
+        
+        res2 = get_chart_data(1, db)
+        print("Success charts:", len(res2["value"]), "days of data")
+        
     except Exception as e:
         print("Error:", e)
     finally: