-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathconftest.py
More file actions
45 lines (37 loc) · 1.91 KB
/
conftest.py
File metadata and controls
45 lines (37 loc) · 1.91 KB
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
41
42
43
44
import pytest
from selenium import webdriver
from selenium.webdriver.common.desired_capabilities import DesiredCapabilities
def pytest_addoption(parser):
parser.addoption("--selenium_host", action="store", default="http://localhost:4444/wd/hub", help="use webdriver for browser")
parser.addoption("--browser", action="store", default="Firefox", help="use webdriver for browser")
parser.addoption("--start_url", action="store", default="http://www.github.com/join", help="this is start url")
# def pytest_runtest_makereport(__multicall__, item):
# _driver = item.funcargs['test_driver']
# browser_type = _driver.capabilities['browserName']
# pytest_html = item.config.pluginmanager.getplugin('html')
# report = __multicall__.execute()
# extra = getattr(report, 'extra', [])
# if report.when == 'call':
# url = _driver.current_url
# extra.append(pytest_html.extras.url(url))
# screenshot = _driver.get_screenshot_as_base64()
# extra.append(pytest_html.extras.image(screenshot, ''))
# report.extra = extra
# return report
@pytest.fixture(scope="session")
def selenium_driver(request):
executor = request.config.getoption("--selenium_host")
if request.config.getoption("--browser") == 'Chrome':
driver = webdriver.Remote(command_executor=executor,
desired_capabilities=DesiredCapabilities.CHROME)
elif request.config.getoption("--browser") == 'Firefox':
driver = webdriver.Remote(command_executor=executor,
desired_capabilities=DesiredCapabilities.FIREFOX)
else:
driver = webdriver.Remote(command_executor=executor,
desired_capabilities=DesiredCapabilities.HTMLUNITWITHJS)
driver.maximize_window()
driver.get(request.config.getoption("--start_url"))
request.node._driver = driver
yield driver
driver.quit()