python_selenium_framework/base_test.py at master · Mattwhooo/python_selenium_framework · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
from selenium.webdriver.common.desired_capabilities import DesiredCapabilities
from selenium import webdriver
from selenium.webdriver.chrome.options import Options

class test(object):

    def tearDown(self):
        try:
            self.driver.quit()
            pass
        except:
            pass

    def getDriver(self, browser, local=False):
        if browser == "firefox":
            cap = DesiredCapabilities.FIREFOX
        if browser == "ie":
            cap = DesiredCapabilities.INTERNETEXPLORER
        if browser == "chrome":
            cap = DesiredCapabilities.CHROME
            options = Options()
            options.add_argument("--disable-extensions")
            options.add_argument("test-type")
        if browser == "mobile chrome":
            cap = {}
            cap['browserName'] = "Chrome"
            cap['platformName'] = "Android"
            cap['deviceName'] = "android"
        if browser == "android":
            cap = {}
            cap['browserName'] = "Browser"
            cap['platformName'] = "Android"
            cap['deviceName'] = "android"
        if local:
          return webdriver.Chrome()
        else:
          return webdriver.Remote(
                                  command_executor='http://10.238.242.50:4444/wd/hub',
                                  desired_capabilities=cap
                                 )