mitogen/tests/id_allocation_test.py

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()