djcheckup 0.2.1__tar.gz → 0.2.2__tar.gz

{djcheckup-0.2.1 → djcheckup-0.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: djcheckup
-Version: 0.2.1
+Version: 0.2.2
 Summary: DJ Checkup is a security scanner for Django sites.
 Author: Stuart Maxwell
 Author-email: Stuart Maxwell <stuart@amanzi.nz>

{djcheckup-0.2.1 → djcheckup-0.2.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "djcheckup"
-version = "0.2.1"
+version = "0.2.2"
 description = "DJ Checkup is a security scanner for Django sites."
 readme = "README.md"
 authors = [{ name = "Stuart Maxwell", email = "stuart@amanzi.nz" }]
@@ -26,5 +26,15 @@ target-version = "py313"
 [tool.ruff.lint]
 select = ["ALL"]
 
+[tool.ruff.lint.extend-per-file-ignores]
+"tests/**" = ["S101", "ANN201", "ANN001", "ARG001"]
+
 [tool.ruff.lint.pydocstyle]
 convention = "google" # Accepts: "google", "numpy", or "pep257".
+
+[dependency-groups]
+dev = ["pytest>=8.4.2"]
+
+[tool.pytest.ini_options]
+pythonpath = "."
+python_files = "test_*.py"

{djcheckup-0.2.1 → djcheckup-0.2.2}/src/djcheckup/checks.py

@@ -286,6 +286,27 @@ class SchemeCheck(_BaseCheck):
         return response.url.scheme == self.end_scheme
 
 
+def create_context(url: httpx.URL, client: httpx.Client, response: httpx.Response) -> SiteCheckContext:
+    """Create a SiteCheckContext context object.
+
+    Args:
+        url: The URL to check.
+        client: The HTTPX client to use for requests.
+        response: The HTTPX response object from the initial request.
+
+    Returns:
+        A SiteCheckContext object containing the provided parameters.
+    """
+    return SiteCheckContext(
+        url=url,
+        client=client,
+        headers=response.headers,
+        cookies=response.cookies.jar,
+        content=response.text,
+        response_url=response.url,
+    )
+
+
 class SiteChecker:
     """Run all the checks for a given site."""
 
@@ -359,14 +380,7 @@ class SiteChecker:
                 message=error_message,
             )
 
-        self.context = SiteCheckContext(
-            url=self.url,
-            client=self.client,
-            headers=response.headers,
-            cookies=response.cookies.jar,
-            content=response.text,
-            response_url=response.url,
-        )
+        self.context = create_context(self.url, self.client, response)
 
         return CheckResponse(
             name=check_name,