def test_puncture_message_conversion(self):
for c in self._conversions:
if isinstance(c, PunctureConversion):
meta = self._di._community.get_meta_message(
PUNCTURE_MESSAGE_NAME)
sender_lan = Address.ipv4("0.0.0.1:1232")
sender_wan = Address.ipv4("23.12.32.123:1232")
sender_id = os.urandom(16)
address_vote = Address.ipv6("[::0]:12145")
endpoint_id = os.urandom(16)
message = meta.impl(
authentication=(self._di._community.my_member, ),
distribution=(self._di._community.claim_global_time(), ),
destination=(Candidate(address_vote.addr(), True), ),
payload=(sender_lan, sender_wan, sender_id, address_vote,
endpoint_id))
encoded = c.encode_payload(message)
placeholder = c.Placeholder(None, meta, 0, encoded, False,
True)
_, x = c.decode_payload(placeholder, 0,
str(encoded[0]) + encoded[1])
self.assertEqual(x.sender_lan, sender_lan)
self.assertEqual(x.sender_wan, sender_wan)
self.assertEqual(x.sender_id, sender_id)
self.assertEqual(x.address_vote, address_vote)
self.assertEqual(x.endpoint_id, endpoint_id)
def test_file_hash_message_conversion(self):
for c in self._conversions:
if isinstance(c, FileHashConversion):
meta = self._di._community.get_meta_message(
FILE_HASH_MESSAGE_NAME)
filename = "asdf.asdf"
dir_ = ""
roothash = "asdfj23j09f09sjfewef"
size = 103425
timestamp = 3232342.2342
address = Address.ipv4("0.0.0.0:1")
message = meta.impl(
authentication=(self._di._community.my_member, ),
distribution=(self._di._community.claim_global_time(),
self._di._community._file_hash_distribution.
claim_sequence_number()),
payload=(filename, dir_, roothash, size, timestamp,
[address]))
encoded = c.encode_payload(message)
placeholder = c.Placeholder(None, meta, 0, encoded, False,
True)
_, x = c.decode_payload(placeholder, 0,
str(encoded[0]) + encoded[1])
self.assertEqual(x.filename, filename)
self.assertEqual(x.directories, dir_)
self.assertEqual(x.roothash, roothash)
self.assertEqual(x.size, size)
self.assertEqual(x.timestamp, timestamp)
self.assertEqual(x.addresses[0], address)
def test_addresses_request_message_conversion(self):
for c in self._conversions:
if isinstance(c, AddressesRequestConversion):
meta = self._di._community.get_meta_message(ADDRESSES_REQUEST_MESSAGE_NAME)
sender_lan = Address.ipv4("0.0.0.1:1232")
sender_wan = Address.ipv4("23.12.32.123:1232")
endpoint_id = os.urandom(16)
wan_address = Address.ipv4("12.23.21.21:1234")
message = meta.impl(authentication=(self._di._community.my_member,),
distribution=(self._di._community.claim_global_time(),),
destination=(Candidate(wan_address.addr(), True),),
payload=(sender_lan, sender_wan, endpoint_id, wan_address))
encoded = c.encode_payload(message)
placeholder = c.Placeholder(None, meta, 0, encoded, False, True)
_, x = c.decode_payload(placeholder, 0, str(encoded[0])+encoded[1])
self.assertEqual(x.sender_lan, sender_lan)
self.assertEqual(x.sender_wan, sender_wan)
self.assertEqual(x.endpoint_id, endpoint_id)
self.assertEqual(x.wan_address, wan_address)
def test_puncture_response_message_conversion(self):
for c in self._conversions:
if isinstance(c, PunctureResponseConversion):
meta = self._di._community.get_meta_message(PUNCTURE_RESPONSE_MESSAGE_NAME)
sender_lan = Address.ipv4("0.0.0.1:1232")
sender_wan = Address.ipv4("23.12.32.123:1232")
address_vote = Address.ipv6("[::0]:12145")
endpoint_id = os.urandom(16)
message = meta.impl(authentication=(self._di._community.my_member,),
distribution=(self._di._community.claim_global_time(),),
destination=(Candidate(address_vote.addr(), True),),
payload=(sender_lan, sender_wan, address_vote, endpoint_id))
encoded = c.encode_payload(message)
placeholder = c.Placeholder(None, meta, 0, encoded, False, True)
_, x = c.decode_payload(placeholder, 0, str(encoded[0])+encoded[1])
self.assertEqual(x.sender_lan, sender_lan)
self.assertEqual(x.sender_wan, sender_wan)
self.assertEqual(x.address_vote, address_vote)
self.assertEqual(x.endpoint_id, endpoint_id)
def test_uav_message_conversion(self):
for c in self._conversions:
if isinstance(c, APIMessageConversion):
meta = self._di._community.get_meta_message(API_MESSAGE_NAME)
data = "asjfdioewf"
addresses = [Address.ipv4("0.0.0.1:1232"), Address.ipv6("[::0]:12145"), Address(port=32532)]
message = meta.impl(authentication=(self._di._community.my_member,),
distribution=(self._di._community.claim_global_time(),),
destination=(Candidate(addresses[0].addr(), True),),
payload=(data,))
encoded = c.encode_payload(message)
placeholder = c.Placeholder(None, meta, 0, encoded, False, True)
_, x = c.decode_payload(placeholder, 0, str(encoded[0])+encoded[1])
self.assertEqual(x.message, data)
def test_addresses_request_message_conversion(self):
for c in self._conversions:
if isinstance(c, AddressesRequestConversion):
meta = self._di._community.get_meta_message(
ADDRESSES_REQUEST_MESSAGE_NAME)
sender_lan = Address.ipv4("0.0.0.1:1232")
sender_wan = Address.ipv4("23.12.32.123:1232")
endpoint_id = os.urandom(16)
wan_address = Address.ipv4("12.23.21.21:1234")
message = meta.impl(
authentication=(self._di._community.my_member, ),
distribution=(self._di._community.claim_global_time(), ),
destination=(Candidate(wan_address.addr(), True), ),
payload=(sender_lan, sender_wan, endpoint_id, wan_address))
encoded = c.encode_payload(message)
placeholder = c.Placeholder(None, meta, 0, encoded, False,
True)
_, x = c.decode_payload(placeholder, 0,
str(encoded[0]) + encoded[1])
self.assertEqual(x.sender_lan, sender_lan)
self.assertEqual(x.sender_wan, sender_wan)
self.assertEqual(x.endpoint_id, endpoint_id)
self.assertEqual(x.wan_address, wan_address)
def test_addresses_message_conversion(self):
for c in self._conversions:
if isinstance(c, AddressesConversion):
meta = self._di._community.get_meta_message(ADDRESSES_MESSAGE_NAME)
ids = [os.urandom(16) for _ in range(3)]
addresses = [Address.ipv4("0.0.0.1:1232"), Address.ipv6("[::0]:12145"), Address(port=32532)]
id_addresses = zip(ids, addresses, addresses)
message = meta.impl(authentication=(self._di._community.my_member,),
distribution=(self._di._community.claim_global_time(),),
destination=(Candidate(id_addresses[0][1].addr(), True),),
payload=(id_addresses,))
encoded = c.encode_payload(message)
placeholder = c.Placeholder(None, meta, 0, encoded, False, True)
_, x = c.decode_payload(placeholder, 0, str(encoded[0])+encoded[1])
self.assertEqual(x.id_addresses, id_addresses)
def test_file_hash_message_conversion(self):
for c in self._conversions:
if isinstance(c, FileHashConversion):
meta = self._di._community.get_meta_message(FILE_HASH_MESSAGE_NAME)
filename = "asdf.asdf"
dir_ = ""
roothash = "asdfj23j09f09sjfewef"
size = 103425
timestamp = 3232342.2342
address = Address.ipv4("0.0.0.0:1")
message = meta.impl(authentication=(self._di._community.my_member,), distribution=(self._di._community.claim_global_time(), self._di._community._file_hash_distribution.claim_sequence_number()),
payload=(filename, dir_, roothash, size, timestamp, [address]))
encoded = c.encode_payload(message)
placeholder = c.Placeholder(None, meta, 0, encoded, False, True)
_, x = c.decode_payload(placeholder, 0, str(encoded[0])+encoded[1])
self.assertEqual(x.filename, filename)
self.assertEqual(x.directories, dir_)
self.assertEqual(x.roothash, roothash)
self.assertEqual(x.size, size)
self.assertEqual(x.timestamp, timestamp)
self.assertEqual(x.addresses[0], address)
def test_uav_message_conversion(self):
for c in self._conversions:
if isinstance(c, APIMessageConversion):
meta = self._di._community.get_meta_message(API_MESSAGE_NAME)
data = "asjfdioewf"
addresses = [
Address.ipv4("0.0.0.1:1232"),
Address.ipv6("[::0]:12145"),
Address(port=32532)
]
message = meta.impl(
authentication=(self._di._community.my_member, ),
distribution=(self._di._community.claim_global_time(), ),
destination=(Candidate(addresses[0].addr(), True), ),
payload=(data, ))
encoded = c.encode_payload(message)
placeholder = c.Placeholder(None, meta, 0, encoded, False,
True)
_, x = c.decode_payload(placeholder, 0,
str(encoded[0]) + encoded[1])
self.assertEqual(x.message, data)
def test_addresses_message_conversion(self):
for c in self._conversions:
if isinstance(c, AddressesConversion):
meta = self._di._community.get_meta_message(
ADDRESSES_MESSAGE_NAME)
ids = [os.urandom(16) for _ in range(3)]
addresses = [
Address.ipv4("0.0.0.1:1232"),
Address.ipv6("[::0]:12145"),
Address(port=32532)
]
id_addresses = zip(ids, addresses, addresses)
message = meta.impl(
authentication=(self._di._community.my_member, ),
distribution=(self._di._community.claim_global_time(), ),
destination=(Candidate(id_addresses[0][1].addr(), True), ),
payload=(id_addresses, ))
encoded = c.encode_payload(message)
placeholder = c.Placeholder(None, meta, 0, encoded, False,
True)
_, x = c.decode_payload(placeholder, 0,
str(encoded[0]) + encoded[1])
self.assertEqual(x.id_addresses, id_addresses)
def test_no_port_ipv4(self):
addr = Address.ipv4("0.0.0.0")
self.assertEqual(addr.port, 0)
self.assertEqual(addr.ip, "0.0.0.0")
self.assertEqual(addr.family, AF_INET)
def test_no_port_ipv4(self):
addr = Address.ipv4("0.0.0.0")
self.assertEqual(addr.port, 0)
self.assertEqual(addr.ip, "0.0.0.0")
self.assertEqual(addr.family, AF_INET)
