From ccaf6b0b2d9a7dc612b5129e1c2841a04a2b587c Mon Sep 17 00:00:00 2001 From: Jordan Cook Date: Sat, 21 May 2022 12:54:56 -0500 Subject: Normalize ordering and whitespace for multi-value request headers --- tests/unit/test_cache_keys.py | 17 +++++++++++++---- 1 file changed, 13 insertions(+), 4 deletions(-) (limited to 'tests') diff --git a/tests/unit/test_cache_keys.py b/tests/unit/test_cache_keys.py index 3b774a6..3371d70 100644 --- a/tests/unit/test_cache_keys.py +++ b/tests/unit/test_cache_keys.py @@ -4,7 +4,7 @@ This just contains tests for some extra edge cases not covered elsewhere. import json import pytest -from requests import PreparedRequest, Request +from requests import Request from requests_cache.cache_keys import MAX_NORM_BODY_SIZE, create_key, normalize_request @@ -119,11 +119,20 @@ def test_normalize_request__ovsersized_body(): assert normalize_request(request, ignored_parameters=['param']).body == encoded_body +def test_normalize_request__headers(): + request = Request( + method='GET', + url='https://img.site.com/base/img.jpg', + headers={'Accept': 'gzip, deflate,Venmo, PayPal, '}, + ) + norm_request = normalize_request(request.prepare()) + assert norm_request.headers == {'Accept': 'deflate, gzip, paypal, venmo'} + + def test_remove_ignored_headers__empty(): - request = PreparedRequest() - request.prepare( + request = Request( method='GET', url='https://img.site.com/base/img.jpg', headers={'foo': 'bar'}, ) - assert normalize_request(request, ignored_parameters=None).headers == request.headers + assert normalize_request(request.prepare(), ignored_parameters=None).headers == request.headers -- cgit v1.2.1