Add new website-traffic-generation methods

Author: mdmintz

help_docs/method_summary.md

@@ -191,10 +191,16 @@ 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.generate_referral_chain(pages)
+
+self.generate_traffic_chain(pages, loops=1)
+
 ########
 
 self.wait_for_element_present(selector, by=By.CSS_SELECTOR,

seleniumbase/fixtures/base_case.py

@@ -1824,6 +1824,8 @@ 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.
@@ -1843,7 +1845,7 @@ def generate_referral(self, start_page, destination_page):
         referral_link = ('''<a class='analytics referral test' href='%s' '''
                          '''style='font-family: Arial,sans-serif; '''
                          '''font-size: 30px; color: #18a2cd'>'''
-                         '''* Magic Link Button! *</a>''' % destination_page)
+                         '''Magic Link Button</a>''' % destination_page)
         self.execute_script(
             '''document.body.innerHTML = \"%s\"''' % referral_link)
         time.sleep(0.1)
@@ -1861,6 +1863,32 @@ def generate_traffic(self, start_page, destination_page, loops=1):
             self.generate_referral(start_page, destination_page)
             time.sleep(0.05)
 
+    def generate_referral_chain(self, pages):
+        """ Use this method to chain the action of creating button links on
+            one website page that will take you to the next page.
+            (When you want to create a referral to a website for traffic
+            generation without increasing the bounce rate, you'll want to visit
+            at least one additional page on that site with a button click.) """
+        if not type(pages) is tuple and not type(pages) is list:
+            raise Exception(
+                "Exception: Expecting a list of website pages for chaining!")
+        if len(pages) < 2:
+            raise Exception(
+                "Exception: At least two website pages required for chaining!")
+        for page in pages:
+            # Find out if any of the web pages are invalid before continuing
+            if not page_utils.is_valid_url(page):
+                raise Exception(
+                    "Exception: Website page {%s} is not a valid URL!" % page)
+        for page in pages:
+            self.generate_referral(None, page)
+
+    def generate_traffic_chain(self, pages, loops=1):
+        """ Similar to generate_referral_chain(), but for multiple loops. """
+        for loop in range(loops):
+            self.generate_referral_chain(pages)
+            time.sleep(0.05)
+
     ############
 
     def wait_for_element_present(self, selector, by=By.CSS_SELECTOR,