Add HTTP deserializer tests

Author: JordonPhillips

packages/smithy-http/tests/unit/test_serializers.py

@@ -1,6 +1,7 @@
 #  Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
 #  SPDX-License-Identifier: Apache-2.0
 import datetime
+from asyncio import iscoroutinefunction
 from decimal import Decimal
 from dataclasses import dataclass, field
 from typing import ClassVar, Self, Any, Protocol
@@ -40,8 +41,9 @@
     TIMESTAMP,
 )
 from smithy_core.aio.types import AsyncBytesReader
+from smithy_http.deserializers import HTTPResponseDeserializer
 from smithy_json import JSONCodec
-
+from smithy_http.aio import HTTPResponse as _HTTPResponse
 from smithy_http import tuples_to_fields, Fields
 from smithy_http.serializers import HTTPRequestSerializer, HTTPResponseSerializer
 
@@ -388,6 +390,7 @@ def _consumer(schema: Schema, de: ShapeDeserializer) -> None:
                         cls.SCHEMA.members["boolean_list_member"],
                         lambda d: list_value.append(d.read_boolean(BOOLEAN)),
                     )
+                    kwargs["boolean_list_member"] = list_value
                 case 2:
                     kwargs["integer_member"] = de.read_integer(
                         cls.SCHEMA.members["integer_member"]
@@ -564,7 +567,7 @@ class HTTPImplicitPayload:
                 "target": STRING,
                 "traits": [HTTPHeaderTrait("header")],
             },
-            "payload_member": {"index": 0, "target": STRING},
+            "payload_member": {"index": 1, "target": STRING},
         },
     )
 
@@ -587,7 +590,11 @@ def deserialize(cls, deserializer: ShapeDeserializer) -> Self:
         def _consumer(schema: Schema, de: ShapeDeserializer) -> None:
             match schema.expect_member_index():
                 case 0:
-                    kwargs["payload"] = de.read_string(cls.SCHEMA.members["payload"])
+                    kwargs["header"] = de.read_string(cls.SCHEMA.members["header"])
+                case 1:
+                    kwargs["payload_member"] = de.read_string(
+                        cls.SCHEMA.members["payload_member"]
+                    )
                 case _:
                     raise Exception(f"Unexpected schema: {schema}")
 
@@ -1281,6 +1288,21 @@ class HTTPMessageTestCase:
 ]
 
 
+EMPTY_PREFIX_HEADER_DESER_CASES: list[HTTPMessageTestCase] = [
+    HTTPMessageTestCase(
+        HTTPEmptyPrefixHeaders(
+            string_member="string",
+            string_map_member={"foo": "bar", "baz": "bam", "string": "string"},
+        ),
+        HTTPMessage(
+            fields=tuples_to_fields(
+                [("foo", "bar"), ("baz", "bam"), ("string", "string")]
+            ),
+        ),
+    ),
+]
+
+
 QUERY_CASES: list[HTTPMessageTestCase] = [
     HTTPMessageTestCase(
         HTTPQuery(boolean_member=True),
@@ -1529,6 +1551,9 @@ class HTTPMessageTestCase:
         HTTPStructuredPayload(payload=HTTPStringPayload(payload="foo")),
         HTTPMessage(body=BytesIO(b'{"payload":"foo"}')),
     ),
+]
+
+ASYNC_STREAMING_PAYLOAD_CASES: list[HTTPMessageTestCase] = [
     HTTPMessageTestCase(
         HTTPStreamingPayload(payload=AsyncBytesReader(b"\xde\xad\xbe\xef")),
         HTTPMessage(body=AsyncBytesReader(b"\xde\xad\xbe\xef")),
@@ -1544,8 +1569,9 @@ class HTTPMessageTestCase:
 REQUEST_CASES.extend(PAYLOAD_CASES)
 
 REQUEST_SER_CASES: list[HTTPMessageTestCase] = []
-REQUEST_CASES.extend(REQUEST_CASES)
-REQUEST_CASES.extend(EMPTY_PREFIX_HEADER_SER_CASES)
+REQUEST_SER_CASES.extend(REQUEST_CASES)
+REQUEST_SER_CASES.extend(EMPTY_PREFIX_HEADER_SER_CASES)
+REQUEST_SER_CASES.extend(ASYNC_STREAMING_PAYLOAD_CASES)
 
 
 @pytest.mark.parametrize("case", REQUEST_SER_CASES)
@@ -1580,7 +1606,7 @@ async def test_serialize_http_request(case: HTTPMessageTestCase) -> None:
 RESPONSE_CASES.extend(PAYLOAD_CASES)
 
 RESPONSE_SER_CASES: list[HTTPMessageTestCase] = []
-RESPONSE_CASES.extend(RESPONSE_CASES)
+RESPONSE_SER_CASES.extend(RESPONSE_CASES)
 RESPONSE_SER_CASES.extend(EMPTY_PREFIX_HEADER_SER_CASES)
 
 
@@ -1602,3 +1628,40 @@ async def test_serialize_http_response(case: HTTPMessageTestCase) -> None:
         expected_body_value = await AsyncBytesReader(case.request.body).read()
         assert actual_body_value == expected_body_value
         assert type(actual.body) is type(case.request.body)
+
+
+RESPONSE_DESER_CASES: list[HTTPMessageTestCase] = []
+RESPONSE_DESER_CASES.extend(RESPONSE_CASES)
+RESPONSE_DESER_CASES.extend(EMPTY_PREFIX_HEADER_DESER_CASES)
+
+
+# TODO: Move this to a separate file
+@pytest.mark.parametrize("case", RESPONSE_DESER_CASES)
+async def test_deserialize_http_response(case: HTTPMessageTestCase) -> None:
+    body = case.request.body
+    if (read := getattr(body, "read", None)) is not None and iscoroutinefunction(read):
+        body = BytesIO(await read())
+    deserializer = HTTPResponseDeserializer(
+        payload_codec=JSONCodec(),
+        http_trait=case.http_trait,
+        response=_HTTPResponse(
+            body=case.request.body,
+            status=case.request.status,
+            fields=case.request.fields,
+        ),
+        body=body,
+    )
+    actual = type(case.shape).deserialize(deserializer)
+    assert actual == case.shape
+
+
+async def test_deserialize_http_response_with_async_stream() -> None:
+    stream = AsyncBytesReader(b"\xde\xad\xbe\xef")
+
+    deserializer = HTTPResponseDeserializer(
+        payload_codec=JSONCodec(),
+        http_trait=HTTPTrait({"method": "POST", "code": 200, "uri": "/"}),
+        response=_HTTPResponse(body=stream, status=200, fields=Fields()),
+    )
+    actual = HTTPStreamingPayload.deserialize(deserializer)
+    assert actual == HTTPStreamingPayload(stream)