Use dotenv to import worker keys into mitmproxy; add line breaks to d…

…ocstrings in mitmproxy test.
cisagov · Mar 16, 2024 · 9c0523e · 9c0523e
1 parent 02bde87
commit 9c0523e
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/backend/worker/mitmproxy_sign_requests.py b/backend/worker/mitmproxy_sign_requests.py
@@ -4,10 +4,13 @@
 import traceback
 
 # Third-Party Libraries
+from dotenv import load_dotenv
 from mitmproxy import http
 import requests
 from requests_http_signature import HTTPSignatureHeaderAuth
 
+load_dotenv()
+
 
 class SignRequests:
     """

diff --git a/backend/worker/test_mitmproxy_sign_requests.py b/backend/worker/test_mitmproxy_sign_requests.py
@@ -20,7 +20,8 @@ def test_user_agent_and_signature():
     """
     This function tests the SignRequests class with a user agent and signature set.
 
-    It creates an instance of the SignRequests class with a user agent and signature, makes a request, and verifies the signature.
+    It creates an instance of the SignRequests class with a user agent and signature, makes a request, and verifies the
+    signature.
     """
     sr = SignRequests(
         key_id="crossfeed",
@@ -45,7 +46,8 @@ def test_no_user_agent_or_signature_set():
     """
     This function tests the SignRequests class without a user agent and signature set.
 
-    It creates an instance of the SignRequests class without a user agent and signature, makes a request, and checks that no user agent, date, or signature headers are set.
+    It creates an instance of the SignRequests class without a user agent and signature, makes a request, and checks
+    that no user agent, date, or signature headers are set.
     """
     sr = SignRequests(key_id="", public_key="", private_key="", user_agent="")
     with taddons.context():