Add methods for injecting html into a web page

Author: mdmintz

help_docs/method_summary.md

@@ -126,6 +126,12 @@ self.select_option_by_value(dropdown_selector, option,
                             dropdown_by=By.CSS_SELECTOR,
                             timeout=None)
 
+self.load_html_string(html_string, new_page=True)
+
+self.load_html_file(html_file, new_page=True)
+
+self.open_html_file(html_file)
+
 self.execute_script(script)
 
 self.execute_async_script(script, timeout=None)
@@ -294,6 +300,8 @@ self.add_css_style(css_style)
 
 self.add_js_code_from_link(js_link)
 
+self.add_js_code(js_code)
+
 self.add_meta_tag(http_equiv=None, content=None)
 
 ########

seleniumbase/fixtures/base_case.py

@@ -1372,6 +1372,88 @@ def select_option_by_value(self, dropdown_selector, option,
                              dropdown_by=dropdown_by, option_by="value",
                              timeout=timeout)
 
+    def load_html_string(self, html_string, new_page=True):
+        """ Loads an HTML string into the web browser.
+            If new_page==True, the page will switch to: "data:text/html,"
+            If new_page==False, will load HTML into the current page. """
+
+        soup = self.get_beautiful_soup(html_string)
+        scripts = soup.findAll("script")
+        for script in scripts:
+            html_string = html_string.replace(str(script), "")
+        soup = self.get_beautiful_soup(html_string)
+
+        found_head = False
+        found_body = False
+        html_head = None
+        html_body = None
+        if soup.head and len(str(soup.head)) > 12:
+            found_head = True
+            html_head = str(soup.head)
+            html_head = re.escape(html_head)
+            html_head = self.__escape_quotes_if_needed(html_head)
+            html_head = html_head.replace('\\ ', ' ')
+        if soup.body and len(str(soup.body)) > 12:
+            found_body = True
+            html_body = str(soup.body)
+            html_body = re.escape(html_body)
+            html_body = self.__escape_quotes_if_needed(html_body)
+            html_body = html_body.replace('\\ ', ' ')
+        html_string = re.escape(html_string)
+        html_string = self.__escape_quotes_if_needed(html_string)
+        html_string = html_string.replace('\\ ', ' ')
+
+        if new_page:
+            self.open("data:text/html,")
+        inner_head = '''document.getElementsByTagName("head")[0].innerHTML'''
+        inner_body = '''document.getElementsByTagName("body")[0].innerHTML'''
+        if not found_body:
+            self.execute_script(
+                '''%s = \"%s\"''' % (inner_body, html_string))
+        elif found_body and not found_head:
+            self.execute_script(
+                '''%s = \"%s\"''' % (inner_body, html_body))
+        elif found_body and found_head:
+            self.execute_script(
+                '''%s = \"%s\"''' % (inner_head, html_head))
+            self.execute_script(
+                '''%s = \"%s\"''' % (inner_body, html_body))
+        else:
+            raise Exception("Logic Error!")
+
+        for script in scripts:
+            js_code = script.string
+            js_code_lines = js_code.split('\n')
+            new_lines = []
+            for line in js_code_lines:
+                line = line.strip()
+                new_lines.append(line)
+            js_code = '\n'.join(new_lines)
+            js_utils.add_js_code(self.driver, js_code)
+
+    def load_html_file(self, html_file, new_page=True):
+        """ Loads a local html file into the browser from a relative file path.
+            If new_page==True, the page will switch to: "data:text/html,"
+            If new_page==False, will load HTML into the current page.
+            Local images and other local src content WILL BE IGNORED. """
+        if len(html_file) < 6 or not html_file.endswith(".html"):
+            raise Exception('Expecting a ".html" file!')
+        abs_path = os.path.abspath('.')
+        file_path = abs_path + "/%s" % html_file
+        f = open(file_path, 'r')
+        html_string = f.read().strip()
+        f.close()
+        self.load_html_string(html_string, new_page)
+
+    def open_html_file(self, html_file):
+        """ Opens a local html file into the browser from a relative file path.
+            The URL displayed in the web browser will start with "file://". """
+        if len(html_file) < 6 or not html_file.endswith(".html"):
+            raise Exception('Expecting a ".html" file!')
+        abs_path = os.path.abspath('.')
+        file_path = abs_path + "/%s" % html_file
+        self.open("file://" + file_path)
+
     def execute_script(self, script):
         return self.driver.execute_script(script)
 
@@ -2610,6 +2692,9 @@ def add_css_style(self, css_style):
     def add_js_code_from_link(self, js_link):
         js_utils.add_js_code_from_link(self.driver, js_link)
 
+    def add_js_code(self, js_code):
+        js_utils.add_js_code(self.driver, js_code)
+
     def add_meta_tag(self, http_equiv=None, content=None):
         js_utils.add_meta_tag(
             self.driver, http_equiv=http_equiv, content=content)

seleniumbase/fixtures/js_utils.py

@@ -349,7 +349,20 @@ def add_js_code_from_link(driver, js_link):
         '''script.onload = function() { null };'''
         '''script.appendChild(document.createTextNode("%s"));'''
         '''body.appendChild(script);''')
-    js_code = js_code.replace('\n', '')
+    js_code = js_code.replace('\n', ' ')
+    js_code = escape_quotes_if_needed(js_code)
+    driver.execute_script(add_js_code_script % js_code)
+
+
+def add_js_code(driver, js_code):
+    add_js_code_script = (
+        '''var body = document.getElementsByTagName('body').item(0);'''
+        '''var script = document.createElement("script");'''
+        '''script.type = "text/javascript";'''
+        '''script.onload = function() { null };'''
+        '''script.appendChild(document.createTextNode("%s"));'''
+        '''body.appendChild(script);''')
+    js_code = js_code.replace('\n', ' ')
     js_code = escape_quotes_if_needed(js_code)
     driver.execute_script(add_js_code_script % js_code)