Set some things more simply

Author: adamtheturtle

tests/mock_vws/test_authorization_header.py

@@ -37,9 +37,7 @@ def test_missing(self, endpoint: TargetAPIEndpoint) -> None:
 
         headers.pop('Authorization', None)
 
-        endpoint.prepared_request.prepare_headers(  # type: ignore
-            headers=headers,
-        )
+        endpoint.prepared_request.headers = headers
         session = requests.Session()
         response = session.send(  # type: ignore
             request=endpoint.prepared_request,
@@ -64,9 +62,7 @@ def test_incorrect(self, endpoint: TargetAPIEndpoint) -> None:
             'Date': date,
         }
 
-        endpoint.prepared_request.prepare_headers(  # type: ignore
-            headers=headers,
-        )
+        endpoint.prepared_request.headers = headers
         session = requests.Session()
         response = session.send(  # type: ignore
             request=endpoint.prepared_request,

tests/mock_vws/test_date_header.py

@@ -52,10 +52,7 @@ def test_no_date_header(
             'Authorization': authorization_string,
         }
         headers.pop('Date', None)
-
-        endpoint.prepared_request.prepare_headers(  # type: ignore
-            headers=headers,
-        )
+        endpoint.prepared_request.headers = headers
         session = requests.Session()
         response = session.send(  # type: ignore
             request=endpoint.prepared_request,
@@ -107,9 +104,7 @@ def test_incorrect_date_format(
             'Date': date_incorrect_format,
         }
 
-        endpoint.prepared_request.prepare_headers(  # type: ignore
-            headers=headers,
-        )
+        endpoint.prepared_request.headers = headers
         session = requests.Session()
         response = session.send(  # type: ignore
             request=endpoint.prepared_request,
@@ -171,9 +166,7 @@ def test_date_out_of_range(
             'Date': date,
         }
 
-        endpoint.prepared_request.prepare_headers(  # type: ignore
-            headers=headers,
-        )
+        endpoint.prepared_request.headers = headers
         session = requests.Session()
         response = session.send(  # type: ignore
             request=endpoint.prepared_request,
@@ -227,9 +220,7 @@ def test_date_in_range(
             'Date': date,
         }
 
-        endpoint.prepared_request.prepare_headers(  # type: ignore
-            headers=headers,
-        )
+        endpoint.prepared_request.headers = headers
         session = requests.Session()
         response = session.send(  # type: ignore
             request=endpoint.prepared_request,

tests/mock_vws/test_invalid_json.py

@@ -64,14 +64,8 @@ def test_invalid_json(
             'Date': date,
         }
 
-        endpoint.prepared_request.prepare_body(  # type: ignore
-            data=content,
-            files=None,
-        )
-
-        endpoint.prepared_request.prepare_headers(  # type: ignore
-            headers=headers,
-        )
+        endpoint.prepared_request.body = content
+        endpoint.prepared_request.headers = headers
         endpoint.prepared_request.prepare_content_length(  # type: ignore
             body=content,
         )

tests/mock_vws/test_unexpected_json.py

@@ -56,13 +56,8 @@ def test_does_not_take_data(
             'Content-Type': content_type,
         }
 
-        endpoint.prepared_request.prepare_body(  # type: ignore
-            data=content,
-            files=None,
-        )
-        endpoint.prepared_request.prepare_headers(  # type: ignore
-            headers=headers,
-        )
+        endpoint.prepared_request.body = content
+        endpoint.prepared_request.headers = headers
         endpoint.prepared_request.prepare_content_length(  # type: ignore
             body=content,
         )