fix getting composite metadata on router via annotation

Author: jell-o-fishi

rsocket/routing/request_router.py

@@ -174,8 +174,8 @@ def _collect_route_arguments(self,
         for parameter in route_signature.parameters:
             parameter_type = route_signature.parameters[parameter]
 
-            if 'composite_metadata' == parameter or parameter_type is CompositeMetadata:
-                route_kwargs['composite_metadata'] = composite_metadata
+            if 'composite_metadata' == parameter or parameter_type.annotation is CompositeMetadata:
+                route_kwargs[parameter] = composite_metadata
             else:
                 payload_data = payload
 

tests/rsocket/test_routing.py

@@ -141,6 +141,25 @@ async def response(composite_metadata: CompositeMetadata):
         assert result.metadata == b'test.path'
 
 
+async def test_routed_request_response_composite_according_to_type(lazy_pipe):
+    router = RequestRouter()
+
+    def handler_factory():
+        return RoutingRequestHandler(router)
+
+    @router.response('test.path')
+    async def response(metadata: CompositeMetadata):
+        return create_response(metadata=metadata.items[0].tags[0])
+
+    async with lazy_pipe(
+            client_arguments={'metadata_encoding': WellKnownMimeTypes.MESSAGE_RSOCKET_COMPOSITE_METADATA},
+            server_arguments={'handler_factory': handler_factory}) as (server, client):
+        result = await client.request_response(Payload(metadata=composite(route('test.path'))))
+
+        assert result.metadata == b'test.path'
+
+
+
 async def test_routed_request_response_properly_finished_accept_payload_and_metadata(lazy_pipe):
     router = RequestRouter()