mitogen/tests/id_allocation_test.py

import unittest

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)
        assert id_ == (self.router.id_allocator.next_id - 1)


if __name__ == '__main__':
    unittest.main()