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)