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()
        # Master's allocator named the context 1.
        self.assertEquals(1, context.context_id)

        # First call from slave allocates a block (2..1001)
        id_ = context.call(id_allocation.allocate_an_id)
        self.assertEqual(id_, 2)

        # Second call from slave allocates from block (3..1001)
        id_ = context.call(id_allocation.allocate_an_id)
        self.assertEqual(id_, 3)

        # Subsequent master allocation does not collide
        c2 = self.router.local()
        self.assertEquals(1002, c2.context_id)


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