fix api tests

golemfoundation · May 16, 2024 · 95af7bb · 95af7bb
1 parent 8cf2143
commit 95af7bb
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/backend/app/__init__.py b/backend/app/__init__.py
@@ -31,7 +31,9 @@ def create_app(config=None):
         template_folder=f"{config.PROJECT_ROOT}/templates",
         static_folder=f"{config.PROJECT_ROOT}/static",
     )
-    app.teardown_request(teardown_session)
+
+    app.teardown_request(teardown_session)(app)
+
     app.config.from_object(config)
 
     register_extensions(app)
@@ -60,5 +62,6 @@ def register_errorhandlers(app):
     app.register_error_handler(Exception, handler)
 
 
-def teardown_session(*args, **kwargs):
-    db.session.remove()
+def teardown_session(app: Flask, *args, **kwargs):
+    with app.app_context():
+        db.session.remove()