first steps of testing query params with http message handler

Author: mrtedn21

tests/test_http_message_handler.py

@@ -2,10 +2,19 @@
 
 import pytest
 
+from typing import Iterable
 from martin_eden.core import HttpMessageHandler
+from martin_eden.database import Base, MarshmallowToDataclass, SqlAlchemyToMarshmallow
 from martin_eden.http_utils import HttpHeadersParser
 from martin_eden.routing import register_route
 from tests.conftest import base_http_request, base_http_result_headers
+from datetime import date
+from typing import TYPE_CHECKING
+
+from sqlalchemy import ForeignKey
+from sqlalchemy.orm import Mapped, mapped_column, relationship
+
+from martin_eden.database import Base
 
 pytest_plugins = ('pytest_asyncio',)
 
@@ -15,6 +24,31 @@ async def get_openapi_schema() -> str:
     return 'test'
 
 
+class TestModel(Base):
+    __tablename__ = 'test'
+    pk: Mapped[int] = mapped_column(primary_key=True)
+    name: Mapped[str]
+
+
+class TestSchema(TestModel, metaclass=SqlAlchemyToMarshmallow):
+    pass
+
+
+class TestDataclass(TestSchema, metaclass=MarshmallowToDataclass):
+    pass
+
+
+@register_route(
+    '/test_query/', 'get',
+    response_schema=TestSchema(),
+    query_params={TestModel: ['name']},
+)
+async def get_users(query_params: Iterable) -> str:
+    print()
+    print(query_params)
+    return query_params
+
+
 @pytest.fixture
 def http_get_request():
     return base_http_request.encode('utf8')
@@ -88,3 +122,15 @@ async def test_openapi_schema(http_get_request):
     assert openapi_result['paths'] == {
         '/test/': {'get': {'operationId': 'test_get'}}
     }
+
+
+#@pytest.mark.asyncio
+#async def test_query_params(http_get_request):
+#    http_get_request = http_get_request.replace(
+#        b'/users/', b'/test_query/?test_model__name__like=martin',
+#    )
+#
+#    handler = HttpMessageHandler(http_get_request)
+#    response = await handler.handle_request()
+#
+#    print(response)