class HM(object):
def __init__(self):
self.w = WebUI()
self.mode = None
self.configer = Configuration()
def admin_mode(self):
self.mode = 'admin'
def vhm_mode(self):
self.mode = 'vhm'
def is_admin_mode(self):
return self.mode == 'admin'
def is_vhm_mode(self):
return self.mode == 'vhm'
def _login(self, server, user, passwd):
self.server = server
self.user = user
self.passwd = passwd
self.w.info('trying to login %s as %s:%s' % (self.server, self.user, self.passwd))
self.w.get(self.server)
self.w.wait_until_element_displayed(LoginPage.get('login_btn'))
self.w.check_title(LoginPage.get('login_page_title'), 'check title after open')
self.w.input(LoginPage.get('username_txt'), self.user)
self.w.input(LoginPage.get('password_txt'), self.passwd)
self.w.info('login', True)
self.w.click(LoginPage.get('login_btn'))
def login(self, server, user, passwd):
self.admin_mode()
self._login(server, user, passwd)
self.w.wait_until_element_displayed(MainPage.get('monitor_lnk'))
self.w.register_cleanup(self.logout)
def login_by_vhm(self, server, vhm_user, vhm_passwd, hive_name=None, hive_mgt_passwd=None, quick_ssid_passwd=None):
self.vhm_mode()
self._login(server, vhm_user, vhm_passwd)
def __tmp(w):
return w.s.title != LoginPage.get('login_page_title')
self.w.wait_until(__tmp, msg='login page navigated')
if self.w.title == EULAPage.get('eula_page_title'):
self.w.wait_until_element_displayed(EULAPage.get('agree_btn'))
self.w.info('agree with EUAL', True)
self.w.click(EULAPage.get('agree_btn'))
self.w.wait_until_element_displayed(WelcomePage.get('continue_btn'))
self.w.check_title(WelcomePage.get('welcome_page_title'))
self.w.input(WelcomePage.get('hive_name_txt'), hive_name)
self.w.input(WelcomePage.get('hive_mgt_passwd_txt'), hive_mgt_passwd)
self.w.input(WelcomePage.get('hive_mgt_passwd_confirm_txt'), hive_mgt_passwd)
self.w.check(WelcomePage.get('enterprise_mode_radio'))
self.w.wait_until_element_displayed(WelcomePage.get('quick_ssid_passwd_txt'))
self.w.input(WelcomePage.get('quick_ssid_passwd_txt'), quick_ssid_passwd)
self.w.input(WelcomePage.get('quick_ssid_passwd_confirm_txt'), quick_ssid_passwd)
self.w.info('handle initial configuration', True)
self.w.click(WelcomePage.get('continue_btn'))
self.w.check_title(MainPage.get('main_page_title'), 'check title after login')
time.sleep(5)
self.w.register_cleanup(self.logout)
def to_hiveaps_table(self, type='all'):
self.w.click(MainPage.get('monitor_lnk'))
self.w.wait_until_title_present(HiveAPsPage.get('allAPs_page_title'))
if type == 'all':
self.w.click(HiveAPsPage.get('hive_aps_lnk'))
self.w.wait_until_title_present(HiveAPsPage.get('hiveAPs_page_title'))
self.w.click(HiveAPsPage.get('auto_refresh_off'))
selecter = self.w.get_selecter(HiveAPsPage.get('device_per_page_sel'))
if selecter.first_selected_option.text != '500':
selecter.select_by_value('500')
self.w.debug('after select page size', True)
elif type == 'rogue':
self.w.click(RogueAPsPage.get('rogue_aps_lnk'))
self.w.wait_until_title_present(RogueAPsPage.get('rogueAPs_page_title'))
selecter = self.w.get_selecter(RogueAPsPage.get('device_per_page_sel'))
if selecter.first_selected_option.text != '500':
selecter.select_by_value('500')
def select_ap(self, text='', by='default'):
if by == 'default':
try:
self.w.wait_until_element_displayed(APsTablePage.locate_checkbox(text))
except Exception, e:
self.w.warn('Cannot find: %s' % text, True)
checkbox_disabled = self.w.find_element(APsTablePage.locate_checkbox(text)).get_attribute('disabled')
if checkbox_disabled == 'true':
self.w.warn('not selectable AP: %s' % text, True)
else:
self.w.check(APsTablePage.locate_checkbox(text))
self.w.info('after select AP: %s' % text, True)
elif by == 'linktext':
try:
self.w.wait_until_element_displayed(APsTablePage.locate_checkbox_by_linktext(text))
except Exception, e:
self.w.warn('Cannot find: %s' % text, True)
