test

app/routes.py

@@ -12,7 +12,7 @@ MOCK_USER_ID = 1
 @main.route('/')
 def index():
     # Ensure the mock user exists
-    user = User.query.get(MOCK_USER_ID)
+    user = db.session.get(User, MOCK_USER_ID)
     if not user:
         user = User(id=MOCK_USER_ID, email='prototype@example.com')
         db.session.add(user)