CtlClient infrastructure

1 files changed, 21 insertions, 0 deletions

diff --git a/tests/test_ctl.py b/tests/test_ctl.py
index f58ce5a..ed6a5e4 100644
--- a/tests/test_ctl.py
+++ b/tests/test_ctl.py
@@ -1,5 +1,6 @@
 import os
 import tempfile
+import unittest
 
 from tornado.testing import AsyncTestCase, gen_test
 
@@ -40,3 +41,23 @@ class TestCtlServerCommands(AsyncTestCase):
         ))
         self.assertIn('ok', res)
         self.assertEqual(len(self.ctl_server.app.store.objects), 1)
+
+
+class TestCtlClient(unittest.TestCase):
+    def setUp(self):
+        self.client = ctl.CtlClient(
+            config.Config('tests/data/test_config.yaml')
+            )
+
+    def test_cmd_ping(self):
+        cmd = self.client.ping()
+        self.assertEqual(cmd, {"command": "ping"})
+
+    def test_cmd_add_object(self):
+        obj = {
+            "id": "http://example.org/123456",
+            "type": "object",
+            }
+        cmd = self.client.add_object(obj)
+        self.assertEqual(cmd['command'], 'add_object')
+        self.assertEqual(cmd['obj'], obj)