Update the "sbase mkdir DIR" console script

Author: mdmintz

seleniumbase/console_scripts/sb_mkdir.py

@@ -230,16 +230,19 @@ def main():
     data.append("")
     data.append("    def test_basic(self):")
     data.append('        self.open("https://store.xkcd.com/search")')
-    data.append("        self.type('input[name=\"q\"]', \"xkcd book\\n\")")
+    data.append("        self.type('input[name=\"q\"]', \"xkcd book\")")
+    data.append("        self.click('input[value=\"Search\"]')")
     data.append('        self.assert_text("xkcd: volume 0", "h3")')
     data.append('        self.open("https://xkcd.com/353/")')
     data.append('        self.assert_title("xkcd: Python")')
     data.append("        self.assert_element('img[alt=\"Python\"]')")
     data.append("        self.click('a[rel=\"license\"]')")
     data.append('        self.assert_text("free to copy and reuse")')
     data.append('        self.go_back()')
-    data.append('        self.click("link=About")')
+    data.append('        self.click_link_text("About")')
     data.append('        self.assert_exact_text("xkcd.com", "h2")')
+    data.append('        self.click_link_text("geohashing")')
+    data.append('        self.assert_element("#comic img")')
     data.append("")
     file_path = "%s/%s" % (dir_name, "my_first_test.py")
     file = codecs.open(file_path, "w+", "utf-8")
@@ -357,13 +360,26 @@ def main():
     data.append("class BaseTestCase(BaseCase):")
     data.append("")
     data.append("    def setUp(self):")
-    data.append("        super(BaseTestCase, self).setUp()")
-    data.append("        # << Add custom code AFTER the super() line >>")
+    if sys.version_info[0] >= 3:
+        data.append("        super().setUp()")
+    else:
+        data.append("        super(BaseTestCase, self).setUp()")
+    data.append("        # <<< Run custom code AFTER the super() line >>>")
     data.append("")
     data.append("    def tearDown(self):")
     data.append("        self.save_teardown_screenshot()")
-    data.append("        # << Add custom code BEFORE the super() line >>")
-    data.append("        super(BaseTestCase, self).tearDown()")
+    data.append("        if self.has_exception():")
+    data.append("            # <<< Run custom code if the test failed. >>>")
+    data.append("            pass")
+    data.append("        else:")
+    data.append("            # <<< Run custom code if the test passed. >>>")
+    data.append("            pass")
+    data.append("        # (Wrap unreliable code in a try/except block.)")
+    data.append("        # <<< Run custom code BEFORE the super() line >>>")
+    if sys.version_info[0] >= 3:
+        data.append("        super().tearDown()")
+    else:
+        data.append("        super(BaseTestCase, self).tearDown()")
     data.append("")
     data.append("    def login(self):")
     data.append("        # <<< Placeholder. Add your code here. >>>")