Merge pull request #91 from Geode-solutions/feat/before_teardown_request

feat(utils_functions): before & teardown requests function wrappers

Author: JulienChampagnol

src/opengeodeweb_back/utils_functions.py

@@ -33,6 +33,15 @@ def update_last_request_time(current_app):
         current_app.config.update(LAST_REQUEST_TIME=LAST_REQUEST_TIME)
 
 
+def before_request(current_app):
+    increment_request_counter(current_app)
+
+
+def teardown_request(current_app):
+    decrement_request_counter(current_app)
+    update_last_request_time(current_app)
+
+
 def kill_task(current_app):
     DESKTOP_APP = bool(current_app.config.get("DESKTOP_APP"))
     REQUEST_COUNTER = int(current_app.config.get("REQUEST_COUNTER"))

tests/test_utils_functions.py

@@ -20,6 +20,20 @@ def test_update_last_request_time(app_context):
     assert flask.current_app.config.get("LAST_REQUEST_TIME") >= LAST_REQUEST_TIME
 
 
+def test_before_request(app_context):
+    assert flask.current_app.config.get("REQUEST_COUNTER") == 0
+    utils_functions.before_request(flask.current_app)
+    assert flask.current_app.config.get("REQUEST_COUNTER") == 1
+
+
+def test_teardown_request(app_context):
+    LAST_REQUEST_TIME = flask.current_app.config.get("LAST_REQUEST_TIME")
+    assert flask.current_app.config.get("REQUEST_COUNTER") == 1
+    utils_functions.teardown_request(flask.current_app)
+    assert flask.current_app.config.get("REQUEST_COUNTER") == 0
+    assert flask.current_app.config.get("LAST_REQUEST_TIME") >= LAST_REQUEST_TIME
+
+
 def test_versions():
     list_packages = [
         "OpenGeode-core",