TST: Add some proxy tests

Author: mferrera

pyproject.toml

@@ -69,7 +69,7 @@ norecursedirs = [
     "dist",
     "build",
 ]
-addopts = "-ra --strict-markers"
+addopts = "-ra --strict-markers --basetemp=/tmp/pytest"
 markers = [
     "integration: Marks a test as an integration test",
     "onprem: Marks a test as valid only in an on-prem environment",

src/runrms/api/proxy.py

@@ -279,7 +279,7 @@ def __contains__(self, item: Any) -> Any:
     def __str__(self) -> str:
         """Forward string conversion to worker."""
         if not self._path:
-            return f"<RmsApiProxy connected to {self._zmq_address}"
+            return f"<RmsApiProxy connected to {self._zmq_address}>"
 
         request = Request(msg_type="call", path=[*self._path, "__str__"])
         response = self._send_request(request)
@@ -288,7 +288,7 @@ def __str__(self) -> str:
     def __repr__(self) -> str:
         """Forward repr conversion to worker."""
         if not self._path:
-            return f"<RmsApiProxy connected to {self._zmq_address}"
+            return f"<RmsApiProxy connected to {self._zmq_address}>"
 
         request = Request(msg_type="call", path=[*self._path, "__repr__"])
         response = self._send_request(request)

tests/test_api/test_proxy.py

@@ -73,6 +73,12 @@ def test_private_attr_raises(proxy_with_mocks: RmsApiProxy) -> None:
         _ = proxy_with_mocks._private
 
 
+def test_str_repr_on_root_proxy_gives_instance(proxy_with_mocks: RmsApiProxy) -> None:
+    """Str and repr print the root RmsApiProxy instance with its address."""
+    assert str(proxy_with_mocks).startswith("<RmsApiProxy connected to ipc://")
+    assert repr(proxy_with_mocks).startswith("<RmsApiProxy connected to ipc://")
+
+
 def test_version_special_case(
     proxy_with_mocks: RmsApiProxy, mock_socket: MagicMock
 ) -> None:
@@ -497,6 +503,13 @@ def test_shutdown_sends_request(
     assert request.msg_type == "shutdown"
 
 
+def test_shutdown_on_child_proxy_raises(proxy_with_mocks: RmsApiProxy) -> None:
+    """Accessing attribute returns new proxy with extended path."""
+    child = proxy_with_mocks.some_attr
+    with pytest.raises(RuntimeError, match="Shutdown can only be called on root"):
+        child._shutdown()
+
+
 def test_cleanup_closes_socket(
     proxy_with_mocks: RmsApiProxy, mock_socket: MagicMock
 ) -> None: