Fixed test for walkthrough pages on Chrome browser.

Added global variable BROWSER with default value firefox that updates its value from capabilities.yaml browserName parameter.
Added user agent on Chrome browser for Linux, Windows and Mac to test walkthrough pages, on each OS.

Author: Konstantinacc

bin/seleniumbender

@@ -97,14 +97,17 @@ elif [ "${walkthrough}" -eq 1 ]; then
     IDENTIFIER="walkthrough"
     RETVALS=()
     # Linux
+    export SELENIUM_USER_AGENT_CHROME='Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/47.0.2526.106 Safari/537.36'
     export SELENIUM_USER_AGENT='Mozilla/5.0 (X11; Ubuntu; Linux x86_64; rv:43.0) Gecko/20100101 Firefox/43.0 codebender-selenium'
     tox tests/walkthrough -- --url=${URL} --source=${SOURCE} --plugin
     RETVALS+=($?)
     # Windows
+    export SELENIUM_USER_AGENT_CHROME='Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/47.0.2526.106 Safari/537.36'
     export SELENIUM_USER_AGENT='Mozilla/5.0 (Windows NT 6.1; rv:43.0) Gecko/20100101 Firefox/43.0 codebender-selenium'
     tox tests/walkthrough -- --url=${URL} --source=${SOURCE} --plugin
     RETVALS+=($?)
     # MacOSX
+    export SELENIUM_USER_AGENT_CHROME='Mozilla/5.0 (Macintosh; Intel Mac OS X 10_11_1) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/47.0.2526.106 Safari/537.36'
     export SELENIUM_USER_AGENT='Mozilla/5.0 (Macintosh; Intel Mac OS X 10_11_1; rv:43.0) Gecko/20100101 Firefox/43.0 codebender-selenium'
     tox tests/walkthrough -- --url=${URL} --source=${SOURCE} --plugin
     RETVALS+=($?)

codebender_testing/config.py

@@ -95,6 +95,11 @@ def jsondump(data):
 DEFAULT_USER_AGENT = 'Mozilla/5.0 (X11; Ubuntu; Linux x86_64; rv:43.0) Gecko/20100101 Firefox/43.0 codebender-selenium'
 TESTS_USER_AGENT = os.getenv('SELENIUM_USER_AGENT', DEFAULT_USER_AGENT)
 
+DEFAULT_USER_AGENT_CHROME = 'Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/47.0.2526.106 Safari/537.36'
+TESTS_USER_AGENT_CHROME = os.getenv('SELENIUM_USER_AGENT_CHROME', DEFAULT_USER_AGENT_CHROME)
+
+BROWSER = "firefox"
+
 # Set up Selenium Webdrivers to be used for selenium tests.
 def _get_firefox_profile():
     """Returns the Firefox profile to be used for the FF webdriver.
@@ -140,12 +145,14 @@ def create_webdriver(command_executor, desired_capabilities):
     browser_profile_path = None
 
     if browser_name == "chrome":
+        BROWSER = "chrome"
         desired_capabilities = DesiredCapabilities.CHROME.copy()
         desired_capabilities.update(_capabilities)
         if desired_capabilities["version"] > CHROME_EXT_MAX_CHROME_VERSION:
             # Add new chrome extension to capabilities.
             options = chrome.options.Options()
             options.add_extension(os.path.join(_EXTENSIONS_DIR, _CHROME_APP_FNAME))
+            options.add_argument("--user-agent=" + TESTS_USER_AGENT_CHROME)
             desired_capabilities.update(options.to_capabilities())
             desired_capabilities.update(_capabilities)
         else:
@@ -154,6 +161,7 @@ def create_webdriver(command_executor, desired_capabilities):
                             % (CHROME_EXT_MAX_CHROME_VERSION, desired_capabilities["version"]))
 
     elif browser_name == "firefox":
+        BROWSER = "firefox"
         desired_capabilities = DesiredCapabilities.FIREFOX.copy()
         desired_capabilities.update(_capabilities)
         browser_profile = _get_firefox_profile()

tests/walkthrough/test_walkthrough.py

@@ -2,10 +2,11 @@
 import pytest
 
 from codebender_testing.config import TESTS_USER_AGENT
+from codebender_testing.config import TESTS_USER_AGENT_CHROME
+from codebender_testing.config import BROWSER
 from codebender_testing.utils import SeleniumTestCase
 from codebender_testing.utils import SELECT_BOARD_SCRIPT
 
-
 TEST_BOARD = 'Arduino Uno'
 
 
@@ -28,12 +29,20 @@ def test_page_2(self):
 
     def test_page_3(self):
         """Test page 3"""
-        if 'Linux' in TESTS_USER_AGENT:
-            self.get_element(By.CSS_SELECTOR, '#linux-directions .btn:nth-child(2)').click()
-        elif 'Windows' in TESTS_USER_AGENT:
-            self.get_element(By.CSS_SELECTOR, '#windows-directions .btn:nth-child(2)').click()
-        elif 'Mac' in TESTS_USER_AGENT:
-            self.get_element(By.CSS_SELECTOR, '#mac-directions .btn:nth-child(2)').click()
+        if BROWSER == "firefox":
+            if 'Linux' in TESTS_USER_AGENT:
+                self.get_element(By.CSS_SELECTOR, '#linux-directions .btn:nth-child(2)').click()
+            elif 'Windows' in TESTS_USER_AGENT:
+                self.get_element(By.CSS_SELECTOR, '#windows-directions .btn:nth-child(2)').click()
+            elif 'Mac' in TESTS_USER_AGENT:
+                self.get_element(By.CSS_SELECTOR, '#mac-directions .btn:nth-child(2)').click()
+        elif BROWSER == "chrome":
+            if 'Linux' in TESTS_USER_AGENT_CHROME:
+                self.get_element(By.CSS_SELECTOR, '#linux-directions .btn:nth-child(2)').click()
+            elif 'Windows' in TESTS_USER_AGENT_CHROME:
+                self.get_element(By.CSS_SELECTOR, '#windows-directions .btn:nth-child(2)').click()
+            elif 'Mac' in TESTS_USER_AGENT_CHROME:
+                self.get_element(By.CSS_SELECTOR, '#mac-directions .btn:nth-child(2)').click()
 
     def test_page_4(self):
         """Test page 4"""