def test02_main(self):
call_stack = []
def call_node_simulator(nodeaddr, request):
call_stack.append((nodeaddr, request.parameters['start_key'], request.parameters['end_key']))
return 0, 'ok'
operator = DHTOperator('127.0.0.1:1987', home_dir=TEST_FS_RANGE_DIR)
operator.ranges_table.append(0, 99, 'first_range_holder')
operator.ranges_table.append(100, 149, 'second_range_holder')
operator.ranges_table.append(300, 499, 'third_range_holder')
operator.call_node = call_node_simulator
operator.start_as_dht_member()
operator.start_as_dht_member()
operator.start_as_dht_member()
self.assertEqual(len(call_stack), 3)
self.assertEqual(call_stack[0][0], 'third_range_holder')
self.assertEqual(call_stack[0][1], 400)
self.assertEqual(call_stack[0][2], 499)
self.assertEqual(call_stack[1][0], 'first_range_holder')
self.assertEqual(call_stack[1][1], 50)
self.assertEqual(call_stack[1][2], 99)
self.assertEqual(call_stack[2][0], 'second_range_holder')
self.assertEqual(call_stack[2][1], 125)
self.assertEqual(call_stack[2][2], 149)
operator.start_as_dht_member() #wait timeout
self.assertEqual(call_stack[3][0], 'second_range_holder')
self.assertEqual(call_stack[3][1], 125)
self.assertEqual(call_stack[3][2], 149)
operator.stop()
def test03_discovery_range(self):
shutil.rmtree(TEST_FS_RANGE_DIR)
os.mkdir(TEST_FS_RANGE_DIR)
os.makedirs(os.path.join(TEST_FS_RANGE_DIR, 'dht_range', '%040x_%040x'%(0, 99)))
call_stack = []
def call_node_simulator(nodeaddr, request):
call_stack.append((nodeaddr, request.parameters['start_key'], request.parameters['end_key']))
return 0, 'ok'
operator = DHTOperator('127.0.0.1:1987', home_dir=TEST_FS_RANGE_DIR)
try:
operator.ranges_table.append(0, 99, 'first_range_holder')
mod_index = operator.ranges_table.get_mod_index()
operator.ranges_table.append(100, 149, 'second_range_holder')
operator.ranges_table.append(300, 499, 'third_range_holder')
mod_index1 = operator.ranges_table.get_mod_index()
self.assertNotEqual(mod_index, mod_index1)
tbl_dump = operator.ranges_table.dump()
operator.ranges_table.load(tbl_dump)
mod_index2 = operator.ranges_table.get_mod_index()
self.assertEqual(mod_index1, mod_index2)
operator.call_node = call_node_simulator
operator.start_as_dht_member()
operator.start_as_dht_member()
operator.start_as_dht_member()
self.assertEqual(len(call_stack), 3)
self.assertEqual(call_stack[0][0], 'first_range_holder')
self.assertEqual(call_stack[0][1], 50)
self.assertEqual(call_stack[0][2], 99)
self.assertEqual(call_stack[1][0], 'third_range_holder')
self.assertEqual(call_stack[1][1], 400)
self.assertEqual(call_stack[1][2], 499)
self.assertEqual(call_stack[2][0], 'second_range_holder')
self.assertEqual(call_stack[2][1], 125)
self.assertEqual(call_stack[2][2], 149)
def call_node_error_simulator(nodeaddr, request):
if nodeaddr == 'second_range_holder':
return 666, 'No route to host!'
call_stack.append((nodeaddr, request.parameters['start_key'], request.parameters['end_key']))
return 0, 'ok'
operator.call_node = call_node_error_simulator
operator.start_as_dht_member()
self.assertEqual(call_stack[3][0], 'third_range_holder')
self.assertEqual(call_stack[3][1], 400)
self.assertEqual(call_stack[3][2], 499)
finally:
operator.stop()
