U /ez@sdZddlmZddlZeeZddlmZm Z m Z m Z m Z m Z ddlmZddlZer~ddlZddlmZmZddlmZdZd d d d d ddZejddd ddddZejddd ddddZdS)z0 Define a Pytest plugin for a log file fixture ) annotationsN) TYPE_CHECKINGCallableIteratorListNoReturnSequence)warn)confignodes WebDriver)driverhas_no_console_errorspytest_report_collectionfinishz config.Configz py.path.localzSequence[nodes.Item]z List[str])r startdiritemsreturncCs6|dd}|dkrdnd}dd|d|dgS) z rchromeONZOFFzBokeh selenium tests using z& driver (no-console-error assertions: ) getoptionlower)r rr driver_nameZassertsrC/tmp/pip-unpacked-wheel-f5fndrjf/bokeh/_testing/plugins/selenium.pyr8srsession)ZscopezIterator[WebDriver]) pytestconfigrccs|dd}dddd}dddd}ddd d }|dkrJ|}n(|d krZ|}n|d krj|}ntd |d|V|dS)zG Select and configure a Selenium webdriver for integration tests. rrr )rcSsFddlm}ddlm}|}|d|d|d||dS)NrOptionsr --headlessz --no-sandbox--window-size=1920x1080options)Z!selenium.webdriver.chrome.optionsr!Z#selenium.webdriver.chrome.webdriverr add_argument)r!ZChromer%rrrrGs     zdriver..chromecSs<ddlm}ddlm}|}|d|d||dS)Nrr r r"r#r$)Z"selenium.webdriver.firefox.optionsr!Z$selenium.webdriver.firefox.webdriverr r&)r!ZFirefoxr%rrrfirefoxPs     zdriver..firefoxcSsddlm}|S)Nrr )Z#selenium.webdriver.safari.webdriverr )ZSafarirrrsafariXs zdriver..safarir'r(z(expected 'chrome', 'firefox' or 'safari' N)rr ValueErrorZimplicitly_waitquit)rrrr'r(rrrrr@s  rz&Callable[[WebDriver], bool | NoReturn]cCs<|d}|dkr(ddddd}nddddd}|S) a Provide a function to assert no browser console errors are present. Unfortunately logs are only accessibly with Chrome web driver, see e.g. https://github.com/mozilla/geckodriver/issues/284 For non-Chrome webdrivers this check always returns True. rrr zbool | NoReturn)rrcSs`|d}dd|D}dd|D}t|dkrPt|dkrLtd|dStd|S) NZbrowsercSsg|]}|ddkr|qS)levelZSEVEREget).0xrrr {sz7has_no_console_errors..func..cSsg|]}|ddkr|qS)typenetworkr-)r/lrrrr1|srzPThere were severe network errors (this may or may not have affected your test): TzConsole errors: )Zget_loglenr pytestZfail)rZlogsZ severe_errorsZnon_network_errorsrrrfuncys   z#has_no_console_errors..funccSsdS)NTr)rrrrr7sr)rrr7rrrrjs r)__doc__ __future__rlogging getLogger__name__logtypingrrrrrrwarningsr r6pyZ_pytestr r Z#selenium.webdriver.remote.webdriverr __all__rZfixturerrrrrrs        )