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_type(self):
message = Message()
message.set_packet_type(Message.LIST)
self.assertEqual(message.get_packet_type(), Message.NOTHING)
message.set_packet_type(Message.MEMBER)
self.assertEqual(message.get_packet_type(), Message.NOTHING)
message.set_packet_type(Message.TRANSACTION)
self.assertEqual(message.get_packet_type(), Message.NOTHING)
message.set_packet_type(Message.CHEESE)
self.assertEqual(message.get_packet_type(), Message.NOTHING)
message.set_packet_type(Message.RESPONSE)
self.assertEqual(message.get_packet_type(), Message.RESPONSE)
message.set_packet_type(Message.REQUEST)
self.assertEqual(message.get_packet_type(), Message.REQUEST)
message.set_packet_type(Message.REPORT)
self.assertEqual(message.get_packet_type(), Message.REPORT)
message.set_packet_type(Message.ERROR)
self.assertEqual(message.get_packet_type(), Message.ERROR)
