def test_random_key_in_bucket_range(self):
"""
Ensures the returned key is within the expected bucket range.
"""
parent_node_id = 'deadbeef'
r = RoutingTable(parent_node_id)
bucket = Bucket(1, 2)
r._buckets[0] = bucket
expected = 1
actual = int(r._random_key_in_bucket_range(0), 0)
self.assertEqual(expected, actual)
def test_random_key_in_bucket_range_long(self):
"""
Ensures that random_key_in_bucket_range works with large numbers.
"""
minimum = 978675645342314253647586978
maximum = 978675645342314253647586979
parent_node_id = 'abc'
r = RoutingTable(parent_node_id)
bucket = KBucket(minimum, maximum)
r._buckets[0] = bucket
expected = minimum
actual = int(r._random_key_in_bucket_range(0).encode('hex'), 16)
self.assertEqual(expected, actual)
