Test director methods.

pymeasure · May 16, 2024 · ba40751 · ba40751
1 parent c8833a7
commit ba40751
Showing 1 changed file with 26 additions and 0 deletions.
diff --git a/tests/directors/test_director.py b/tests/directors/test_director.py
@@ -77,6 +77,20 @@ def test_default_actor(self, director: Director):
         assert director._actor_check("") == "actor"
 
 
+def test_get_rpc_capabilities(director: Director):
+    data = {"name": "actor", "methods": []}
+    director.communicator._r = [  # type: ignore
+        Message("director", "actor", conversation_id=cid, message_type=MessageTypes.JSON, data={
+            "id": 1, "result": data, "jsonrpc": "2.0"
+            })]
+    result = director.get_rpc_capabilities()
+    assert director.communicator._s == [  # type: ignore
+        Message("actor", "director", conversation_id=cid, message_type=MessageTypes.JSON, data={
+            "id": 1, "method": "rpc.discover", "jsonrpc": "2.0"
+            })]
+    assert result == data
+
+
 def test_shutdown_actor(director: Director):
     director.communicator._r = [  # type: ignore
         Message("director", "actor", conversation_id=cid, message_type=MessageTypes.JSON, data={
@@ -89,6 +103,18 @@ def test_shutdown_actor(director: Director):
             })]
 
 
+def test_set_actor_log_level(director: Director):
+    director.communicator._r = [  # type: ignore
+        Message("director", "actor", conversation_id=cid, message_type=MessageTypes.JSON, data={
+            "id": 1, "result": None, "jsonrpc": "2.0"
+            })]
+    director.set_actor_log_level(30)
+    assert director.communicator._s == [  # type: ignore
+        Message("actor", "director", conversation_id=cid, message_type=MessageTypes.JSON, data={
+            "id": 1, "method": "set_log_level", "jsonrpc": "2.0", "params": {"level": "WARNING"}
+            })]
+
+
 def test_read_rpc_response(director: Director):
     director.communicator._r = [  # type: ignore
         Message("director", "actor", conversation_id=cid, message_type=MessageTypes.JSON, data={