Fix @class_component signature

Author: stevej2608

src/reactpy_utils/class_component.py

@@ -1,9 +1,10 @@
 from __future__ import annotations
 
 import inspect
-from typing import TYPE_CHECKING, Any, TypeVar, cast
+from typing import TYPE_CHECKING, Any, TypeVar, cast, Callable
 
 from reactpy.core.component import Component
+from reactpy.core.types import ComponentType
 
 if TYPE_CHECKING:
     from reactpy.core.types import VdomDict
@@ -53,6 +54,13 @@ def create_component(*args: Any, key: Any | None = None, **kwargs: Any):
         _comp = cast(_ComponentClass, _comp)
         _comp._sig = sig  # pylint: disable=protected-access
         _comp.key = key
+
+        _comp._args = args
+        _comp._kwargs = kwargs
+
+        _comp.type = cast(Callable[..., ComponentType],comp)
+
+
         return _comp
 
     return create_component  # type: ignore

tests/test_class_component.py

@@ -1,4 +1,6 @@
 import pytest
+from reactpy import html
+from reactpy.core.types import VdomDict
 from reactpy.testing import DisplayFixture
 
 from docs.examples.python.class_component import App
@@ -7,19 +9,37 @@
 from .tooling import page_stable
 
 
-@class_component
-class RenderMissing:
-    def __init__(self):
-        super().__init__()
+def test_render_missing(display: DisplayFixture):
 
+    @class_component
+    class RenderMissing:
+        """Class must have a render() method"""
+
+        def __init__(self):
+            super().__init__()
 
-def test_render_missing(display: DisplayFixture):
     try:
         component = RenderMissing()
         component.render()  # type: ignore
     except NotImplementedError:
         assert True
 
+def test_str(display: DisplayFixture):
+
+    @class_component
+    class TitleComponent:
+
+        def __init__(self, title:str):
+            super().__init__()
+            self.title = title
+
+        def render(self) -> VdomDict:
+            return html.h2(self.title)
+
+    component = TitleComponent(title="Main Page")
+    assert f"{component}" == f"TitleComponent({id(component):02x}, title='Main Page')"
+
+
 
 @pytest.mark.anyio
 async def test_component_class_new(display: DisplayFixture):