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
