| from autotest_lib.client.common_lib import error |
| from autotest_lib.client.common_lib import ui_utils |
| from autotest_lib.client.cros.input_playback import keyboard |
| import logging |
| import time |
| |
| |
| class UIPrinterHelper(object): |
| """Helper lib for ChromeOS UI automation.""" |
| |
| def __init__(self, ui_handler=None, chrome=None): |
| """ui_handler or chrome must be provided. ui_handler is an already |
| existing ui object.""" |
| if not ui_handler and chrome: |
| self.ui = ui_utils.UI_Handler() |
| self.ui.start_ui_root(chrome) |
| elif not chrome and ui_handler: |
| self.ui = ui_handler |
| else: |
| raise error.TestError( |
| "Either the chrome object or ui_handler must be provided.") |
| self._keyboard = None |
| |
| def print_to_custom_printer(self, printer_name, isPDF=False): |
| """Open the printer menu, select the printer given and click print.""" |
| self.open_printer_menu() |
| self.open_see_more_print_sub_menu() |
| self.select_printer_from_see_more_menu(printer_name) |
| if not self.check_print_window_open(): |
| raise error.TestError("Print not open after setting printer.") |
| self.click_print(isPDF) |
| if self._keyboard: |
| self._keyboard.close() |
| |
| def check_print_window_open(self): |
| """Check if the print window is still open.""" |
| start_time = time.time() |
| |
| # Giving up to 5 seconds for the window to load. |
| while time.time() - start_time < 5: |
| if (self.ui.item_present('Destination', role='inlineTextBox') and |
| self.ui.item_present('Cancel', role='button')) and not ( |
| self.ui.item_present('Loading preview')): |
| return True |
| |
| logging.info("Print Window was not open {}" |
| .format(self.ui.get_name_role_list())) |
| self.ui.screenshoter.take_ss() |
| return False |
| |
| def open_printer_menu(self, retry=0): |
| """Open the printer menu via the Chrome Dropdown.""" |
| if not self._is_chrome_menu_open(): |
| self.ui.doDefault_on_obj('Chrome') |
| self.ui.wait_for_ui_obj('/Print/', role='menuItem', isRegex=True) |
| self.ui.doDefault_on_obj('/Print/', isRegex=True) |
| if retry == 0: |
| try: |
| self.wait_for_print_ready() |
| except error.TestError: |
| self.open_printer_menu(retry=1) |
| else: |
| self.wait_for_print_ready() |
| |
| def wait_for_print_ready(self): |
| """Wait for the print page to fully load.""" |
| try: |
| self.ui.wait_for_ui_obj('Fit to width', role='button', timeout=30) |
| except: |
| logging.info("Missing {} in {}".format("Fit to width", self.ui.get_name_role_list())) |
| raise error.TestError("Fit to width no load") |
| self.ui.wait_for_ui_obj('Loading preview', remove=True) |
| |
| def open_see_more_print_sub_menu(self): |
| """For now must pivot there with the KB.""" |
| if not self._keyboard: |
| self._keyboard = keyboard.Keyboard() |
| |
| # Some tests have a notification which causes the chrome menu |
| # to remain open, and loose focus on the print page. |
| if self._is_chrome_menu_open(): |
| self._keyboard.press_key("escape") |
| self.verify_print_menu_open() |
| |
| self.ui.doDefault_on_obj(self._node_name_above_see_more_menu()) |
| self._keyboard.press_key("down") |
| self._keyboard.press_key('enter') |
| |
| def _node_name_above_see_more_menu(self): |
| """ |
| Return the name of the node above see more menu. |
| |
| Right now "See More" is not directly clickable. There is investigation |
| being done as for why, but still no answer. For now we will instead |
| get the name of the node above it on the menu list, to "hover" over, |
| then down arrow + enter to select "See more". |
| There are some empty items that return off the list query, so we have |
| to filter these out. Additionally if for some odd reason we do not |
| find an item directly above the "See more" option, we will default |
| to "Save as PDF". |
| |
| """ |
| temp_list = [] |
| menu_list = self.ui.get_name_role_list(role="menuListOption") |
| for i, item in enumerate(menu_list): |
| if item["name"]: |
| temp_list.append(item["name"]) |
| if "See more" in item["name"]: |
| if len(temp_list) < 2: |
| return "Save as PDF" # Default item above |
| return temp_list[-2] # First non empty item name prior. |
| |
| def _is_chrome_menu_open(self): |
| """Return True if the chrome dropdown menu is still open.""" |
| return self.ui.item_present("/Print/", role="menuItem", isRegex=True) |
| |
| def select_printer_from_see_more_menu(self, printer_name): |
| """Click a printer from the "see more" sub menu within print page.""" |
| if not self.is_see_more_menu_open(): |
| raise error.TestError( |
| "Cannot select printer from sub menu as its not open.") |
| # Easier to find via regex since the actual cell has the name twice. |
| if not self.is_str_regex(printer_name): |
| printer_name = self.make_str_regex(printer_name) |
| self.ui.wait_for_ui_obj(printer_name, isRegex=True, role='cell') |
| self.ui.doDefault_on_obj(printer_name, isRegex=True, role='cell') |
| self.ui.wait_for_ui_obj(printer_name, role='cell', remove=True, isRegex=True) |
| # Wait for the "Setting up " loading icon to finish |
| self.ui.wait_for_ui_obj('Setting up ', remove=True) |
| |
| def click_print(self, isPDF=False): |
| """Click the print button. Click save if PDF.""" |
| self.verify_print_menu_open() |
| self.ui.doDefault_on_obj('Save' if isPDF else 'Print', role='button') |
| if isPDF: |
| pass # TODO implement the save menu |
| |
| def is_see_more_menu_open(self): |
| """Return True if the print menu is open.""" |
| try: |
| self.ui.wait_for_ui_obj("Select a destination", role="dialog") |
| except error.TestError: |
| return False |
| return True |
| |
| def is_print_menu_open(self): |
| """Return True if the print menu is open.""" |
| return self.ui.item_present("Print", role="window") |
| |
| def verify_print_menu_open(self): |
| """Verify print menu is open. If not raise TestError.""" |
| if not self.is_print_menu_open(): |
| raise error.TestError( |
| "Cannot open See more print menu when print screen not open") |
| |
| def is_str_regex(self, var): |
| """Return True if the given var starts and ends with "/".""" |
| if len(var) < 1: |
| return False |
| if var[0] == "/" and var[-1] == "/": |
| return True |
| return False |
| |
| def make_str_regex(self, var): |
| """Return the provided var with a leading and ending "/".""" |
| return ("/" + var + "/") |
