Complex example for creating new Open Browser keywords and using SL f… (robotframework#1434)

aaltat · web-flow · commit a1fd48495b10 · 2019-08-29T18:48:52.000+03:00
More acceptance tests
diff --git a/atest/acceptance/1-plugin/OpenBrowserExample.py b/atest/acceptance/1-plugin/OpenBrowserExample.py
@@ -0,0 +1,65 @@
+from robot.api import logger
+from selenium import webdriver
+
+from SeleniumLibrary.base import LibraryComponent, keyword
+from SeleniumLibrary.keywords import BrowserManagementKeywords
+from SeleniumLibrary.keywords.webdrivertools import WebDriverCreator
+
+
+class OpenBrowserExample(LibraryComponent):
+
+    def __init__(self, ctx):
+        LibraryComponent.__init__(self, ctx)
+        self._new_creator = NewWebDriverCreator(self.log_dir)
+
+    @keyword
+    def open_browser(self, url, browser='firefox', alias=None,
+                     remote_url=False, desired_capabilities=None,
+                     ff_profile_dir=None, options=None, service_log_path=None,
+                     extra_dictionary=None):
+        self._new_creator.extra_dictionary = extra_dictionary
+        browser_manager = BrowserManagementKeywords(self.ctx)
+        browser_manager._make_driver = self._make_driver
+        browser_manager.open_browser(url, browser=browser, alias=alias,
+                                     remote_url=remote_url, desired_capabilities=desired_capabilities,
+                                     ff_profile_dir=ff_profile_dir, options=options,
+                                     service_log_path=service_log_path)
+
+    def _make_driver(self, browser, desired_capabilities=None, profile_dir=None,
+                     remote=None, options=None, service_log_path=None):
+        driver = self._new_creator.create_driver(
+            browser=browser, desired_capabilities=desired_capabilities, remote_url=remote,
+            profile_dir=profile_dir, options=options, service_log_path=service_log_path)
+        driver.set_script_timeout(self.ctx.timeout)
+        driver.implicitly_wait(self.ctx.implicit_wait)
+        if self.ctx.speed:
+            self._monkey_patch_speed(driver)
+        return driver
+
+
+class NewWebDriverCreator(WebDriverCreator):
+
+    def create_driver(self, browser, desired_capabilities, remote_url,
+                      profile_dir=None, options=None, service_log_path=None):
+        self.browser_names['seleniumwire'] = 'seleniumwire'
+        browser = self._normalise_browser_name(browser)
+        creation_method = self._get_creator_method(browser)
+        desired_capabilities = self._parse_capabilities(desired_capabilities, browser)
+        service_log_path = self._get_log_path(service_log_path)
+        options = self.selenium_options.create(self.browser_names.get(browser), options)
+        if service_log_path:
+            logger.info('Browser driver log file created to: %s' % service_log_path)
+            self._create_directory(service_log_path)
+        if (creation_method == self.create_firefox
+                or creation_method == self.create_headless_firefox):
+            return creation_method(desired_capabilities, remote_url, profile_dir,
+                                   options=options, service_log_path=service_log_path)
+        if creation_method == self.create_seleniumwire:
+            return creation_method(desired_capabilities, remote_url, options=options,
+                                   service_log_path=service_log_path)
+        return creation_method(desired_capabilities, remote_url, options=options,
+                               service_log_path=service_log_path)
+
+    def create_seleniumwire(self, desired_capabilities, remote_url, options=None, service_log_path=None):
+        logger.info(self.extra_dictionary)
+        return webdriver.Chrome()
diff --git a/atest/acceptance/1-plugin/plugin_open_browser.robot b/atest/acceptance/1-plugin/plugin_open_browser.robot
@@ -0,0 +1,16 @@
+*** Settings ***
+Library    SeleniumLibrary    plugins=${CURDIR}/OpenBrowserExample.py
+
+*** Variables ***
+${SERVER}=         localhost:7000
+${ROOT}=           http://${SERVER}/html
+&{EXTRA DICTIONARY}    extra=dictionary    key=value
+
+
+*** Test Cases ***
+Open Browser Coplex Example
+    [Documentation]
+    ...    LOG 1:2 INFO GLOB: *'key': *'value'*
+    ...    LOG 1:2 INFO GLOB: *'extra': *'dictionary'*
+    Open Browser    ${ROOT}/forms/prefilled_email_form.html    seleniumwire    extra_dictionary=${EXTRA DICTIONARY}
+    [Teardown]    Close All Browsers
