mirror of
https://github.com/bec-project/bec_widgets.git
synced 2025-07-14 03:31:50 +02:00
29 lines
894 B
Python
29 lines
894 B
Python
import pytest
|
|
|
|
from bec_widgets.cli.rpc.rpc_base import DeletedWidgetError, RPCBase, RPCReference
|
|
|
|
|
|
@pytest.fixture
|
|
def rpc_base():
|
|
yield RPCBase(gui_id="rpc_base_test", object_name="test")
|
|
|
|
|
|
def test_rpc_base(rpc_base):
|
|
"""Test registry and reference creation"""
|
|
registry = {rpc_base._gui_id: rpc_base}
|
|
ref = RPCReference(registry, rpc_base._gui_id)
|
|
|
|
assert ref._gui_id == rpc_base._gui_id
|
|
assert ref.object_name == rpc_base.object_name
|
|
assert ref.__str__() == rpc_base.__str__()
|
|
assert ref.__repr__() == rpc_base.__repr__()
|
|
|
|
# Remove object from registry
|
|
registry.pop(rpc_base._gui_id)
|
|
|
|
assert ref.__str__() == f"<Deleted widget with gui_id {rpc_base._gui_id}>"
|
|
assert ref.__repr__() == f"<Deleted widget with gui_id {rpc_base._gui_id}>"
|
|
|
|
with pytest.raises(DeletedWidgetError):
|
|
ref._root # Object no longer referenced in registry
|