Adding tests for BusPath.parameter_schema and BusPath.response_schema

Author: adamcharnock

lightbus/client.py

@@ -407,11 +407,22 @@ async def lazy_load_now(self):
             )
         )
 
-        # 2. Open the transports
+        # 2. Add any local APIs to the schema
+        for api in self.api_registry.all():
+            await self.schema.add_api(api)
+
+        logger.info(
+            LBullets(
+                "Loaded the following local schemas ({})".format(len(self.schema.remote_schemas)),
+                items=self.schema.local_schemas.keys(),
+            )
+        )
+
+        # 3. Open the transports
         for transport in self.transport_registry.get_all_transports():
             await transport.open()
 
-        # 3. Done
+        # 4. Done
         self._lazy_load_complete = True
 
     # RPCs

lightbus/path.py

@@ -168,8 +168,7 @@ def response_schema(self):
         Only RPCs have responses. Accessing this property for an event will result in a
         SchemaNotFound error.
         """
-        rpc_schema = self.client.schema.get_rpc_schema(self.api_name, self.name)["response"]
-        return rpc_schema["response"]
+        return self.client.schema.get_rpc_schema(self.api_name, self.name)["response"]
 
     def validate_parameters(self, parameters: dict):
         self.client.schema.validate_parameters(self.api_name, self.name, parameters)

tests/test_bus_path_unit.py

@@ -56,3 +56,33 @@ async def test_positional_only_rpc(dummy_bus: lightbus.path.BusPath):
 async def test_positional_only_event(dummy_bus: lightbus.path.BusPath):
     with pytest.raises(InvalidParameters):
         await dummy_bus.my.dummy.event.fire_async(123)
+
+
+@pytest.mark.asyncio
+async def test_parameter_schema_event(dummy_bus: lightbus.path.BusPath, dummy_api):
+    dummy_bus.client.register_api(dummy_api)
+    await dummy_bus.client.lazy_load_now()
+    schema = dummy_bus.my.dummy.my_event.parameter_schema
+    assert schema
+    assert isinstance(schema, dict)
+    assert "$schema" in schema
+
+
+@pytest.mark.asyncio
+async def test_parameter_schema_rpc(dummy_bus: lightbus.path.BusPath, dummy_api):
+    dummy_bus.client.register_api(dummy_api)
+    await dummy_bus.client.lazy_load_now()
+    schema = dummy_bus.my.dummy.my_proc.parameter_schema
+    assert schema
+    assert isinstance(schema, dict)
+    assert "$schema" in schema
+
+
+@pytest.mark.asyncio
+async def test_response_schema_rpc(dummy_bus: lightbus.path.BusPath, dummy_api):
+    dummy_bus.client.register_api(dummy_api)
+    await dummy_bus.client.lazy_load_now()
+    schema = dummy_bus.my.dummy.my_proc.response_schema
+    assert schema
+    assert isinstance(schema, dict)
+    assert "$schema" in schema