Update console scripts

Author: mdmintz

seleniumbase/console_scripts/sb_mkdir.py

@@ -318,6 +318,7 @@ def main():
 
     data = []
     data.append("from seleniumbase import BaseCase")
+    data.append("BaseCase.main(__name__, __file__)")
     data.append("")
     data.append("")
     data.append("class MyTestClass(BaseCase):")
@@ -353,6 +354,7 @@ def main():
 
     data = []
     data.append("from seleniumbase import BaseCase")
+    data.append("BaseCase.main(__name__, __file__)")
     data.append("")
     data.append("")
     data.append("class DemoSiteTests(BaseCase):")
@@ -448,6 +450,7 @@ def main():
     data = []
     data.append("from parameterized import parameterized")
     data.append("from seleniumbase import BaseCase")
+    data.append("BaseCase.main(__name__, __file__)")
     data.append("")
     data.append("")
     data.append("class GoogleTests(BaseCase):")
@@ -485,6 +488,7 @@ def main():
 
     data = []
     data.append("from seleniumbase import BaseCase")
+    data.append("BaseCase.main(__name__, __file__)")
     data.append("")
     data.append("")
     data.append("class BaseTestCase(BaseCase):")
@@ -550,6 +554,7 @@ def main():
 
     data = []
     data.append("from seleniumbase import BaseCase")
+    data.append("BaseCase.main(__name__, __file__)")
     data.append("")
     data.append("")
     data.append("class DataPage:")
@@ -647,6 +652,7 @@ def main():
     data.append('""" Classic Page Object Model with BaseCase inheritance """')
     data.append("")
     data.append("from seleniumbase import BaseCase")
+    data.append("BaseCase.main(__name__, __file__)")
     data.append("")
     data.append("")
     data.append("class LoginPage:")

seleniumbase/console_scripts/sb_mkfile.py

@@ -11,7 +11,7 @@
 
 Options:
     -b / --basic  (Basic boilerplate / single-line test)
-    -r / --rec  (add pdb++ breakpoint for Recorder Mode)
+    -r / --rec  (adds Pdb+ breakpoint for Recorder Mode)
 
 Language Options:
     --en / --English    |    --zh / --Chinese
@@ -45,7 +45,7 @@ def invalid_run_command(msg=None):
     exp += "           sbase mkfile new_test.py\n"
     exp += "  Options:\n"
     exp += "           -b / --basic  (Basic boilerplate / single-line test)\n"
-    exp += "           -r / --rec  (add pdb++ breakpoint for Recorder Mode)\n"
+    exp += "           -r / --rec  (adds Pdb+ breakpoint for Recorder Mode)\n"
     exp += "  Language Options:\n"
     exp += "           --en / --English    |    --zh / --Chinese\n"
     exp += "           --nl / --Dutch      |    --fr / --French\n"
@@ -220,9 +220,12 @@ def main():
         import_line = MD_F.get_import_line(language)
         parent_class = MD_F.get_lang_parent_class(language)
     class_line = "class %s(%s):" % (class_name, parent_class)
+    main_line = "BaseCase.main(__name__, __file__)"
 
     data = []
     data.append("%s" % import_line)
+    if not recorder:
+        data.append(main_line)
     data.append("")
     data.append("")
     data.append("%s" % class_line)
@@ -272,6 +275,10 @@ def main():
                         # Example: self.assert_true("Name" in self.get_title())
                         line = new_line
                         continue
+            if main_line in line:
+                new_main = "%s.main(__name__, __file__)" % parent_class
+                new_line = line.replace(main_line, new_main)
+                found_swap = True
             if found_swap:
                 if new_line.endswith("  # noqa"):  # Remove flake8 skip
                     new_line = new_line[0 : -len("  # noqa")]

seleniumbase/translate/translator.py

@@ -75,11 +75,12 @@ def invalid_run_command(msg=None):
 
 
 def sc_ranges():
-    # Get the ranges of special characters of Chinese, Japanese, and Korean.
+    # Get the ranges of special double-width characters.
     special_char_ranges = [
         {"from": ord("\u4e00"), "to": ord("\u9FFF")},
         {"from": ord("\u3040"), "to": ord("\u30ff")},
         {"from": ord("\uac00"), "to": ord("\ud7a3")},
+        {"from": ord("\uff01"), "to": ord("\uff60")},
     ]
     return special_char_ranges
 
@@ -220,6 +221,19 @@ def process_test_file(code_lines, new_lang):
                 seleniumbase_lines.append(line)
             continue
 
+        if (
+            ".main(__name__, __file__)" in line
+            and detected_lang
+            and new_lang
+            and (detected_lang != new_lang)
+        ):
+            old_basecase = MD_F.get_lang_parent_class(detected_lang)
+            new_basecase = MD_F.get_lang_parent_class(new_lang)
+            if old_basecase in line:
+                new_line = line.replace(old_basecase, new_basecase)
+                seleniumbase_lines.append(new_line)
+                continue
+
         if (
             "self." in line
             and "(" in line