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
if args.gw_mac:
gwinfo_msg.mac = args.gw_mac
if args.gw_vlan:
gwinfo_msg.vlan = args.gw_vlan
else:
gwinfo_msg.vlan = ""
client.SetGatewayInfo(gwinfo_msg)
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
gwinfo_msg.vlan = ""
self._stub.SetGatewayInfo(gwinfo_msg)
gw_info_list = self._stub.ListGatewayInfo(Void())
for gw_info in gw_info_list.gw_list:
gw_ip_get = ipaddress.ip_address(gw_info.ip.address)
self.assertEqual(ipaddr1, gw_ip_get)
self.assertEqual(mac1, gw_info.mac)
