Use the new method for getting the BeautifulSoup soup

Author: mdmintz

seleniumbase/fixtures/base_case.py

@@ -196,9 +196,7 @@ def is_link_text_present(self, link_text):
         """ Returns True if the link text appears in the HTML of the page.
             The element doesn't need to be visible,
             such as elements hidden inside a dropdown selection. """
-        self.wait_for_ready_state_complete()
-        source = self.get_page_source()
-        soup = BeautifulSoup(source, "html.parser")
+        soup = self.get_beautiful_soup()
         html_links = soup.find_all('a')
         for html_link in html_links:
             if html_link.text.strip() == link_text.strip():
@@ -209,9 +207,7 @@ def get_link_attribute(self, link_text, attribute, hard_fail=True):
         """ Finds a link by link text and then returns the attribute's value.
             If the link text or attribute cannot be found, an exception will
             get raised if hard_fail is True (otherwise None is returned). """
-        self.wait_for_ready_state_complete()
-        source = self.get_page_source()
-        soup = BeautifulSoup(source, "html.parser")
+        soup = self.get_beautiful_soup()
         html_links = soup.find_all('a')
         for html_link in html_links:
             if html_link.text.strip() == link_text.strip():
@@ -340,8 +336,7 @@ def click_partial_link_text(self, partial_link_text,
                 element = self.wait_for_partial_link_text(partial_link_text)
                 element.click()
                 return
-            source = self.get_page_source()
-            soup = BeautifulSoup(source, "html.parser")
+            soup = self.get_beautiful_soup()
             html_links = soup.fetch('a')
             for html_link in html_links:
                 if partial_link_text in html_link.text:
@@ -701,8 +696,7 @@ def is_element_in_an_iframe(self, selector, by=By.CSS_SELECTOR):
         selector, by = self.__recalculate_selector(selector, by)
         if self.is_element_present(selector, by=by):
             return False
-        source = self.get_page_source()
-        soup = BeautifulSoup(source, "html.parser")
+        soup = self.get_beautiful_soup()
         iframe_list = soup.select('iframe')
         for iframe in iframe_list:
             iframe_identifier = None
@@ -727,8 +721,7 @@ def switch_to_frame_of_element(self, selector, by=By.CSS_SELECTOR):
         selector, by = self.__recalculate_selector(selector, by)
         if self.is_element_present(selector, by=by):
             return None
-        source = self.get_page_source()
-        soup = BeautifulSoup(source, "html.parser")
+        soup = self.get_beautiful_soup()
         iframe_list = soup.select('iframe')
         for iframe in iframe_list:
             iframe_identifier = None
@@ -2570,8 +2563,7 @@ def __get_href_from_link_text(self, link_text, hard_fail=True):
 
     def __click_dropdown_link_text(self, link_text, link_css):
         """ When a link may be hidden under a dropdown menu, use this. """
-        source = self.get_page_source()
-        soup = BeautifulSoup(source, "html.parser")
+        soup = self.get_beautiful_soup()
         drop_down_list = soup.select('[class*=dropdown]')
         for item in soup.select('[class*=HeaderMenu]'):
             drop_down_list.append(item)