diff options
Diffstat (limited to 'tests')
-rw-r--r-- | tests/test_ctl.py | 21 |
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) |