def test_list_error(self): # Tracker tracker = Tracker(9995) tracker.list = Tracker.populate(40) client1 = Peer(9992) # List Error Message request = Message() request.set_packet_type(Message.ERROR) request.set_packet(Message.LIST) request.set_data(9992) client1.produce_response(IP=socket.gethostbyname(socket.gethostname()), port=9995, message=request) flag = True try: response = client1.consume_receive()[2] except Exception: flag = False client1.client.close() tracker.client.close() client1.server.close() tracker.server.close() self.assertEqual(flag, True)
def test_list_request(self): # Tracker tracker = Tracker(9990) tracker.list = Tracker.populate(40) client1 = Peer(9991) # List Request Message request = Message() request.set_packet_type(Message.REQUEST) request.set_packet(Message.LIST) request.set_data(9991) client1.produce_response(IP=socket.gethostbyname(socket.gethostname()), port=9990, message=request) response = client1.consume_receive()[2] # Checking the sublist flag = True for i in response.get_data(): if not tracker.list.is_member(i): flag = False client1.client.close() tracker.client.close() client1.server.close() tracker.server.close() self.assertEqual(flag, True)
def test_cheese_response(self): message = Message() message.set_packet(Message.CHEESE) message.set_packet_type(Message.RESPONSE) data = { "transactions": [{ "input": [("110812f67fa1e1f0117f6f3d70241c1a42a7b07711a93c2477cc516d9042f9db", 0)], "wallet": [ "110812f67fa1e1f0117f6f3d70241c1a42a7b07711a93c2477cc516d9042f9db01", "110812f67fa1e1f0117f6f3d70241c1a42a7b07711a93c2477cc516d9042f9db02", "00000000000000000000000000000000000000000000000000000000000000000" ], "amount": [10, 10, 0], "signature": [ "110812f67fa1e1f0117f6f3d70241c1a42a7b07711a93c2477cc516d9042f9d" ] }, { "input": [("110812f67fa1e1f0117f6f3d70241c1a42a7b07711a93c2477cc516d9042f9db", 0)], "wallet": [ "110812f67fa1e1f0117f6f3d70241c1a42a7b07711a93c2477cc516d9042f9db01", "110812f67fa1e1f0117f6f3d70241c1a42a7b07711a93c2477cc516d9042f9db02", "00000000000000000000000000000000000000000000000000000000000000000" ], "amount": [10, 10, 0], "signature": [ "110812f67fa1e1f0117f6f3d70241c1a42a7b07711a93c2477cc516d9042f9d" ] }], "nonce": 123456789 } message.set_data(data) writer = MessageWriter(message) string = writer.write() self.assertEqual( string, "CHEESE RESPONSE " + "110812f67fa1e1f0117f6f3d70241c1a42a7b07711a93c2477cc516d9042f9db" + " 0 110812f67fa1e1f0117f6f3d70241c1a42a7b07711a93c2477cc516d9042f9db01 110812f67fa1e1f0117f6f3d70241c1a42a7b07711a93c2477cc516d9042f9db02" + " 00000000000000000000000000000000000000000000000000000000000000000" + " 10 10 0 110812f67fa1e1f0117f6f3d70241c1a42a7b07711a93c2477cc516d9042f9d " + "110812f67fa1e1f0117f6f3d70241c1a42a7b07711a93c2477cc516d9042f9db" + " 0 110812f67fa1e1f0117f6f3d70241c1a42a7b07711a93c2477cc516d9042f9db01 110812f67fa1e1f0117f6f3d70241c1a42a7b07711a93c2477cc516d9042f9db02" + " 00000000000000000000000000000000000000000000000000000000000000000" + " 10 10 0 110812f67fa1e1f0117f6f3d70241c1a42a7b07711a93c2477cc516d9042f9d 123456789\r\n" )
def test_transaction_error(self): message = Message() message.set_packet(Message.TRANSACTION) message.set_packet_type(Message.ERROR) writer = MessageWriter(message) string = writer.write() self.assertEqual(string, "TRANSACTION ERROR\r\n")
def test_list_error(self): message = Message() message.set_packet(Message.LIST) message.set_packet_type(Message.ERROR) writer = MessageWriter(message) string = writer.write() self.assertEqual(string, "LIST ERROR\r\n")
def test_cheese_error(self): message = Message() message.set_packet(Message.CHEESE) message.set_packet_type(Message.ERROR) writer = MessageWriter(message) string = writer.write() self.assertEqual(string, "CHEESE ERROR\r\n")
def test_transaction_request(self): message = Message() message.set_packet(Message.TRANSACTION) message.set_packet_type(Message.REQUEST) writer = MessageWriter(message) string = writer.write() self.assertEqual(string, "TRANSACTION REQUEST\r\n")
def test_list_request(self): message = Message() message.set_packet(Message.LIST) message.set_packet_type(Message.REQUEST) message.set_data(1234) writer = MessageWriter(message) string = writer.write() self.assertEqual(string, "LIST REQUEST 1234\r\n")
def test_member_report(self): message = Message() message.set_packet(Message.MEMBER) message.set_packet_type(Message.REPORT) message.set_data(("192.168.1.1", 2)) writer = MessageWriter(message) string = writer.write() self.assertEqual(string, "MEMBER REPORT 192.168.1.1 2\r\n")
def test_cheese_request(self): message = Message() message.set_packet(Message.CHEESE) message.set_packet_type(Message.REQUEST) message.set_data("af4721920") writer = MessageWriter(message) string = writer.write() self.assertEqual(string, "CHEESE REQUEST af4721920\r\n")
def test_list_response(self): message = Message() message.set_packet(Message.LIST) message.set_packet_type(Message.RESPONSE) message.set_data([("192.168.1.1", 2), ("127.0.0.0.1", 1234)]) writer = MessageWriter(message) string = writer.write() self.assertEqual( string, "LIST RESPONSE 192.168.1.1 2 " + "127.0.0.0.1 1234\r\n")
def process_transaction_request(self): """ We process a TRANSACTION REQUEST """ # We get a list of transactions transaction_list = self.transaction_list.ressource # We get the last transaction last = self.transaction_list.read(transaction_list.get_last) # We send back the response message = Message() message.set_packet(Message.TRANSACTION) message.set_packet_type(Message.RESPONSE) message.set_data(last) return message
def test_member_error(self): # Tracker print('begin') tracker = Tracker(9996) print(tracker.client) print('end') tracker.list = Tracker.populate(5) print('beginC') client1 = Peer(9997) print('endC') print("step 1") # Add the client to Member List print(tracker.list.print_list()) tracker.list.add_member( (socket.gethostbyname(client1.server.host_name), client1.server.port)) print("step 2") print(tracker.list.print_list()) # List Request Message request = Message() request.set_packet_type(Message.REPORT) request.set_packet(Message.MEMBER) request.set_data((socket.gethostbyname(client1.server.host_name), client1.server.port)) print("step 3") client1.produce_response(IP=socket.gethostbyname(socket.gethostname()), port=9996, message=request) print("step 4") # NOOOOOOOOOT FINISHEDDDDDDDDDDDDDD print("step 5") # Checking if Client is in Member List flag = tracker.list.is_member( (socket.gethostbyname(client1.server.host_name), client1.server.port)) print(flag) print("step 6") #time.sleep(2) print('Socket ' + str(client1.client.socket)) client1.client.close() #tracker.client.close() client1.server.close() tracker.server.close() print("fini") self.assertEqual(flag, True)
def process_message(self, tuple): (ip, socket, message) = tuple # Handling Messages if message.get_packet() == Message.LIST: # List REQUEST # Creating a RESPONSE message if message.get_packet_type() == Message.REQUEST: # Adding member to list port = int(message.get_data()) member_list = self.member_list.ressource self.member_list.write(member_list.add_member, (ip, port)) response = Message() response.set_packet(Message.LIST) response.set_packet_type(Message.RESPONSE) member_list = self.member_list.ressource sublist = self.member_list.read(member_list.get_sublist) response.set_data(sublist) self.produce_response(socket=socket, message=response, close=True) # List ERROR elif message.get_packet_type() == Message.ERROR: print(message.get_data()) # Member REPORT elif (message.get_packet_type() == Message.REPORT and message.get_packet() == Message.MEMBER): try: ip_port = message.get_data() except ValueError: raise Exception('Error: Invalid IP Port') (ip, port) = ip_port self.produce_ping(ip, port) else: print('Error: No Message Type Detected\n')
def create_transaction(self): """ We will create a transaction after pushing the button """ # We get the amount amount = int(self.entry_amount.get()) # We create the transaction transaction_user = Transaction.create_user(self.member.money_list, amount, self.entry_receiver.get()) # We add it to the transactions list transaction_list = self.member.transaction_list.ressource self.member.transaction_list.write(transaction_list.add, transaction_user) # We broadcast the transaction message = Message() message.set_packet(Message.TRANSACTION) message.set_packet_type(Message.BROADCAST) message.set_data(transaction_user) self.member.broadcast(message) self.entry_amount.delete(0, 'end') self.entry_receiver.delete(0, 'end')
def test_transaction_response(self): message = Message() message.set_packet(Message.TRANSACTION) message.set_packet_type(Message.RESPONSE) data = { "input": [("110812f67fa1e1f0117f6f3d70241c1a4" + "2a7b07711a93c2477cc516d9042f9db", 0)], "wallet": [ "110812f67fa1e1f0117f6f3d70241c1a" + "42a7b07711a93c2477cc516d9042f9db01", "110812f67fa1e1f0117f6f3d70241c1a42" + "a7b07711a93c2477cc516d9042f9db02", "0000000000000000000000000000000000" + "0000000000000000000000000000000" ], "amount": [10, 10, 0], "signature": [ "110812f67fa1e1f0117f6f3d70241" + "c1a42a7b07711a93c2477cc516d9042f9d" ] } message.set_data(data) writer = MessageWriter(message) string = writer.write() self.assertEqual( string, "TRANSACTION RESPONSE 110812f67fa1e1f0117f6f3d70241c1a42a7b07711a93c2477cc516d9042f9db" + " 0 110812f67fa1e1f0117f6f3d70241c1a42a7b07711a93c2477cc516d9042f9db01 110812f67fa1e1f0117f6f3d70241c1a42a7b07711a93c2477cc516d9042f9db02" + " 00000000000000000000000000000000000000000000000000000000000000000" + " 10 10 0 110812f67fa1e1f0117f6f3d70241c1a42a7b07711a93c2477cc516d9042f9d\r\n" )
def test_set_packet(self): message = Message() message.set_packet(Message.LIST) self.assertEqual(message.get_packet(), Message.LIST) message.set_packet(Message.MEMBER) self.assertEqual(message.get_packet(), Message.MEMBER) message.set_packet(Message.TRANSACTION) self.assertEqual(message.get_packet(), Message.TRANSACTION) message.set_packet(Message.CHEESE) self.assertEqual(message.get_packet(), Message.CHEESE) message.set_packet(Message.RESPONSE) self.assertEqual(message.get_packet(), Message.NOTHING) message.set_packet(Message.REQUEST) self.assertEqual(message.get_packet(), Message.NOTHING) message.set_packet(Message.REPORT) self.assertEqual(message.get_packet(), Message.NOTHING) message.set_packet(Message.ERROR) self.assertEqual(message.get_packet(), Message.NOTHING)