From b3575eb06852b456cde915dfda281a3e778e3aeb Mon Sep 17 00:00:00 2001 From: wakonig_k Date: Wed, 19 Jun 2024 11:42:53 +0200 Subject: [PATCH] test: moved rpc_classes test --- tests/unit_tests/test_generate_cli_client.py | 14 -------------- tests/unit_tests/test_plugin_utils.py | 14 ++++++++++++++ 2 files changed, 14 insertions(+), 14 deletions(-) create mode 100644 tests/unit_tests/test_plugin_utils.py diff --git a/tests/unit_tests/test_generate_cli_client.py b/tests/unit_tests/test_generate_cli_client.py index 9583b287..69d80606 100644 --- a/tests/unit_tests/test_generate_cli_client.py +++ b/tests/unit_tests/test_generate_cli_client.py @@ -97,17 +97,3 @@ def test_client_generator_with_black_formatting(): generated_output_formatted = isort.code(generated_output_formatted) 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 diff --git a/tests/unit_tests/test_plugin_utils.py b/tests/unit_tests/test_plugin_utils.py new file mode 100644 index 00000000..82276f3d --- /dev/null +++ b/tests/unit_tests/test_plugin_utils.py @@ -0,0 +1,14 @@ +from bec_widgets.utils.plugin_utils import get_rpc_classes + + +def test_client_generator_classes(): + out = 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