gecko-dev/browser/extensions/loop/test/functional/loopTestDriver.py

193 lines
7.3 KiB
Python

# taken from https://github.com/mozilla-b2g/gaia/blob/master/tests/python/gaia-ui-tests/gaiatest/gaia_test.py#L858
# XXX factor out into utility object for use by other tests
from marionette_driver.by import By
from marionette_driver.errors import NoSuchElementException, StaleElementException
# noinspection PyUnresolvedReferences
from marionette_driver import Wait
import re
import urlparse
import time
from serversetup import ROOMS_WEB_APP_URL_BASE
from config import USE_LOCAL_STANDALONE
class LoopTestDriver():
def setUp(self, marionette):
self.marionette = marionette
# XXX This should be unnecessary once bug 1254132 is fixed.
def set_context(self, context):
self.context = context
self.marionette.set_context(context)
def wait_for_element_displayed(self, by, locator, timeout=None):
Wait(self.marionette, timeout,
ignored_exceptions=[NoSuchElementException, StaleElementException])\
.until(lambda m: m.find_element(by, locator).is_displayed())
return self.marionette.find_element(by, locator)
def wait_for_subelement_displayed(self, parent, by, locator, timeout=None):
Wait(self.marionette, timeout,
ignored_exceptions=[NoSuchElementException, StaleElementException])\
.until(lambda m: parent.find_element(by, locator).is_displayed())
return parent.find_element(by, locator)
# XXX workaround for Marionette bug 1094246
def wait_for_element_exists(self, by, locator, timeout=None):
Wait(self.marionette, timeout,
ignored_exceptions=[NoSuchElementException, StaleElementException]) \
.until(lambda m: m.find_element(by, locator))
return self.marionette.find_element(by, locator)
def wait_for_element_enabled(self, element, timeout=10):
Wait(self.marionette, timeout) \
.until(lambda e: element.is_enabled(),
message="Timed out waiting for element to be enabled")
def wait_for_element_property_to_be_false(self, element, property, timeout=10):
# XXX We have to switch between get_attribute and get_property here as the
# content mode now required get_property for real properties of HTMLElements.
# However, in some places (e.g. switch_to_chatbox), we're still operating in
# a chrome mode. So we have to use get_attribute for these.
# Bug 1277065 should fix this for marionette, alternately this should go
# away when the e10s bug 1254132 is fixed.
def check_property(m):
if self.context == "content":
return not element.get_property(property)
return element.get_attribute(property) == "false"
Wait(self.marionette, timeout) \
.until(check_property,
message="Timeout out waiting for " + property + " to be false")
def open_panel(self):
self.set_context("chrome")
self.marionette.switch_to_frame()
button = self.marionette.find_element(By.ID, "loop-button")
# click the element
button.click()
def switch_to_panel(self):
self.set_context("chrome")
# switch to the frame
frame = self.marionette.find_element(By.ID, "loop-panel-iframe")
self.marionette.switch_to_frame(frame)
def switch_to_chatbox(self):
self.set_context("chrome")
self.marionette.switch_to_frame()
contentBox = "content"
if self.e10s_enabled:
contentBox = "remote-content"
# Added time lapse to allow for DOM to catch up
time.sleep(2)
# XXX should be using wait_for_element_displayed, but need to wait
# for Marionette bug 1094246 to be fixed.
chatbox = self.wait_for_element_exists(By.TAG_NAME, 'chatbox')
script = ("return document.getAnonymousElementByAttribute("
"arguments[0], 'anonid', '" + contentBox + "');")
frame = self.marionette.execute_script(script, [chatbox])
self.marionette.switch_to_frame(frame)
def local_start_a_conversation(self):
button = self.wait_for_element_displayed(By.CSS_SELECTOR, ".new-room-view .btn-info")
self.wait_for_element_enabled(button, 120)
button.click()
def local_close_share_panel(self):
copyLink = self.wait_for_element_displayed(By.CLASS_NAME, "btn-copy")
self.wait_for_element_enabled(copyLink, 120)
copyLink.click()
def switch_to_standalone(self):
self.set_context("content")
def load_homepage(self):
self.switch_to_standalone()
self.marionette.navigate("about:home")
def adjust_url(self, room_url):
if USE_LOCAL_STANDALONE != "1":
return room_url
# If we're not using the local standalone, then we need to adjust the room
# url that the server gives us to use the local standalone.
return re.sub("https?://.*/", ROOMS_WEB_APP_URL_BASE + "/", room_url)
def get_text_from_clipboard(self):
"""
This assumes we're already in the chrome context!
"""
script = ("""
var clipboard = Components.classes["@mozilla.org/widget/clipboard;1"]
.getService(Components.interfaces.nsIClipboard);
var trans = Components.classes["@mozilla.org/widget/transferable;1"]
.createInstance(Components.interfaces.nsITransferable);
trans.init(null);
trans.addDataFlavor("text/unicode");
clipboard.getData(trans, clipboard.kGlobalClipboard);
var str = new Object();
var strLength = new Object();
trans.getTransferData("text/unicode", str, strLength);
if (str)
str = str.value.QueryInterface(Components.interfaces.nsISupportsString);
return str ? str.data.substring(0, strLength.value / 2) : "";
""")
return self.marionette.execute_script(script, sandbox="system")
def local_get_and_verify_room_url(self):
self.switch_to_chatbox()
button = self.wait_for_element_displayed(By.CLASS_NAME, "btn-copy")
button.click()
# click the element
room_url = self.get_text_from_clipboard()
room_url = self.adjust_url(room_url)
self.assertIn(urlparse.urlparse(room_url).scheme, ['http', 'https'],
"room URL returned by server: '" + room_url +
"' has invalid scheme")
return room_url
# Assumes the standalone or the conversation window is selected first.
def check_video(self, selector):
video = self.wait_for_element_displayed(By.CSS_SELECTOR, selector, 30)
self.wait_for_element_property_to_be_false(video, "paused")
self.wait_for_element_property_to_be_false(video, "ended")
def local_check_room_self_video(self):
self.switch_to_chatbox()
# expect a video container on desktop side
media_container = self.wait_for_element_displayed(By.CLASS_NAME, "media-layout")
self.assertEqual(media_container.tag_name, "div", "expect a video container")
self.check_video(".local-video")
def standalone_load_and_join_room(self, url):
self.switch_to_standalone()
self.marionette.navigate(url)
# Join the room - the first time around, the tour will be displayed
# so we look for its close button.
join_button = self.wait_for_element_displayed(By.CLASS_NAME, "button-got-it")
self.wait_for_element_enabled(join_button, 120)
join_button.click()