def test_set_gw_info(self):
mac1 = "22:22:c6:d0:02:3c"
ipaddr1 = ipaddress.ip_address("10.1.1.11")
gwinfo_msg = GWInfo()
gwinfo_msg.ip.version = IPBlock.IPV4
gwinfo_msg.ip.address = ipaddr1.packed
gwinfo_msg.mac = mac1
self._stub.SetGatewayInfo(gwinfo_msg)
gw_info = self._stub.GetGatewayInfo(Void())
gw_ip_get = ipaddress.ip_address(gw_info.ip.address)
self.assertEqual(ipaddr1, gw_ip_get)
self.assertEqual(mac1, gw_info.mac)
def set_gw_ip_addressk_handler(client, args):
try:
ipaddr = ipaddress.ip_address(args.gwip)
except ValueError:
print("Error: invalid IP address format: %s" % args.gwip)
return
gwinfo_msg = GWInfo()
if ipaddr.version == 4:
gwinfo_msg.ip.version = IPBlock.IPV4
else:
print("Error: IP version %d is not supported yet" % ipaddr.version)
return
gwinfo_msg.ip.address = ipaddr.packed
gwinfo_msg.mac = ""
client.SetGatewayInfo(gwinfo_msg)
