def testIPv6(self):
ipv6 = ipaddress.IPv6Address('2001:db8::1000')
address = jobs_pb2.NetworkAddress()
address.address_type = jobs_pb2.NetworkAddress.INET6
address.packed_bytes = ipv6.packed
out = io.StringIO()
pp = pretty.PrettyPrinter(out)
representer.network_address_pretty(address, pp, cycle=False)
expected = 'inet6 2001:db8::1000'
self.assertEqual(out.getvalue(), expected)
def testIPv4(self):
ipv4 = ipaddress.IPv4Address('42.0.255.32')
address = jobs_pb2.NetworkAddress()
address.address_type = jobs_pb2.NetworkAddress.INET
address.packed_bytes = ipv4.packed
out = io.StringIO()
pp = pretty.PrettyPrinter(out)
representer.network_address_pretty(address, pp, cycle=False)
expected = 'inet 42.0.255.32'
self.assertEqual(out.getvalue(), expected)