Merge pull request #139 from seleniumbase/traffic-generation

Adding methods for testing analytics software

Author: mdmintz

help_docs/method_summary.md

@@ -125,6 +125,10 @@ self.pick_select_option_by_index(dropdown_selector, option,
 self.pick_select_option_by_value(dropdown_selector, option,
     dropdown_by=By.CSS_SELECTOR, timeout=settings.SMALL_TIMEOUT)
 
+self.generate_referral(start_page, destination_page)
+
+self.generate_traffic(start_page, destination_page, loops=1)
+
 ########
 
 self.wait_for_element_present(selector, by=By.CSS_SELECTOR,

seleniumbase/fixtures/base_case.py

@@ -872,6 +872,36 @@ def pick_select_option_by_value(self, dropdown_selector, option,
                                  dropdown_by=dropdown_by, option_by="value",
                                  timeout=timeout)
 
+    def generate_referral(self, start_page, destination_page):
+        """ This method opens the start_page, creates a referral link there,
+            and clicks on that link, which goes to the destination_page.
+            (This generates real traffic for testing analytics software.) """
+        if not page_utils.is_valid_url(destination_page):
+            raise Exception(
+                "Exception: destination_page [%s] is not a valid URL!"
+                % destination_page)
+        if start_page:
+            if not page_utils.is_valid_url(start_page):
+                raise Exception(
+                    "Exception: start_page [%s] is not a valid URL! "
+                    "(Use an empty string or None to start from current page.)"
+                    % start_page)
+            self.open(start_page)
+            time.sleep(0.03)
+        referral_link = ('''<a class='analytics referral test' href='%s'>'''
+                         '''Generate Free Referral!</a>''' % destination_page)
+        self.execute_script(
+            '''document.body.innerHTML = \"%s\"''' % referral_link)
+        time.sleep(0.05)
+        self.click("a.analytics")  # Clicks the generated button
+        time.sleep(0.03)
+
+    def generate_traffic(self, start_page, destination_page, loops=1):
+        """ Similar to generate_referral(), but can do multiple loops. """
+        for loop in range(loops):
+            self.generate_referral(start_page, destination_page)
+            time.sleep(0.05)
+
     ############
 
     def wait_for_element_present(self, selector, by=By.CSS_SELECTOR,

seleniumbase/fixtures/page_utils.py

@@ -1,6 +1,7 @@
 """
 This module contains useful utility methods.
 """
+import re
 import requests
 
 
@@ -42,6 +43,21 @@ def is_xpath_selector(selector):
     return False
 
 
+def is_valid_url(url):
+    regex = re.compile(
+        r'^(?:http)s?://'  # http:// or https://
+        r'(?:(?:[A-Z0-9](?:[A-Z0-9-]{0,61}[A-Z0-9])?\.)+'
+        r'(?:[A-Z]{2,6}\.?|[A-Z0-9-]{2,}\.?)|'  # domain...
+        r'localhost|'  # localhost...
+        r'\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3})'  # ...or ip
+        r'(?::\d+)?'  # optional port
+        r'(?:/?|[/?]\S+)$', re.IGNORECASE)
+    if regex.match(url):
+        return True
+    else:
+        return False
+
+
 def _download_file_to(file_url, destination_folder, new_file_name=None):
     if new_file_name:
         file_name = new_file_name

server_setup.py

@@ -8,7 +8,7 @@
 
 setup(
     name='seleniumbase',
-    version='1.5.2',
+    version='1.5.3',
     description='Web Automation & Testing Framework - http://seleniumbase.com',
     long_description='Web Automation and Testing Framework - seleniumbase.com',
     platforms='Mac * Windows * Linux * Docker',

setup.py

@@ -8,7 +8,7 @@
 
 setup(
     name='seleniumbase',
-    version='1.5.2',
+    version='1.5.3',
     description='Web Automation & Testing Framework - http://seleniumbase.com',
     long_description='Web Automation and Testing Framework - seleniumbase.com',
     platforms='Mac * Windows * Linux * Docker',