appium python unittest_Python unittest appium

import unittest

from appium import webdriver

from appium.common.exceptions import NoSuchContextException

import desired_capabilities

class ContextSwitchingTests(unittest.TestCase):

def setUp(self):

desired_caps = desired_capabilities.get_desired_capabilities('selendroid-test-app.apk')

self.driver = webdriver.Remote('http://localhost:4723/wd/hub', desired_caps)

def test_contexts_list(self):

self._enter_webview()

contexts = self.driver.contexts

self.assertEqual(2, len(contexts))

def test_move_to_correct_context(self):

self._enter_webview()

self.assertEqual('WEBVIEW_io.selendroid.testapp', self.driver.current_context)

def test_actually_in_webview(self):

self._enter_webview()

self.driver.find_element_by_css_selector('input[type=submit]').click()

el = self.driver.find_element_by_xpath("//h1[contains(., 'This is my way')]")

self.assertIsNot(None, el)

def test_move_back_to_native_context(self):

self._enter_webview()

self.driver.switch_to.context(None)

self.assertEqual('NATIVE_APP', self.driver.current_context)

def test_set_invalid_context(self):

self.assertRaises(NoSuchContextException, self.driver.switch_to.context, 'invalid name')

def tearDown(self):

self.driver.quit()

def _enter_webview(self):

btn = self.driver.find_element_by_name('buttonStartWebviewCD')

btn.click()

self.driver.switch_to.context('WEBVIEW')

if __name__ == "__main__":

suite = unittest.TestLoader().loadTestsFromTestCase(ContextSwitchingTests)

unittest.TextTestRunner(verbosity=2).run(suite)