def test_forwards_two_new_nodes(self): test_port_2 = 32323 test_files_2 = ["2_file_a.txt", "2_file_b.txt"] self.send_broadcast(broadcast_message(TEST_PORT, TEST_FILES)) self.send_broadcast(broadcast_message(test_port_2, test_files_2)) sleep(TEST_WAIT) self.receiver.tell.assert_any_call(remote_files_message(TEST_IP, TEST_PORT, TEST_FILES)) self.receiver.tell.assert_any_call(remote_files_message(TEST_IP, test_port_2, test_files_2))
def test_forwards_two_new_nodes(self): test_port_2 = 32323 test_files_2 = ["/2_file_a.txt", "/2_file_b.txt"] self.send_broadcast_message(broadcast_message(TEST_PORT, TEST_FILES)) self.send_broadcast_message( broadcast_message(test_port_2, test_files_2)) sleep(TEST_WAIT) self.receiver.tell.assert_any_call( remote_files_message(TEST_IP, TEST_PORT, TEST_FILES)) self.receiver.tell.assert_any_call( remote_files_message(TEST_IP, test_port_2, test_files_2))
def test_doesnt_report_own_node(self): own_ip = socket.gethostbyname(socket.gethostname()) self.send_broadcast_message(broadcast_message(TEST_PORT, TEST_FILES), own_ip) sleep(TEST_WAIT) self.receiver.tell.assert_not_called()
def test_broadcast_with_updated_files(self): files = ["/ann_file_1.txt", "/ann_file_2.txt"] self.announcer.tell(local_files_message(files)) sleep(TEST_WAIT) received = self.mockedSocket.received_data() expected = broadcast_message(TEST_PORT, files) self.assertTrue( expected in received, "\n expected: [%s]" "\n received: [%s]" % (expected, received))
def test_broadcast_with_updated_files(self): files = ["ann_file_1.txt", "ann_file_2.txt"] self.announcer.tell(local_files_message(files)) sleep(TEST_WAIT) received = self.mockedSocket.received() expected = broadcast_message(TEST_PORT, files) self.assertTrue(expected in received, "\n expected: [%s]" "\n received: [%s]" % (expected, received))
def test_no_reaction_on_unknown_message(self): payload = 'bla bla bla' udp_packet = IP(src=TEST_IP) / UDP(dport=self.broadcast_port) / payload sendrecv.send(udp_packet) sleep(TEST_WAIT) self.receiver.tell.assert_not_called() self.send_broadcast_message(broadcast_message(TEST_PORT, TEST_FILES)) sleep(TEST_WAIT) self.receiver.tell.assert_called_with( remote_files_message(TEST_IP, TEST_PORT, TEST_FILES))
def test_received_broadcast(self): sleep(TEST_WAIT) received = self.mockedSocket.received_data() expected = broadcast_message(TEST_PORT, []) self.assertTrue(expected in received, received)
def test_received_broadcast(self): sleep(TEST_WAIT) received = self.mockedSocket.received() expected = broadcast_message(TEST_PORT, []) self.assertTrue(expected in received, received)
def test_forwards_new_node(self): self.send_broadcast_message(broadcast_message(TEST_PORT, TEST_FILES)) sleep(TEST_WAIT) self.receiver.tell.assert_called_with( remote_files_message(TEST_IP, TEST_PORT, TEST_FILES))
def test_forwards_new_node(self): self.send_broadcast(broadcast_message(TEST_PORT, TEST_FILES)) sleep(TEST_WAIT) self.receiver.tell.assert_called_with(remote_files_message(TEST_IP, TEST_PORT, TEST_FILES))