Fix tests

smartystreets · May 31, 2024 · fb17c04 · fb17c04
1 parent 4ac794e
commit fb17c04
Showing 1 changed file with 8 additions and 10 deletions.
diff --git a/test/custom_header_test.py b/test/custom_header_test.py
@@ -16,11 +16,11 @@ def test_custom_headers_set(self):
         sender = smarty.RequestsSender
         header = {'Test-User-Agent': 'Test-Agent', 'Test-Content-Type': 'Test-Type'}
         sender = smarty.CustomHeaderSender(header, sender)
-        smartyrequest = smarty.Request()
-        smartyrequest.url_prefix = "http://localhost"
-        smartyrequest.payload = "This is the test content."
+        smarty_request = smarty.Request()
+        smarty_request.url_prefix = "http://localhost"
+        smarty_request.payload = "This is the test content."
 
-        request = sender.build_request(smartyrequest)
+        request = sender.build_request(smarty_request)
 
         self.assertEqual('Test-Agent', request.headers['Test-User-Agent'])
         self.assertEqual('Test-Type', request.headers['Test-Content-Type'])
@@ -29,13 +29,11 @@ def test_custom_headers_used(self):
         sender = smarty.RequestsSender()
         header = {'User-Agent': 'Test-Agent', 'Content-Type': 'Test-Type'}
         sender = smarty.CustomHeaderSender(header, sender)
-        smartyrequest = smarty.Request()
-        smartyrequest.url_prefix = "http://localhost"
-        smartyrequest.payload = "This is the test content."
+        smarty_request = smarty.Request()
+        smarty_request.url_prefix = "http://localhost"
+        smarty_request.payload = "This is the test content."
 
-        request = sender.build_request(smartyrequest)
-
-        request = smarty.requests_sender.build_request(request)
+        request = sender.build_request(smarty_request)
 
         self.assertEqual('Test-Agent', request.headers['User-Agent'])
         self.assertEqual('Test-Type', request.headers['Content-Type'])