mockstack 0.11.0__tar.gz → 0.11.1__tar.gz

{mockstack-0.11.0 → mockstack-0.11.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mockstack
-Version: 0.11.0
+Version: 0.11.1
 Summary: An API mocking workhorse
 Author-email: Adam Ever-Hadani <mockstack.contact@gmail.com>
 License-Expression: MIT

{mockstack-0.11.0 → mockstack-0.11.1}/mockstack/strategies/proxyrules.py

@@ -25,7 +25,7 @@ from mockstack.templating import templates_env_provider
 def maybe_update_response_headers(
     response_headers: ResponseHeaders,
     *,
-    request_headers: Headers,
+    content_length: int,
 ) -> ResponseHeaders:
     """Update the response headers if needed, e.g. to adjust for compression etc."""
     _headers = response_headers.copy()
@@ -34,6 +34,8 @@ def maybe_update_response_headers(
         # If the response is compressed, we need to remove the content-encoding header
         # since httpx will automatically decompress the response while proxying.
         _headers["content-encoding"] = "identity"
+        # content length needs to be adjusted as well for uncompressed content.
+        _headers["content-length"] = str(content_length)
 
     return _headers
 
@@ -209,9 +211,10 @@ class ProxyRulesStrategy(BaseStrategy, CreateMixin):
 
             resp = await client.send(req, stream=False)
             content = resp.read()
+
             response_headers = maybe_update_response_headers(
                 resp.headers,
-                request_headers=request.headers,
+                content_length=len(content),
             )
 
             return Response(

{mockstack-0.11.0 → mockstack-0.11.1}/mockstack/tests/strategies/test_proxyrules.py

@@ -359,15 +359,18 @@ async def test_proxy_rules_strategy_reverse_proxy(settings_reverse_proxy, span):
 
         mock_client.send.assert_called_once()
 
-    def test_maybe_update_response_headers_updates_content_encoding():
-        """Test maybe_update_response_headers updates content-encoding."""
-        headers = Headers(
-            {"content-encoding": "gzip", "content-type": "application/json"}
-        )
-        request_headers = Headers({"accept-encoding": "gzip"})
 
-        updated_headers = maybe_update_response_headers(headers, request_headers)
+def test_maybe_update_response_headers_updates_content_encoding():
+    """Test maybe_update_response_headers updates content-encoding."""
+    response_headers = httpx.Headers(
+        {"content-encoding": "gzip", "content-type": "application/json"}
+    )
+
+    updated_headers = maybe_update_response_headers(
+        response_headers=response_headers,
+        content_length=100,
+    )
 
-        assert updated_headers["content-encoding"] == "identity"
-        assert updated_headers["content-type"] == "application/json"
-        assert updated_headers["accept-encoding"] == "gzip"
+    assert updated_headers["content-encoding"] == "identity"
+    assert updated_headers["content-type"] == "application/json"
+    assert updated_headers["content-length"] == "100"

{mockstack-0.11.0 → mockstack-0.11.1}/mockstack.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mockstack
-Version: 0.11.0
+Version: 0.11.1
 Summary: An API mocking workhorse
 Author-email: Adam Ever-Hadani <mockstack.contact@gmail.com>
 License-Expression: MIT