import enum
from importlib import reload
from types import SimpleNamespace
from unittest.mock import MagicMock, call, patch

from bec_widgets.cli import client
from bec_widgets.cli.rpc.rpc_base import RPCBase
from bec_widgets.cli.rpc.rpc_widget_handler import RPCWidgetHandler
from bec_widgets.utils.bec_widget import BECWidget
from bec_widgets.utils.plugin_utils import BECClassContainer, BECClassInfo
from bec_widgets.widgets.containers.dock.dock import BECDock


def test_rpc_widget_handler():
    handler = RPCWidgetHandler()
    assert "Image" in handler.widget_classes
    assert "RingProgressBar" in handler.widget_classes


class _TestPluginWidget(BECWidget): ...


@patch(
    "bec_widgets.cli.rpc.rpc_widget_handler.get_all_plugin_widgets",
    return_value=BECClassContainer(
        [
            BECClassInfo(name="DeviceComboBox", obj=_TestPluginWidget, module="", file=""),
            BECClassInfo(name="NewPluginWidget", obj=_TestPluginWidget, module="", file=""),
        ]
    ),
)
def test_duplicate_plugins_not_allowed(_):
    handler = RPCWidgetHandler()
    assert handler.widget_classes["DeviceComboBox"] is not _TestPluginWidget
    assert handler.widget_classes["NewPluginWidget"] is _TestPluginWidget