| # Copyright 2016 The Chromium Authors. All rights reserved. |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| |
| """Class to control Netgear WNR1000V4 router.""" |
| |
| import logging |
| |
| import ap_spec |
| import dynamic_ap_configurator |
| |
| |
| class NetgearWNR1000V4APConfigurator( |
| dynamic_ap_configurator.DynamicAPConfigurator): |
| """Configurator for Netgear WNR1000V4 router.""" |
| |
| |
| def get_number_of_pages(self): |
| """Returns the number of available pages.""" |
| return 1 |
| |
| |
| def is_update_interval_supported(self): |
| """ |
| Returns True if setting the PSK refresh interval is supported. |
| |
| @return True is supported; False otherwise. |
| """ |
| return False |
| |
| |
| def get_supported_bands(self): |
| """Returns a list of dictionary with the supported channel per band.""" |
| return [{'band': ap_spec.BAND_2GHZ, |
| 'channels': [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11]}] |
| |
| |
| def get_supported_modes(self): |
| """Returns a list of dictionary with the supported band and modes.""" |
| return [{'band': ap_spec.BAND_2GHZ, |
| 'modes': [ap_spec.MODE_B, ap_spec.MODE_G, ap_spec.MODE_N]}] |
| |
| |
| def is_security_mode_supported(self, security_mode): |
| """Returns if a given security_type is supported. |
| |
| @param security_mode: one security modes defined in the APSpec. |
| |
| @return True if the security mode is supported; False otherwise. |
| """ |
| return security_mode in (ap_spec.SECURITY_TYPE_DISABLED, |
| ap_spec.SECURITY_TYPE_WPA2PSK, |
| ap_spec.SECURITY_TYPE_MIXED) |
| |
| |
| def is_spec_supported(self, spec): |
| """ |
| Returns if a given spec is supported by the router. |
| |
| @param spec: an instance of the |
| autotest_lib.server.cros.ap_configurators.APSpec class. |
| |
| @return: True if supported. False otherwise. |
| """ |
| if spec.security == ap_spec.SECURITY_TYPE_MIXED and spec.mode == ap_spec.MODE_N: |
| return False |
| return True |
| |
| |
| def _switch_to_frame1(self): |
| """Switch to iframe formframe.""" |
| self.driver.switch_to_default_content() |
| self.wait_for_object_by_xpath('//div[@id="formframe_div"]') |
| frame = self.driver.find_element_by_xpath('//iframe[@name="formframe"]') |
| self.driver.switch_to_frame(frame) |
| |
| |
| def _switch_to_frame2(self): |
| """Switch to wl2gsetting iframe within formframe.""" |
| self._switch_to_frame1() |
| xpath = '//div[@id="main"]//iframe[@id="2g_setting"]' |
| self.wait_for_object_by_xpath(xpath) |
| setframe = self.driver.find_element_by_xpath(xpath) |
| self.driver.switch_to_frame(setframe) |
| |
| |
| def _logout_from_previous_netgear_login(self): |
| """Logout if already logged into this router.""" |
| self.click_button_by_id('yes') |
| self.navigate_to_page(1) |
| |
| |
| def navigate_to_page(self, page_number): |
| """Navigates to the page corresponding to the given page number. |
| |
| This method performs the translation between a page number and a url to |
| load. This is used internally by apply_settings. |
| |
| @param page_number: page number of the page to load. |
| |
| """ |
| if self.object_by_id_exist('yes'): |
| self._logout_from_previous_netgear_login() |
| |
| self.get_url(self.admin_interface_url) |
| self.wait_for_object_by_xpath('//div[@id="wireless"]') |
| self.click_button_by_xpath('//div[@id="wireless"]') |
| self._switch_to_frame1() |
| self.wait_for_object_by_xpath('//input[@name="ssid"]') |
| |
| |
| def save_page(self, page_number): |
| """Saves the given page. |
| |
| @param page_number: Page number of the page to save. |
| """ |
| self.driver.switch_to_default_content() |
| self._switch_to_frame1() |
| xpath = '//input[@name="Apply"]' |
| self.wait_for_object_by_xpath(xpath) |
| self.click_button_by_xpath(xpath) |
| self.wait_for_object_by_xpath(xpath, wait_time=30) |
| |
| |
| def set_radio(self, enabled=True): |
| """Turns the radio on and off. |
| |
| @param enabled: True to turn on the radio; False otherwise. |
| """ |
| logging.debug('Cannot turn OFF radio!') |
| |
| |
| def set_ssid(self, ssid): |
| """Sets the SSID of the wireless network. |
| |
| @param ssid: name of the wireless network. |
| """ |
| self.add_item_to_command_list(self._set_ssid, (ssid,), 1, 900) |
| |
| |
| def _set_ssid(self, ssid): |
| """Sets the SSID of the wireless network. |
| |
| @param ssid: name of the wireless network. |
| """ |
| xpath = '//input[@maxlength="32" and @name="ssid"]' |
| self.set_content_of_text_field_by_xpath(ssid, xpath, abort_check=True) |
| self._ssid = ssid |
| |
| |
| def set_channel(self, channel): |
| """Sets the channel of the wireless network. |
| |
| @param channel: integer value of the channel. |
| """ |
| self.add_item_to_command_list(self._set_channel, (channel,), 1, 900) |
| |
| |
| def _set_channel(self, channel): |
| """Sets the channel of the wireless network. |
| |
| @param channel: integer value of the channel. |
| """ |
| position = self._get_channel_popup_position(channel) |
| channel_choices = [ '01', '02', '03', '04', '05', '06', |
| '07', '08', '09', '10', '11'] |
| xpath = '//select[@name="w_channel"]' |
| # Channel pop up is empty at this point, Save and Reload page |
| if self.number_of_items_in_popup_by_xpath(xpath) == 0: |
| self.save_page(1) |
| self._switch_to_frame2() |
| self.select_item_from_popup_by_xpath(channel_choices[position], |
| xpath) |
| |
| |
| def set_mode(self, mode, band=None): |
| """Sets the mode. |
| |
| @param mode: must be one of the modes listed in __init__(). |
| @param band: the band to select. |
| |
| """ |
| self.add_item_to_command_list(self._set_mode, (mode,), 1, 900) |
| |
| |
| def _set_mode(self, mode, band=None): |
| """Sets the mode. |
| |
| @param mode: must be one of the modes listed in __init__(). |
| @param band: the band to select. |
| |
| """ |
| if mode == ap_spec.MODE_G or mode == ap_spec.MODE_B: |
| mode = 'Up to 54 Mbps' |
| elif mode == ap_spec.MODE_N: |
| mode = 'Up to 300 Mbps' |
| else: |
| raise RuntimeError('Unsupported mode passed.') |
| xpath = '//select[@name="opmode"]' |
| self.select_item_from_popup_by_xpath(mode, xpath) |
| |
| |
| def set_band(self, band): |
| """Sets the band of the wireless network. |
| |
| @param band: Constant describing the band type. |
| """ |
| self.current_band = ap_spec.BAND_2GHZ |
| |
| |
| def set_security_disabled(self): |
| """Disables the security of the wireless network.""" |
| self.add_item_to_command_list(self._set_security_disabled, (), 1, 900) |
| |
| |
| def _set_security_disabled(self): |
| """Disables the security of the wireless network.""" |
| self.driver.switch_to_default_content() |
| self._switch_to_frame2() |
| xpath = '//input[@name="security_type" and @value="Disable"]' |
| self.click_button_by_xpath(xpath) |
| |
| |
| def set_security_wep(self, value, authentication): |
| """Enables WEP security for the wireless network. |
| |
| @param key_value: encryption key to use. |
| @param authentication: one of two supported WEP authentication types: |
| open or shared. |
| """ |
| logging.debug('WEP mode is not supported.') |
| |
| |
| def set_security_wpapsk(self, security, key, update_interval=None): |
| """Enables WPA2 and Mixed modes using a private security key for |
| the wireless network. |
| |
| @param security: Required security for AP configuration. |
| @param shared_key: shared encryption key to use. |
| @param update_interval: number of seconds to wait before updating. |
| """ |
| self.add_item_to_command_list(self._set_security_wpapsk, |
| (security, key,), |
| 1, 900) |
| |
| |
| def _set_security_wpapsk(self, security, key, update_interval=None): |
| """Enables WPA2 and Mixed modes using a private security key for |
| the wireless network. |
| |
| @param security: Required security for AP configuration. |
| @param shared_key: shared encryption key to use. |
| @param update_interval: number of seconds to wait before updating. |
| """ |
| self._switch_to_frame2() |
| |
| if security == ap_spec.SECURITY_TYPE_WPA2PSK: |
| xpath = '//input[@name="security_type" and @value="WPA2-PSK"]' |
| elif security == ap_spec.SECURITY_TYPE_MIXED: |
| xpath = '//input[@name="security_type" and @value="AUTO-PSK"]' |
| else: |
| raise RunTimeException('Invalid security mode %s sent' % security) |
| |
| self.click_button_by_xpath(xpath) |
| xpath = '//input[@name="passphrase"]' |
| self.set_content_of_text_field_by_xpath(key, xpath, abort_check=True) |
| |
| |
| def is_visibility_supported(self): |
| """ |
| Returns if AP supports setting the visibility (SSID broadcast). |
| |
| @return True if supported; False otherwise. |
| """ |
| return True |
| |
| |
| def set_visibility(self, visible=True): |
| """ |
| Sets SSID broadcast ON by clicking the checkbox. |
| |
| @param visible: True to enable SSID broadcast. False otherwise. |
| """ |
| self.add_item_to_command_list(self._set_visibility, (visible,), 1, 900) |
| |
| |
| def _set_visibility(self, visible=True): |
| """ |
| Sets SSID broadcast ON by clicking the checkbox. |
| |
| @param visible: True to enable SSID broadcast. False otherwise. |
| """ |
| self._switch_to_frame2() |
| xpath = '//input[@name="ssid_bc" and @type="checkbox"]' |
| check_box = self.wait_for_object_by_xpath(xpath) |
| value = check_box.is_selected() |
| if (visible and not value) or (not visible and value): |
| self.click_button_by_xpath(xpath) |
