# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
from marionette_driver import By, Wait
from gaiatest.apps.base import Base
[docs]class Wallpaper(Base):
name = "Wallpaper"
_stock_wallpapers_locator = (By.CLASS_NAME, 'wallpaper')
[docs] def tap_wallpaper_by_index(self, index):
Wait(self.marionette).until(
lambda m: len(m.find_elements(*self._stock_wallpapers_locator)) >= (index + 1),
message='%d wallpaper(s) not present after timeout' % (index + 1))
self.tap_element_from_system_app(
self.marionette.find_elements(*self._stock_wallpapers_locator)[index])
self.wait_to_not_be_displayed()
self.apps.switch_to_displayed_app()