adjusting jwt test to adjust value in the MultiProcessing and removing the clearing input values to make sure that we were only triggering callbacks when we desired.

BryanSchroeder · BryanSchroeder · commit 2473546a3d83 · 2024-12-09T14:53:33.000-05:00
diff --git a/tests/integration/renderer/test_request_hooks.py b/tests/integration/renderer/test_request_hooks.py
@@ -1,8 +1,10 @@
+import asyncio
 import json
 import functools
 import flask
 import pytest
 from tests.utils import test_async
+from multiprocessing import Value
 
 from flaky import flaky
 
@@ -201,12 +203,9 @@ def update_output(value):
     assert dash_duo.get_logs() == []
 
 
-@flaky(max_runs=3)
+# @flaky(max_runs=3)
 @pytest.mark.parametrize("expiry_code", [401, 400])
 def test_rdrh003_refresh_jwt(expiry_code, dash_duo):
-    if test_async():
-        return  # if async, bypass this test as this ends up wrapping async funcs and results in 3 failed requests
-
     app = Dash(__name__)
 
     app.index_string = """<!DOCTYPE html>
@@ -248,28 +247,35 @@ def test_rdrh003_refresh_jwt(expiry_code, dash_duo):
         ]
     )
 
-    @app.callback(Output("output-1", "children"), [Input("input", "value")])
+    @app.callback(Output("output-1", "children"), [Input("input", "value")
+                                                   ],prevent_initial_call=True)
     def update_output(value):
+        jwt_token.value = len(value) + 1
         return value
 
-    required_jwt_len = 0
+    jwt_token = Value("i", 0)
 
     # test with an auth layer that requires a JWT with a certain length
     def protect_route(func):
         @functools.wraps(func)
         def wrap(*args, **kwargs):
             try:
+
                 if flask.request.method == "OPTIONS":
                     return func(*args, **kwargs)
                 token = flask.request.headers.environ.get("HTTP_AUTHORIZATION")
-                if required_jwt_len and (
-                    not token or len(token) != required_jwt_len + len("Bearer ")
+                if jwt_token.value and (
+                    not token or len(token) != jwt_token.value + len("Bearer ")
                 ):
                     # Read the data to prevent bug with base http server.
                     flask.request.get_json(silent=True)
                     flask.abort(expiry_code, description="JWT Expired " + str(token))
             except HTTPException as e:
                 return e
+            if asyncio.iscoroutinefunction(func):
+                if test_async():
+                    from asgiref.sync import async_to_sync   # pylint: disable=unused-import, # noqa: F401
+                    return async_to_sync(func)(*args, **kwargs)
             return func(*args, **kwargs)
 
         return wrap
@@ -287,22 +293,21 @@ def wrap(*args, **kwargs):
     _in = dash_duo.find_element("#input")
     dash_duo.clear_input(_in)
 
-    required_jwt_len = 1
-
-    _in.send_keys("fired request")
+    dash_duo.wait_for_text_to_equal("#output-1", "")
 
-    dash_duo.wait_for_text_to_equal("#output-1", "fired request")
+    _in.send_keys(".")
+    dash_duo.wait_for_text_to_equal("#output-1", ".")
     dash_duo.wait_for_text_to_equal("#output-token", ".")
 
-    required_jwt_len = 2
-
-    dash_duo.clear_input(_in)
-    _in.send_keys("fired request again")
-
-    dash_duo.wait_for_text_to_equal("#output-1", "fired request again")
+    _in.send_keys(".")
+    dash_duo.wait_for_text_to_equal("#output-1", "..")
     dash_duo.wait_for_text_to_equal("#output-token", "..")
 
-    assert len(dash_duo.get_logs()) == 2
+    _in.send_keys(".")
+    dash_duo.wait_for_text_to_equal("#output-1", "...")
+    dash_duo.wait_for_text_to_equal("#output-token", "...")
+
+    assert len(dash_duo.get_logs()) == 3
 
 
 def test_rdrh004_layout_hooks(dash_duo):
