From 5d2abce4a0e8e0f94e02c75cbea88132aaf19322 Mon Sep 17 00:00:00 2001 From: Elena ``of Valhalla'' Grandi Date: Sun, 23 Jun 2019 11:11:20 +0200 Subject: CtlClient infrastructure --- tests/test_ctl.py | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) (limited to 'tests') 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) -- cgit v1.2.3