Bug 941749 - A test managing open windows and tabs with marionette.;r=automatedtester

testing/marionette/client/marionette/tests/unit/test_window_handles.py

@@ -0,0 +1,126 @@
+# 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_test import MarionetteTestCase
+from marionette import Keys
+
+
+class TestWindowHandles(MarionetteTestCase):
+
+    def test_new_tab_window_handles(self):
+        keys = [Keys.SHIFT]
+        if self.marionette.session_capabilities['platformName'] == 'DARWIN':
+            keys.append(Keys.META)
+        else:
+            keys.append(Keys.CONTROL)
+        keys.append('a')
+
+        # Put some history in the tab so this results in a fresh tab opening.
+        self.marionette.navigate("about:blank")
+        self.marionette.navigate("data:text/html, <div>Text</div>")
+        self.marionette.navigate("about:blank")
+
+        origin_win = self.marionette.current_window_handle
+
+        with self.marionette.using_context("chrome"):
+            main_win = self.marionette.find_element("id", "main-window")
+            main_win.send_keys(*keys)
+
+        self.wait_for_condition(lambda mn: len(mn.window_handles) == 2)
+        handles = self.marionette.window_handles
+        handles.remove(origin_win)
+        addons_page = handles.pop()
+        self.marionette.switch_to_window(addons_page)
+        self.assertEqual(self.marionette.get_url(), "about:addons")
+        self.marionette.close()
+
+        self.marionette.switch_to_window(origin_win)
+        self.assertEqual(self.marionette.get_url(), "about:blank")
+
+    def test_link_opened_tab_window_handles(self):
+        tab_testpage = self.marionette.absolute_url("newTab.html")
+        self.marionette.navigate(tab_testpage)
+        start_win = self.marionette.current_window_handle
+        link = self.marionette.find_element("id", "new-tab")
+        link.click()
+        self.wait_for_condition(lambda mn: len(mn.window_handles) == 2)
+
+        handles = self.marionette.window_handles
+        handles.remove(start_win)
+        dest_win = handles.pop()
+
+        self.marionette.switch_to_window(dest_win)
+        self.assertEqual(self.marionette.get_url(), "about:blank")
+        self.assertEqual(self.marionette.title, "")
+
+        self.marionette.switch_to_window(start_win)
+
+        self.assertIn('newTab.html', self.marionette.get_url())
+        self.assertEqual(self.marionette.title, "Marionette New Tab Link")
+
+        self.marionette.close()
+        self.marionette.switch_to_window(dest_win)
+        self.assertEqual(self.marionette.get_url(), "about:blank")
+
+    def test_tab_and_window_handles(self):
+        start_tab = self.marionette.current_window_handle
+        start_chrome_window = self.marionette.chrome_window_handle
+        tab_open_page = self.marionette.absolute_url("newTab.html")
+        window_open_page = self.marionette.absolute_url("test_windows.html")
+
+        # Open a new tab and switch to it.
+        self.marionette.navigate(tab_open_page)
+        link = self.marionette.find_element("id", "new-tab")
+        link.click()
+
+        self.wait_for_condition(lambda mn: len(mn.window_handles) == 2)
+        self.assertEqual(len(self.marionette.chrome_window_handles), 1)
+        self.assertEqual(self.marionette.chrome_window_handle, start_chrome_window)
+
+        handles = self.marionette.window_handles
+        handles.remove(start_tab)
+
+        new_tab = handles.pop()
+        self.marionette.switch_to_window(new_tab)
+        self.assertEqual(self.marionette.get_url(), "about:blank")
+
+        # Open a new window from the new tab.
+        self.marionette.navigate(window_open_page)
+
+        link = self.marionette.find_element("link text", "Open new window")
+        link.click()
+        self.wait_for_condition(lambda mn: len(mn.window_handles) == 3)
+
+        self.assertEqual(len(self.marionette.chrome_window_handles), 2)
+        self.assertEqual(self.marionette.chrome_window_handle, start_chrome_window)
+
+        # Find the new window and switch to it.
+        handles = self.marionette.window_handles
+        handles.remove(start_tab)
+        handles.remove(new_tab)
+        new_window = handles.pop()
+
+        self.marionette.switch_to_window(new_window)
+        results_page = self.marionette.absolute_url("resultPage.html")
+        self.assertEqual(self.marionette.get_url(), results_page)
+
+        self.assertEqual(len(self.marionette.chrome_window_handles), 2)
+        self.assertNotEqual(self.marionette.chrome_window_handle, start_chrome_window)
+
+        # Return to our original tab and close it.
+        self.marionette.switch_to_window(start_tab)
+        self.marionette.close()
+        self.assertEquals(len(self.marionette.window_handles), 2)
+
+        # Close the opened window and carry on in our new tab.
+        self.marionette.switch_to_window(new_window)
+        self.marionette.close()
+        self.assertEqual(len(self.marionette.window_handles), 1)
+
+        self.marionette.switch_to_window(new_tab)
+        self.assertEqual(self.marionette.get_url(), results_page)
+        self.marionette.navigate("about:blank")
+
+        self.assertEqual(len(self.marionette.chrome_window_handles), 1)
+        self.assertEqual(self.marionette.chrome_window_handle, start_chrome_window)

testing/marionette/client/marionette/tests/unit/unit-tests.ini

@@ -106,6 +106,8 @@ b2g = false
 b2g = false
 [test_window_position.py]
 b2g = false
+[test_window_handles.py]
+b2g = false
 
 [test_appcache.py]
 [test_screenshot.py]

testing/marionette/client/marionette/www/newTab.html

@@ -0,0 +1,13 @@
+<!-- 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/. -->
+
+<!DOCTYPE html>
+<html>
+<head>
+<title>Marionette New Tab Link</title>
+</head>
+<body>
+  <a href="about:blank" id="new-tab" target="_blank" class="linkClass">Click me!</a>
+</body>
+</html>