feat(cli): auto-discover rpc-enabled widgets

tests/unit_tests/test_generate_cli_client.py

@@ -1,7 +1,6 @@
 from textwrap import dedent
 
 import black
-import pytest
 
 from bec_widgets.cli.generate_cli import ClientGenerator
 
@@ -33,16 +32,31 @@ class MockBECFigure:
 
 def test_client_generator_with_black_formatting():
     generator = ClientGenerator()
-    generator.generate_client([MockBECWaveform1D, MockBECFigure])
+    rpc_classes = {
+        "connector_classes": [MockBECWaveform1D, MockBECFigure],
+        "top_level_classes": [MockBECFigure],
+    }
+    generator.generate_client(rpc_classes)
 
     # Format the expected output with black to ensure it matches the generator output
     expected_output = dedent(
         '''\
         # This file was automatically generated by generate_cli.py
 
-        from bec_widgets.cli.client_utils import rpc_call, RPCBase, BECGuiClientMixin
+        import enum
         from typing import Literal, Optional, overload
 
+        from bec_widgets.cli.client_utils import BECGuiClientMixin, RPCBase, rpc_call
+
+
+        class Widgets(str, enum.Enum):
+            """
+            Enum for the available widgets.
+            """
+
+            MockBECFigure = "MockBECFigure"
+
+
         class MockBECWaveform1D(RPCBase):
             @rpc_call
             def set_frequency(self, frequency: float) -> list:
@@ -79,3 +93,17 @@ def test_client_generator_with_black_formatting():
     )
 
     assert expected_output_formatted == generated_output_formatted
+
+
+def test_client_generator_classes():
+    generator = ClientGenerator()
+    out = generator.get_rpc_classes("bec_widgets")
+    assert list(out.keys()) == ["connector_classes", "top_level_classes"]
+    connector_cls_names = [cls.__name__ for cls in out["connector_classes"]]
+    top_level_cls_names = [cls.__name__ for cls in out["top_level_classes"]]
+
+    assert "BECFigure" in connector_cls_names
+    assert "BECWaveform" in connector_cls_names
+    assert "BECDockArea" in top_level_cls_names
+    assert "BECFigure" in top_level_cls_names
+    assert "BECWaveform" not in top_level_cls_names