set_rate_limit_group_test use context generator utils

Author: bitterpanda63

aikido_zen/middleware/set_rate_limit_group_test.py

@@ -1,6 +1,6 @@
 import pytest
-from aikido_zen.context import get_current_context, Context
 from aikido_zen.thread.thread_cache import get_cache
+import aikido_zen.test_utils as test_utils
 from .set_rate_limit_group import set_rate_limit_group
 
 
@@ -15,31 +15,8 @@ def run_around_tests():
     get_cache().reset()
 
 
-def set_context_and_lifecycle():
-    wsgi_request = {
-        "REQUEST_METHOD": "GET",
-        "HTTP_HEADER_1": "header 1 value",
-        "HTTP_HEADER_2": "Header 2 value",
-        "RANDOM_VALUE": "Random value",
-        "HTTP_COOKIE": "sessionId=abc123xyz456;",
-        "wsgi.url_scheme": "http",
-        "HTTP_HOST": "localhost:8080",
-        "PATH_INFO": "/hello",
-        "QUERY_STRING": "user=JohnDoe&age=30&age=35",
-        "CONTENT_TYPE": "application/json",
-        "REMOTE_ADDR": "198.51.100.23",
-    }
-    context = Context(
-        req=wsgi_request,
-        body=None,
-        source="flask",
-    )
-    context.set_as_current_context()
-    return context
-
-
 def test_set_rate_limit_group_valid_group_id(caplog):
-    context1 = set_context_and_lifecycle()
+    context1 = test_utils.generate_and_set_context()
     assert context1.rate_limit_group is None
     set_rate_limit_group("group1")
     assert context1.rate_limit_group == "group1"
@@ -49,15 +26,15 @@ def test_set_rate_limit_group_valid_group_id(caplog):
 
 
 def test_set_rate_limit_group_empty_group_id(caplog):
-    context1 = set_context_and_lifecycle()
+    context1 = test_utils.generate_and_set_context()
     assert context1.rate_limit_group is None
     set_rate_limit_group("")
     assert context1.rate_limit_group is None
     assert "Group ID cannot be empty." in caplog.text
 
 
 def test_set_rate_limit_group_none_group_id(caplog):
-    context1 = set_context_and_lifecycle()
+    context1 = test_utils.generate_and_set_context()
     assert context1.rate_limit_group is None
     set_rate_limit_group(None)
     assert context1.rate_limit_group is None
@@ -73,30 +50,30 @@ def test_set_rate_limit_group_no_context(caplog):
 
 
 def test_set_rate_limit_group_middleware_already_executed(caplog):
-    context1 = set_context_and_lifecycle()
+    context1 = test_utils.generate_and_set_context()
     context1.executed_middleware = True
     set_rate_limit_group("group1")
     assert "must be called before the Zen middleware is executed" in caplog.text
     assert context1.rate_limit_group is "group1"
 
 
 def test_set_rate_limit_group_non_string_group_id(caplog):
-    context1 = set_context_and_lifecycle()
+    context1 = test_utils.generate_and_set_context()
     assert context1.rate_limit_group is None
     set_rate_limit_group(123)
     assert context1.rate_limit_group == "123"
 
 
 def test_set_rate_limit_group_non_string_group_id_non_number(caplog):
-    context1 = set_context_and_lifecycle()
+    context1 = test_utils.generate_and_set_context()
     assert context1.rate_limit_group is None
     set_rate_limit_group({"a": "b"})
     assert context1.rate_limit_group is None
     assert "Group ID must be a string or a number" in caplog.text
 
 
 def test_set_rate_limit_group_overwrite_existing_group():
-    context1 = set_context_and_lifecycle()
+    context1 = test_utils.generate_and_set_context()
     assert context1.rate_limit_group is None
     set_rate_limit_group("group1")
     assert context1.rate_limit_group == "group1"

aikido_zen/test_utils/__init__.py

@@ -31,3 +31,6 @@ def __init__(self):
         self.source = "flask"
         self.route = "/"
         self.parsed_userinput = {}
+        self.user = None
+        self.rate_limit_group = None
+        self.executed_middleware = False