def test_addresses_from_pool_wrong(self):
name = 'SomeService'
with self.assertRaises(ValueError):
list(get_addresses_from_pool(name, ('localhost', 'unknown')))
with self.assertRaises(ValueError):
list(get_addresses_from_pool(name, ('localhost', '')))
with self.assertRaises(ValueError):
list(get_addresses_from_pool(name, ('localhost', object())))
def test_addresses_from_pool_digit(self):
name = 'SomeService'
address = 'localhost', 10500
self.assertEqual((address,),
tuple(get_addresses_from_pool(name, address)))
self.assertEqual((address,),
tuple(get_addresses_from_pool(name, (address[0], str(address[1])))))
self.assertEqual((address,),
tuple(get_addresses_from_pool(name, address, (11000, 12000))))
def test_addresses_from_pool_auto(self):
name = 'SomeService'
address = 'localhost', None
self.assertEqual(((address[0], 0),),
tuple(get_addresses_from_pool(name, address)))
g = get_addresses_from_pool(name, address, (11000, 12000))
self.assertEqual((address[0], 11538), g.next())
self.assertEqual((address[0], 11539), g.next())
skip(g, 99)
self.assertEqual((address[0], 11639), g.next())
skip(g, 12000 - 11639 - 1)
self.assertEqual((address[0], 11000), g.next())
skip(g, 11538 - 11000 - 2)
self.assertEqual((address[0], 11537), g.next())
self.assertEqual([], list(g))