Update example tests

Author: mdmintz

examples/boilerplates/samples/file_parsing/parse_files.py

@@ -1,14 +1,10 @@
-"""
-Demonstration of parsing data from files.
-In this example, login information is pulled for tests.
-"""
-
+"""Example of parsing data from files."""
 from seleniumbase import BaseCase
 
 
 class ParseTestCase(BaseCase):
     def setUp(self):
-        super(ParseTestCase, self).setUp()
+        super().setUp()
 
     def get_login_credentials(self, user_type):
         # Example of parsing data from a file (Method 1)

examples/coffee_cart_tests.py

@@ -24,12 +24,18 @@ def test_coffee_promo_with_preview(self):
         self.click('div[data-test="Americano"]')
         self.click('div[data-test="Cafe_Latte"]')
         self.assert_exact_text("cart (3)", 'a[aria-label="Cart page"]')
-        self.assert_text("Get an extra cup of Mocha for $4.", "div.promo")
-        self.click("div.promo button.yes")
-        self.assert_exact_text("cart (4)", 'a[aria-label="Cart page"]')
+        promo = False
+        total_string = "Total: $33.00"
+        if self.is_element_visible("div.promo"):
+            self.assert_text("Get an extra cup of Mocha for $4.", "div.promo")
+            self.click("div.promo button.yes")
+            self.assert_exact_text("cart (4)", 'a[aria-label="Cart page"]')
+            promo = True
+            total_string = "Total: $37.00"
         self.hover('button[data-test="checkout"]')
-        self.assert_text("(Discounted) Mocha", "ul.cart-preview")
-        self.assert_exact_text("Total: $37.00", 'button[data-test="checkout"]')
+        if promo:
+            self.assert_text("(Discounted) Mocha", "ul.cart-preview")
+        self.assert_exact_text(total_string, 'button[data-test="checkout"]')
         self.click('button[data-test="checkout"]')
         self.type("input#name", "Selenium Coffee")
         self.type("input#email", "[email protected]")

examples/my_first_test.py

@@ -1,6 +1,11 @@
 """A complete end-to-end test for an e-commerce website."""
 from seleniumbase import BaseCase
 
+if __name__ == "__main__":  # If "python" called
+    from pytest import main
+    from sys import argv
+    main([*argv, "-s"])  # Run pytest, same args
+
 
 class MyTestClass(BaseCase):
     def test_swag_labs(self):
@@ -28,12 +33,6 @@ def test_swag_labs(self):
         self.assert_element("div#login_button_container")
 
 
-if __name__ == "__main__":  # If "python", run pytest
-    from pytest import main
-    from sys import argv
-    main([*argv, "-s"])  # Run pytest using same args
-
-
 #######################################################################
 #
 #    ****  NOTES / USEFUL INFO  ****

examples/swag_labs_user_tests.py

@@ -91,4 +91,4 @@ def tearDown(self):
             self.js_click_if_present("a#logout_sidebar_link")
         except Exception:
             pass
-        super(SwagLabsTests, self).tearDown()
+        super().tearDown()

examples/test_coffee_cart.py

@@ -1,6 +1,11 @@
 """Use SeleniumBase to test the Coffee Cart App."""
 from seleniumbase import BaseCase
 
+if __name__ == "__main__":  # If "python" called
+    from pytest import main
+    from sys import argv
+    main([*argv, "-s"])  # Run pytest, same args
+
 
 class CoffeeCartTest(BaseCase):
     def test_coffee_cart(self):
@@ -9,15 +14,9 @@ def test_coffee_cart(self):
         self.click('div[data-sb="Flat-White"]')
         self.click('div[data-sb="Cafe-Latte"]')
         self.click('a[aria-label="Cart page"]')
-        self.assert_exact_text("Total: $53.00", 'button.pay')
-        self.click('button.pay')
+        self.assert_exact_text("Total: $53.00", "button.pay")
+        self.click("button.pay")
         self.type("input#name", "Selenium Coffee")
         self.type("input#email", "[email protected]")
         self.click("button#submit-payment")
         self.assert_text("Thanks for your purchase.", "#app .success")
-
-
-if __name__ == "__main__":  # If "python", run pytest
-    from pytest import main
-    from sys import argv
-    main([*argv, "-s"])  # Run pytest using same args

examples/test_mfa_login.py

@@ -1,5 +1,10 @@
 from seleniumbase import BaseCase
 
+if __name__ == "__main__":  # If "python" called
+    from pytest import main
+    from sys import argv
+    main([*argv, "-s"])  # Run pytest, same args
+
 
 class TestMFALogin(BaseCase):
     def test_mfa_login(self):
@@ -14,9 +19,3 @@ def test_mfa_login(self):
         self.click_link("Sign out")  # Link must be "a" tag. Not "button".
         self.assert_element('a:contains("Sign in")')
         self.assert_exact_text("You have been signed out!", "#top_message")
-
-
-if __name__ == "__main__":  # If "python", run pytest
-    from pytest import main
-    from sys import argv
-    main([*argv, "-s"])  # Run pytest using same args

examples/test_override_sb_fixture.py

@@ -20,14 +20,14 @@ def get_new_driver(self, *args, **kwargs):
             return webdriver.Chrome(options=options)
 
         def setUp(self):
-            super(BaseClass, self).setUp()
+            super().setUp()
 
         def base_method(self):
             pass
 
         def tearDown(self):
             self.save_teardown_screenshot()
-            super(BaseClass, self).tearDown()
+            super().tearDown()
 
     sb = BaseClass("base_method")
     sb.setUpClass()

examples/test_swag_labs.py

@@ -1,5 +1,10 @@
 from seleniumbase import BaseCase
 
+if __name__ == "__main__":  # If "python" called
+    from pytest import main
+    from sys import argv
+    main([*argv, "-s"])  # Run pytest, same args
+
 
 class SwagLabsTests(BaseCase):
     def login_to_swag_labs(self, username="standard_user"):
@@ -77,10 +82,4 @@ def tearDown(self):
             self.js_click_if_present("a#logout_sidebar_link")
         except Exception:
             pass
-        super(SwagLabsTests, self).tearDown()
-
-
-if __name__ == "__main__":  # If "python", run pytest
-    from pytest import main
-    from sys import argv
-    main([*argv, "-s"])  # Run pytest using same args
+        super().tearDown()

examples/verify_undetected.py

@@ -5,9 +5,9 @@
 """
 from seleniumbase import BaseCase
 
-if __name__ == "__main__":  # If "python", use "pytest"
+if __name__ == "__main__":
     from pytest import main
-    main([__file__, "--uc"])
+    main([__file__, "--uc", "-s"])
 
 
 class UndetectedTest(BaseCase):