Merge ca10298 into f2433e5

simonsobs · May 26, 2022 · 7997f83 · 7997f83
2 parents f2433e5 + ca10298
commit 7997f83
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/ocs/ocs_client.py b/ocs/ocs_client.py
@@ -112,6 +112,9 @@ def __init__(self, instance_id, **kwargs):
             setattr(self, _opname_to_attr(name),
                     _get_op('process', name, encoded, self._client))
 
+    def __repr__(self):
+        return f"OCSClient('{self.instance_id}')"
+
 
 def _humanized_time(t):
     if abs(t) < 1.:

diff --git a/tests/test_ocs_client.py b/tests/test_ocs_client.py
@@ -161,6 +161,11 @@ def test_matched_client_object(self):
         assert hasattr(client, 'process_name')
         assert hasattr(client, 'task_name')
 
+    @patch('ocs.site_config.get_control_client', fake_get_control_client)
+    def test_ocsclient_repr(self):
+        client = OCSClient('agent-id')
+        assert repr(client) == "OCSClient('agent-id')"
+
 
 class TestOCSReply:
     """Test various scenarios in OCSReply decoding. Since the representation