import unittest2 import testlib import id_allocation class SlaveTest(testlib.RouterMixin, testlib.TestCase): def test_slave_allocates_id(self): context = self.router.local() id_ = context.call(id_allocation.allocate_an_id) self.assertEqual(id_, self.router.id_allocator.next_id - 1) if __name__ == '__main__': unittest2.main()