Add support for open_notifications

Author: imurchie

appium/webdriver/mobilecommand.py

@@ -19,6 +19,7 @@ class MobileCommand(object):
     SWITCH_TO_CONTEXT = 'switchToContext'
     TOUCH_ACTION = 'touchAction'
     MULTI_ACTION = 'multiAction'
+    OPEN_NOTIFICATIONS = 'openNotifications'
 
     # Appium Commands
     GET_APP_STRINGS = 'getAppStrings'

appium/webdriver/webdriver.py

@@ -542,6 +542,12 @@ def shake(self):
         self.execute(Command.SHAKE)
         return self
 
+    def open_notifications(self):
+        """Open notification shade in Android (API Level 18 and above)
+        """
+        self.execute(Command.OPEN_NOTIFICATIONS, {})
+        return self
+
     def _addCommands(self):
         self.command_executor._commands[Command.CONTEXTS] = \
             ('GET', '/session/$sessionId/contexts')
@@ -594,6 +600,8 @@ def _addCommands(self):
             ('POST', '/session/$sessionId/appium/app/reset')
         self.command_executor._commands[Command.HIDE_KEYBOARD] = \
             ('POST', '/session/$sessionId/appium/device/hide_keyboard')
+        self.command_executor._commands[Command.OPEN_NOTIFICATIONS] = \
+            ('POST', '/session/$sessionId/appium/device/open_notifications')
 
 
 # monkeypatched method for WebElement

test/functional/android/appium_tests.py

@@ -17,6 +17,8 @@
 import json
 from time import sleep
 
+from selenium.common.exceptions import NoSuchElementException
+
 from appium import webdriver
 import desired_capabilities
 
@@ -53,47 +55,6 @@ def test_current_activity(self):
         activity = self.driver.current_activity
         self.assertEqual('.ApiDemos', activity)
 
-    def test_set_value(self):
-        el = self.driver.find_element_by_class_name('android.widget.ListView')
-        js_params = {'element': el.id, 'text': "Views"}
-        self.driver.execute_script("mobile: scrollTo", js_params)
-
-        el = self.driver.find_element_by_name('Views')
-        el.click()
-
-        el = self.driver.find_element_by_name('Auto Complete')
-        el.click()
-
-        el = self.driver.find_element_by_name('4. Contacts')
-        el.click()
-
-        el = self.driver.find_element_by_class_name('android.widget.EditText')
-        self.driver.set_value(el, 'Isaac')
-
-        text = el.get_attribute('text')
-        self.assertEqual('Isaac', text)
-
-    def test_element_set_value(self):
-        el = self.driver.find_element_by_class_name('android.widget.ListView')
-        js_params = {'element': el.id, 'text': "Views"}
-        self.driver.execute_script("mobile: scrollTo", js_params)
-
-        el = self.driver.find_element_by_name('Views')
-        el.click()
-
-        el = self.driver.find_element_by_name('Auto Complete')
-        el.click()
-
-        el = self.driver.find_element_by_name('4. Contacts')
-        el.click()
-        sleep(SLEEPY_TIME)
-
-        el = self.driver.find_element_by_class_name('android.widget.EditText')
-        el.set_value('Isaac')
-
-        text = el.get_attribute('text')
-        self.assertEqual('Isaac', text)
-
     def test_pull_file(self):
         data = self.driver.pull_file('data/local/tmp/strings.json')
         strings = json.loads(data.decode('base64', 'strict'))
@@ -157,6 +118,36 @@ def test_reset(self):
         el = self.driver.find_element_by_name('App')
         self.assertIsNotNone(el)
 
+    def test_open_notifications(self):
+        self.driver.find_element_by_android_uiautomator('new UiSelector().text("App")').click()
+        self.driver.find_element_by_android_uiautomator('new UiSelector().text("Notification")').click()
+        self.driver.find_element_by_android_uiautomator('new UiSelector().text("Status Bar")').click()
+
+        self.driver.find_element_by_android_uiautomator('new UiSelector().text(":-|")').click()
+
+        self.driver.open_notifications()
+        sleep(1)
+        self.assertRaises(NoSuchElementException, \
+            self.driver.find_element_by_android_uiautomator, 'new UiSelector().text(":-|")')
+
+        els = self.driver.find_elements_by_class_name('android.widget.TextView')
+        # sometimes numbers shift
+        title = False
+        body = False
+        for el in els:
+            text = el.text
+            if text == 'Mood ring':
+                title = True
+            elif text == 'I am ok':
+                body = True
+        self.assertTrue(title)
+        self.assertTrue(body)
+
+        self.driver.keyevent(4)
+        sleep(1)
+        self.driver.find_element_by_android_uiautomator('new UiSelector().text(":-|")')
+
 
 if __name__ == "__main__":
-    unittest.main()
+    suite = unittest.TestLoader().loadTestsFromTestCase(AppiumTests)
+    unittest.TextTestRunner(verbosity=2).run(suite)