def upgrade_boot(self, host, user, pwd, version, addr):
     #点击系统设置菜单
     UpgradeControl.System_menu(self)
     #点击升级菜单
     UpgradeControl.Upgrade_menu(self)
     #升级方式切换为HTTP
     UpgradeControl.set_firmware_protocal(self, 'HTTP')
     #设置固件服务器,输入的地址为本机的ip地址
     UpgradeControl.set_FM_server(self, addr)
     #取消启动时检查
     UpgradeControl.set_on_boot(self)
     #点击保存
     UpgradeControl.save(self)
     UpgradeControl.apply(self)
     #点击重启
     UpgradeControl.reboot(self)
     #弹出的提示窗口中,点击确认
     UpgradeControl.notice_ok(self)
     # time.sleep(500)
     UpgradeBusiness.confirm_AP_upgrade_finish_after_reboot(
         self, host, user, pwd, version)
     print "reboot and upgrade ap successfully!"
     #ping AP的ip,ping通返回0
     result1 = UpgradeControl.get_ping(self, host)
     #登录AP后台取出版本号
     #登录AP后台取出版本号
     result2 = UpgradeControl.get_router_version(self, host, user, pwd)
     print result1, result2
     if (result1 == 0) and (version in result2):
         return True
     else:
         return False
 def check_reboot_reset_button(self):
     #点击系统设置菜单
     UpgradeControl.System_menu(self)
     #点击升级菜单
     UpgradeControl.Upgrade_menu(self)
     result1 = UpgradeControl.get_reboot(self)
     result2 = UpgradeControl.get_reset(self)
     return result1, result2
 def web_reboot_backup(self):
     #点击系统设置菜单
     UpgradeControl.System_menu(self)
     #点击升级菜单
     UpgradeControl.Upgrade_menu(self)
     #点击重启按钮
     UpgradeControl.reboot(self)
     #弹出的提示窗口中,点击确认
     UpgradeControl.notice_ok(self)
     print "set web reboot ap successfully!"
 def web_reboot(self, host):
     #点击系统设置菜单
     UpgradeControl.System_menu(self)
     #点击升级菜单
     UpgradeControl.Upgrade_menu(self)
     #点击重启按钮
     UpgradeControl.reboot(self)
     #弹出的提示窗口中,点击确认
     UpgradeControl.notice_ok(self)
     UpgradeBusiness.confirm_AP_reboot_finish(self, host)
     print "set web reboot ap successfully!"
 def check_reset_confirm(self):
     #点击系统设置菜单
     UpgradeControl.System_menu(self)
     #点击升级菜单
     UpgradeControl.Upgrade_menu(self)
     #点击重启
     UpgradeControl.factory_reset(self)
     #判断是否会弹出提示框,有则返回True,没有则返回False
     element = self.driver.find_element_by_xpath(
         ".//div[@class='modal-footer']//button[@class='btn btn-primary']")
     result = element.is_displayed()
     print result
     return result
 def check_ok_reboot_confirm(self, AP_IP):
     #点击系统设置菜单
     UpgradeControl.System_menu(self)
     #点击升级菜单
     UpgradeControl.Upgrade_menu(self)
     #点击重启
     UpgradeControl.reboot(self)
     #点击确定
     UpgradeControl.notice_ok(self)
     result1 = UpgradeControl.get_ping(self, AP_IP)
     time.sleep(200)
     result2 = UpgradeControl.get_ping(self, AP_IP)
     return result1, result2
 def web_factory_reset_backup(self, host):
     #首先等待3分钟再复位,防止slave ap和master ap还未建立tcp连接时复位导致的问题
     time.sleep(180)
     #点击系统设置菜单
     UpgradeControl.System_menu(self)
     #点击升级菜单
     UpgradeControl.Upgrade_menu(self)
     #点击恢复出厂
     UpgradeControl.factory_reset(self)
     #弹出的提示窗口中,点击确认
     UpgradeControl.notice_ok(self)
     UpgradeBusiness.confirm_AP_reboot_finish(self, host)
     print "click factory reset in webpage successfully!"
 def set_upgrade_mode(self, addr, mode):
     #点击系统设置菜单
     UpgradeControl.System_menu(self)
     time.sleep(10)
     #点击升级菜单
     UpgradeControl.Upgrade_menu(self)
     #选择升级方式
     UpgradeControl.set_firmware_protocal(self, mode)
     #设置固件服务器,输入的地址为本机的ip地址
     UpgradeControl.set_FM_server(self, addr)
     #点击保存
     UpgradeControl.save(self)
     UpgradeControl.apply(self)
     time.sleep(10)
     print "set upgrade mode is %s successfully!" % mode
 def check_cancel_reboot_confirm(self, AP_IP):
     #点击系统设置菜单
     UpgradeControl.System_menu(self)
     #点击升级菜单
     UpgradeControl.Upgrade_menu(self)
     #点击重启
     UpgradeControl.reboot(self)
     #点击取消
     UpgradeControl.notice_cancel(self)
     #判断是否会弹出提示框,有则返回True,没有则返回False
     element = self.driver.find_element_by_xpath(
         ".//div[@class='modal-footer']//button[@class='btn btn-primary']")
     result1 = element.is_displayed()
     print result1
     result2 = UpgradeControl.get_ping(self, AP_IP)
     return result1, result2