Move content type to target API

Author: adamtheturtle

tests/mock_vws/test_date_header.py

@@ -36,8 +36,6 @@ def test_no_date_header(
         A `BAD_REQUEST` response is returned when no `Date` header is given.
         """
         endpoint_headers = dict(endpoint.prepared_request.headers)
-        content_type = endpoint_headers.get('Content-Type', '')
-        assert isinstance(content_type, str)
         content = endpoint.prepared_request.body or b''
         assert isinstance(content, bytes)
 
@@ -46,7 +44,7 @@ def test_no_date_header(
             secret_key=vuforia_database_keys.server_secret_key,
             method=str(endpoint.prepared_request.method),
             content=content,
-            content_type=content_type,
+            content_type=endpoint.auth_header_content_type,
             date='',
             request_path=endpoint.prepared_request.path_url,
         )
@@ -93,8 +91,6 @@ def test_incorrect_date_format(
             date_incorrect_format = now.strftime('%a %b %d %H:%M:%S %Y')
 
         endpoint_headers = dict(endpoint.prepared_request.headers)
-        content_type = endpoint_headers.get('Content-Type', '')
-        assert isinstance(content_type, str)
         content = endpoint.prepared_request.body or b''
         assert isinstance(content, bytes)
 
@@ -103,7 +99,7 @@ def test_incorrect_date_format(
             secret_key=vuforia_database_keys.server_secret_key,
             method=str(endpoint.prepared_request.method),
             content=content,
-            content_type=content_type,
+            content_type=endpoint.auth_header_content_type,
             date=date_incorrect_format,
             request_path=endpoint.prepared_request.path_url,
         )
@@ -160,8 +156,6 @@ def test_date_out_of_range(
             date = rfc_1123_date()
 
         endpoint_headers = dict(endpoint.prepared_request.headers)
-        content_type = endpoint_headers.get('Content-Type', '')
-        assert isinstance(content_type, str)
         content = endpoint.prepared_request.body or b''
         assert isinstance(content, bytes)
 
@@ -170,7 +164,7 @@ def test_date_out_of_range(
             secret_key=vuforia_database_keys.server_secret_key,
             method=str(endpoint.prepared_request.method),
             content=content,
-            content_type=content_type,
+            content_type=endpoint.auth_header_content_type,
             date=date,
             request_path=endpoint.prepared_request.path_url,
         )
@@ -219,8 +213,6 @@ def test_date_in_range(
             date = rfc_1123_date()
 
         endpoint_headers = dict(endpoint.prepared_request.headers)
-        content_type = endpoint_headers.get('Content-Type', '')
-        assert isinstance(content_type, str)
         content = endpoint.prepared_request.body or b''
         assert isinstance(content, bytes)
 
@@ -229,7 +221,7 @@ def test_date_in_range(
             secret_key=vuforia_database_keys.server_secret_key,
             method=str(endpoint.prepared_request.method),
             content=content,
-            content_type=content_type,
+            content_type=endpoint.auth_header_content_type,
             date=date,
             request_path=endpoint.prepared_request.path_url,
         )

tests/mock_vws/test_invalid_json.py

@@ -45,17 +45,15 @@ def test_invalid_json(
             date = rfc_1123_date()
 
         endpoint_headers = dict(endpoint.prepared_request.headers)
-        content_type = endpoint_headers.get('Content-Type', '')
-        assert isinstance(content_type, str)
-        takes_data = bool(content_type)
+        takes_data = bool(endpoint.auth_header_content_type)
         endpoint_headers = dict(endpoint.prepared_request.headers)
 
         authorization_string = authorization_header(
             access_key=vuforia_database_keys.server_access_key,
             secret_key=vuforia_database_keys.server_secret_key,
             method=str(endpoint.prepared_request.method),
             content=content,
-            content_type=content_type,
+            content_type=endpoint.auth_header_content_type,
             date=date,
             request_path=endpoint.prepared_request.path_url,
         )

tests/mock_vws/test_unexpected_json.py

@@ -40,7 +40,6 @@ def test_does_not_take_data(
         date = rfc_1123_date()
 
         endpoint_headers = dict(endpoint.prepared_request.headers)
-        assert isinstance(content_type, str)
 
         authorization_string = authorization_header(
             access_key=vuforia_database_keys.server_access_key,

tests/mock_vws/utils.py

@@ -94,10 +94,16 @@ def __init__(
                 example path is requested with the method.
             successful_headers_status_code: The expected status code if the
                 example path is requested with the method.
+            auth_header_content_type: The content type to use for the
+                `Authorization` header.
         """
         self.prepared_request = prepared_request
         self.successful_headers_status_code = successful_headers_status_code
         self.successful_headers_result_code = successful_headers_result_code
+        headers = prepared_request.headers
+        content_type = headers.get('Content-Type', '')
+        assert isinstance(content_type, str)
+        self.auth_header_content_type: str = content_type
 
 
 def assert_vws_failure(