Update examples

Author: mdmintz

examples/boilerplates/samples/sb_swag_test.py

@@ -1,16 +1,18 @@
 """Classic Page Object Model with the "sb" fixture."""
+from seleniumbase import BaseCase
+BaseCase.main(__name__, __file__)
 
 
 class LoginPage:
-    def login_to_swag_labs(self, sb, username):
+    def login_to_swag_labs(self, sb: BaseCase, username):
         sb.open("https://www.saucedemo.com")
         sb.type("#user-name", username)
         sb.type("#password", "secret_sauce")
         sb.click('input[type="submit"]')
 
 
 class MyTests:
-    def test_swag_labs_login(self, sb):
+    def test_swag_labs_login(self, sb: BaseCase):
         LoginPage().login_to_swag_labs(sb, "standard_user")
         sb.assert_element("div.inventory_list")
         sb.assert_element('div:contains("Sauce Labs Backpack")')

examples/boilerplates/samples/swag_labs_test.py

@@ -4,7 +4,7 @@
 
 
 class LoginPage:
-    def login_to_swag_labs(self, sb, username):
+    def login_to_swag_labs(self, sb: BaseCase, username):
         sb.open("https://www.saucedemo.com")
         sb.type("#user-name", username)
         sb.type("#password", "secret_sauce")

examples/hack_the_planet.py

@@ -301,11 +301,7 @@ def test_all_your_base_are_belong_to_us(self):
 
         self.open("https://git-scm.com/")
         self.set_text_content("span#tagline", aybabtu)
-        self.set_text_content("#nav-about h3", ayb)
-        self.set_text_content("#nav-documentation h3", abtu)
-        self.highlight("span#tagline", loops=8, scroll=False)
-        self.highlight("#nav-about h3", loops=5, scroll=False)
-        self.highlight("#nav-documentation h3", loops=6, scroll=False)
+        self.highlight("span#tagline", loops=10, scroll=False)
 
         self.open("https://pragprog.com/")
         self.set_text_content("header p", aybabtu)

examples/sb_fixture_tests.py

@@ -1,5 +1,9 @@
+from seleniumbase import BaseCase
+BaseCase.main(__name__, __file__)
+
+
 # "sb" pytest fixture test in a method with no class
-def test_sb_fixture_with_no_class(sb):
+def test_sb_fixture_with_no_class(sb: BaseCase):
     sb.open("seleniumbase.io/simple/login")
     sb.type("#username", "demo_user")
     sb.type("#password", "secret_pass")
@@ -13,7 +17,7 @@ def test_sb_fixture_with_no_class(sb):
 
 # "sb" pytest fixture test in a method inside a class
 class Test_SB_Fixture:
-    def test_sb_fixture_inside_class(self, sb):
+    def test_sb_fixture_inside_class(self, sb: BaseCase):
         sb.open("seleniumbase.io/simple/login")
         sb.type("#username", "demo_user")
         sb.type("#password", "secret_pass")

examples/test_override_sb_fixture.py

@@ -1,5 +1,7 @@
 """Overriding the "sb" fixture to override the driver."""
 import pytest
+from seleniumbase import BaseCase
+BaseCase.main(__name__, __file__)
 
 
 @pytest.fixture()
@@ -66,7 +68,7 @@ def tearDown(self):
             sb._needs_tearDown = False
 
 
-def test_override_fixture_no_class(sb):
+def test_override_fixture_no_class(sb: BaseCase):
     sb.open("https://seleniumbase.io/demo_page")
     sb.type("#myTextInput", "This is Automated")
     sb.set_value("input#mySlider", "100")
@@ -77,7 +79,7 @@ def test_override_fixture_no_class(sb):
 
 
 class TestOverride:
-    def test_override_fixture_inside_class(self, sb):
+    def test_override_fixture_inside_class(self, sb: BaseCase):
         sb.open("https://seleniumbase.io/demo_page")
         sb.type("#myTextInput", "This is Automated")
         sb.set_value("input#mySlider", "100")

examples/test_request_sb_fixture.py

@@ -1,6 +1,10 @@
+from seleniumbase import BaseCase
+BaseCase.main(__name__, __file__)
+
+
 # Use the pytest "request" fixture to get the "sb" fixture (no class)
 def test_request_sb_fixture(request):
-    sb = request.getfixturevalue("sb")
+    sb: BaseCase = request.getfixturevalue("sb")
     sb.open("https://seleniumbase.io/demo_page")
     sb.assert_text("SeleniumBase", "#myForm h2")
     sb.assert_element("input#myTextInput")
@@ -12,7 +16,7 @@ def test_request_sb_fixture(request):
 # Use the pytest "request" fixture to get the "sb" fixture (in class)
 class Test_Request_Fixture:
     def test_request_sb_fixture_in_class(self, request):
-        sb = request.getfixturevalue("sb")
+        sb: BaseCase = request.getfixturevalue("sb")
         sb.open("https://seleniumbase.io/demo_page")
         sb.assert_element("input#myTextInput")
         sb.type("#myTextarea", "Automated")

examples/test_sb_fixture.py

@@ -1,5 +1,9 @@
+from seleniumbase import BaseCase
+BaseCase.main(__name__, __file__)
+
+
 # "sb" pytest fixture test in a method with no class
-def test_sb_fixture_with_no_class(sb):
+def test_sb_fixture_with_no_class(sb: BaseCase):
     sb.open("seleniumbase.io/help_docs/install/")
     sb.type('input[aria-label="Search"]', "GUI Commander")
     sb.click('mark:contains("Commander")')
@@ -8,7 +12,7 @@ def test_sb_fixture_with_no_class(sb):
 
 # "sb" pytest fixture test in a method inside a class
 class Test_SB_Fixture:
-    def test_sb_fixture_inside_class(self, sb):
+    def test_sb_fixture_inside_class(self, sb: BaseCase):
         sb.open("seleniumbase.io/help_docs/install/")
         sb.type('input[aria-label="Search"]', "GUI Commander")
         sb.click('mark:contains("Commander")')

examples/test_usefixtures.py

@@ -1,4 +1,6 @@
 import pytest
+from seleniumbase import BaseCase
+BaseCase.main(__name__, __file__)
 
 
 @pytest.mark.usefixtures("sb")
@@ -7,7 +9,7 @@ def test_usefixtures_on_class(self):
         if not hasattr(self, "sb"):
             print("This test is for pytest only!")
             return
-        sb = self.sb
+        sb: BaseCase = self.sb
         sb.open("https://seleniumbase.io/realworld/login")
         sb.type("#username", "demo_user")
         sb.type("#password", "secret_pass")