Update example tests

Author: mdmintz

examples/boilerplates/samples/google_objects.py

@@ -6,6 +6,7 @@
 class HomePage(object):
     dialog_box = '[role="dialog"] div'
     search_box = 'input[title="Search"]'
+    list_box = '[role="listbox"]'
     search_button = 'input[value="Google Search"]'
     feeling_lucky_button = '''input[value="I'm Feeling Lucky"]'''
 

examples/boilerplates/samples/google_test.py

@@ -10,15 +10,12 @@ class GoogleTests(BaseCase):
 
     def test_google_dot_com(self):
         self.open('https://google.com')
-        try:
-            # Remove the Privacy Checkup box if present.
-            self.assert_text('Privacy Checkup', HomePage.dialog_box, timeout=2)
-            self.click('link=NO, THANKS')
-        except Exception:
-            pass  # Google may have removed the Privacy Checkup. Continue test.
+        self.update_text(HomePage.search_box, 'github')
+        self.assert_element(HomePage.list_box)
         self.assert_element(HomePage.search_button)
         self.assert_element(HomePage.feeling_lucky_button)
-        self.update_text(HomePage.search_box, 'github\n')
+        self.click(HomePage.search_button)
         self.assert_text('github.com', ResultsPage.search_results)
         self.click_link_text('Images')
-        self.assert_element('img[alt="Image result for github"]')
+        source = self.get_page_source()
+        self.assertTrue("Image result for github" in source)

seleniumbase/console_scripts/sb_mkdir.py

@@ -217,26 +217,20 @@ def main():
         data.append("")
         data.append("    def test_google_dot_com(self):")
         data.append("        self.open('https://google.com')")
-        data.append("        try:")
-        data.append("            # Remove the Privacy Checkup box if present.")
-        data.append("            self.assert_text('Privacy Checkup', "
-                    "HomePage.dialog_box, timeout=3)")
-        data.append("            self.click('link=NO, THANKS')")
-        data.append("        except Exception:")
-        data.append("            # Google may have removed it. Continue test.")
-        data.append("            pass")
+        data.append(
+            "        self.update_text(HomePage.search_box, 'github')")
+        data.append("        self.assert_element(HomePage.list_box)")
         data.append("        self.assert_element(HomePage.search_button)")
         data.append(
             "        self.assert_element(HomePage.feeling_lucky_button)")
-        data.append(
-            "        self.update_text(HomePage.search_box, 'github\\n')")
+        data.append("        self.click(HomePage.search_button)")
         data.append(
             "        self.assert_text('github.com', "
             "ResultsPage.search_results)")
         data.append("        self.click_link_text('Images')")
+        data.append("        source = self.get_page_source()")
         data.append(
-            "        self.assert_element("
-            "'img[alt=\"Image result for github\"]')")
+            "        self.assertTrue('Image result for github' in source)")
         data.append("")
         file_path = "%s/%s" % (dir_name_3, "google_test.py")
         file = codecs.open(file_path, "w+", "utf-8")
@@ -247,6 +241,7 @@ def main():
         data.append("class HomePage(object):")
         data.append("    dialog_box = '[role=\"dialog\"] div'")
         data.append("    search_box = 'input[title=\"Search\"]'")
+        data.append("    list_box = '[role=\"listbox\"]'")
         data.append("    search_button = 'input[value=\"Google Search\"]'")
         data.append(
             "    feeling_lucky_button = "