class DirectoryServiceTest(unittest.TestCase):
def setUp(self):
self.ds = DirectoryService(["tst://socket"], "The Process")
self.ds.run(block=False)
self.s = self.ds._server_socks["tst://socket"]
def test_gc(self):
msgs = []
def log_debug(msg):
msgs.append(msg)
self.ds.log_debug = log_debug
self.s = None
self.ds = None
self.assertEqual(msgs, ['Garbage collecting loop', 'Stopping loop', 'Waiting for loop to finish',
"Loop finished", 'Loop garbage collected'])
def test_update_instance_attributes(self):
d1 = self.ds.createMockDevice2("MD1")
self.assertEqual(d1.child, None)
self.assertEqual(d1.attributes["child"].typ.labels, ("MD1",))
l = MagicMock()
d1.add_listener(l, "attributes.child")
d2 = self.ds.createMockDevice2("MD2")
self.assertEqual(d2.child, None)
self.assertEqual(d1.attributes["child"].typ.labels, ("MD1", "MD2"))
self.assertEqual(d2.attributes["child"].typ.labels, ("MD1", "MD2"))
l.assert_called_once_with(
d1.attributes["child"], dict(type=d1.attributes["child"].typ))
self.assertRaises(
AssertionError, d1.attributes["child"].update, "MD2ss")
d1.child = "MD2"
self.assertEqual(d1.child, d2)
self.assertEqual(d1.attributes["child"].to_dict()["value"], "MD2")
class ZmqSystemTest(unittest.TestCase):
# class ZmqSystemTest(object):
def setUp(self):
# Spawn ds under cothread
self.ds = DirectoryService(["zmq://ipc:///tmp/sock.ipc"])
self.ds.run(block=False)
self.lp = Process(
[], "Local Process", ds_string="zmq://ipc:///tmp/sock.ipc")
self.lp.run(block=False)
self.lp.ds.createCounter(name="The Counter")
self.c = self.lp.get_device("The Counter")
def test_simple_function(self):
import cothread
cothread.Sleep(0.2)
start = self.c.getCount()
self.assertEqual(self.c.hello(), "world")
# Hello world takes about 10 ticks
self.assertAlmostEqual(self.c.getCount(), start + 10, delta=3)
# Do a long running call
s = cothread.Spawn(self.c.longHello)
# Check it returns immediately
self.assertAlmostEqual(self.c.getCount(), start + 10, delta=3)
self.assertEqual(self.c.hello(), "world")
# Hello world takes 10 ticks
self.assertAlmostEqual(self.c.getCount(), start + 20, delta=3)
self.assertEqual(s.Wait(), "long world")
# Long hello takes about 50 ticks from send
self.assertAlmostEqual(self.c.getCount(), start + 60, delta=8)
s.Wait()
def tearDown(self):
self.c = None
self.lp.ds.exit()
self.lp.exit()
self.lp = None
self.ds.loop_wait()
self.ds = None
