Update migration examples

Author: mdmintz

examples/migration/raw_selenium/flaky_messy_raw.py

@@ -1,4 +1,5 @@
 """Flaky Raw Selenium Example - (This test does NOT use SeleniumBase)"""
+import pytest
 import sys
 from selenium import webdriver
 from selenium.webdriver.common.by import By
@@ -63,3 +64,8 @@ def test_add_item_to_cart(self):
         self.driver.find_element(by_css, "#react-burger-menu-btn").click()
         self.driver.find_element(by_css, "a#logout_sidebar_link").click()
         self.driver.find_element(by_css, "input#login-button")
+
+
+# When run with "python" instead of "pytest"
+if __name__ == "__main__":
+    pytest.main([__file__])

examples/migration/raw_selenium/long_messy_raw.py

@@ -1,4 +1,5 @@
 """Long & Messy Raw Selenium Example - (This test does NOT use SeleniumBase)"""
+import pytest
 import sys
 from selenium import webdriver
 from selenium.webdriver.common.by import By
@@ -87,3 +88,8 @@ def test_add_item_to_cart(self):
         WebDriverWait(self.driver, 10).until(
             EC.visibility_of_element_located((by_css, "input#login-button"))
         )
+
+
+# When run with "python" instead of "pytest"
+if __name__ == "__main__":
+    pytest.main([__file__])

examples/migration/raw_selenium/messy_raw.py

@@ -1,4 +1,5 @@
 """Messy Raw Selenium Example - (This test does NOT use SeleniumBase)"""
+import pytest
 import sys
 from selenium import webdriver
 from selenium.webdriver.support import expected_conditions as EC
@@ -75,3 +76,8 @@ def test_add_item_to_cart(self):
         self.wait_for_element_clickable("#react-burger-menu-btn").click()
         self.wait_for_element_clickable("a#logout_sidebar_link").click()
         self.wait_for_element_visible("input#login-button")
+
+
+# When run with "python" instead of "pytest"
+if __name__ == "__main__":
+    pytest.main([__file__])

examples/migration/raw_selenium/refined_raw.py

@@ -1,4 +1,5 @@
 """Refined Raw Selenium Example - (This test does NOT use SeleniumBase)"""
+import pytest
 import sys
 from selenium import webdriver
 from selenium.webdriver.support import expected_conditions as EC
@@ -117,3 +118,8 @@ def test_add_item_to_cart(self):
         self.click("#react-burger-menu-btn")
         self.click("a#logout_sidebar_link")
         self.assert_element("input#login-button")
+
+
+# When run with "python" instead of "pytest"
+if __name__ == "__main__":
+    pytest.main([__file__])

examples/migration/raw_selenium/simple_sbase.py

@@ -1,4 +1,5 @@
 """Clean SeleniumBase Example - (Uses simple, reliable methods)"""
+import pytest
 from seleniumbase import BaseCase
 
 
@@ -18,3 +19,8 @@ def test_add_item_to_cart(self):
         self.click("#react-burger-menu-btn")
         self.click("a#logout_sidebar_link")
         self.assert_element("input#login-button")
+
+
+# When run with "python" instead of "pytest"
+if __name__ == "__main__":
+    pytest.main([__file__])