def test_grow(self):
rendezvous = RendezvousHash()
placements = {}
for i in range(10):
rendezvous.add_node(str(i))
placements[str(i)] = []
for i in range(1000):
node = rendezvous.find_node(str(i))
placements[node].append(i)
new_placements = {}
for i in range(20):
rendezvous.add_node(str(i))
new_placements[str(i)] = []
for i in range(1000):
node = rendezvous.find_node(str(i))
new_placements[node].append(i)
keys = [k for sublist in placements.values() for k in sublist]
new_keys = [k for sublist in new_placements.values() for k in sublist]
self.assertEqual(sorted(keys), sorted(new_keys))
added = 0
removed = 0
for node, assignments in new_placements.items():
after = set(assignments)
before = set(placements.get(node, []))
removed += len(before.difference(after))
added += len(after.difference(before))
self.assertEqual(added, removed)
self.assertEqual(1062, (added + removed))
def test_grow(self):
rendezvous = RendezvousHash()
placements = {}
for i in range(10):
rendezvous.add_node(str(i))
placements[str(i)] = []
for i in range(1000):
node = rendezvous.find_node(str(i))
placements[node].append(i)
new_placements = {}
for i in range(20):
rendezvous.add_node(str(i))
new_placements[str(i)] = []
for i in range(1000):
node = rendezvous.find_node(str(i))
new_placements[node].append(i)
keys = [k for sublist in placements.values() for k in sublist]
new_keys = [k for sublist in new_placements.values() for k in sublist]
self.assertEqual(sorted(keys), sorted(new_keys))
added = 0
removed = 0
for node, assignments in new_placements.items():
after = set(assignments)
before = set(placements.get(node, []))
removed += len(before.difference(after))
added += len(after.difference(before))
self.assertEqual(added, removed)
self.assertEqual(1062, (added + removed))
def test_find_node_after_addition(self):
nodes = ['0', '1', '2']
rendezvous = RendezvousHash(nodes=nodes)
self.assertEqual('0', rendezvous.find_node('ok'))
self.assertEqual('1', rendezvous.find_node('mykey'))
self.assertEqual('2', rendezvous.find_node('wat'))
self.assertEqual('2', rendezvous.find_node('lol'))
rendezvous.add_node('3')
self.assertEqual('0', rendezvous.find_node('ok'))
self.assertEqual('1', rendezvous.find_node('mykey'))
self.assertEqual('2', rendezvous.find_node('wat'))
self.assertEqual('3', rendezvous.find_node('lol'))
def test_find_node_after_addition(self):
nodes = ['0', '1', '2']
rendezvous = RendezvousHash(nodes=nodes)
self.assertEqual('0', rendezvous.find_node('ok'))
self.assertEqual('1', rendezvous.find_node('mykey'))
self.assertEqual('2', rendezvous.find_node('wat'))
self.assertEqual('2', rendezvous.find_node('lol'))
rendezvous.add_node('3')
self.assertEqual('0', rendezvous.find_node('ok'))
self.assertEqual('1', rendezvous.find_node('mykey'))
self.assertEqual('2', rendezvous.find_node('wat'))
self.assertEqual('3', rendezvous.find_node('lol'))
def test_add_node(self):
rendezvous = RendezvousHash()
rendezvous.add_node('1')
self.assertEqual(1, len(rendezvous.nodes))
rendezvous.add_node('1')
self.assertEqual(1, len(rendezvous.nodes))
rendezvous.add_node('2')
self.assertEqual(2, len(rendezvous.nodes))
rendezvous.add_node('1')
self.assertEqual(2, len(rendezvous.nodes))
def test_add_node(self):
rendezvous = RendezvousHash()
rendezvous.add_node('1')
self.assertEqual(1, len(rendezvous.nodes))
rendezvous.add_node('1')
self.assertEqual(1, len(rendezvous.nodes))
rendezvous.add_node('2')
self.assertEqual(2, len(rendezvous.nodes))
rendezvous.add_node('1')
self.assertEqual(2, len(rendezvous.nodes))
