import pytest
from selenium.common.exceptions import WebDriverException

from wrapped_driver import WrappedDriver


def test_empty_chromedriver_path():
    """Assert error is raised if no chromedriver path is used"""
    with pytest.raises(WebDriverException):
        WrappedDriver(executable_path="", headless=True)


def test_no_chromedriver_path():
    """Assert error is raised if no chromedriver path is used"""
    with pytest.raises(TypeError):
        WrappedDriver(headless=True)

[(9, 'pytest.raises', 'pytest.raises', False, 'import pytest\n'), (10, 'wrapped_driver.WrappedDriver', 'WrappedDriver', False, 'from wrapped_driver import WrappedDriver\n'), (15, 'pytest.raises', 'pytest.raises', False, 'import pytest\n'), (16, 'wrapped_driver.WrappedDriver', 'WrappedDriver', False, 'from wrapped_driver import WrappedDriver\n')]