Update console scripts test generation

Author: mdmintz

README.md

@@ -451,6 +451,7 @@ ui_tests/
         ├── __init__.py
         ├── google_objects.py
         ├── google_test.py
+        ├── sb_swag_test.py
         └── swag_labs_test.py
 ```
 

seleniumbase/console_scripts/ReadMe.md

@@ -93,6 +93,7 @@ ui_tests/
         ├── __init__.py
         ├── google_objects.py
         ├── google_test.py
+        ├── sb_swag_test.py
         └── swag_labs_test.py
 ```
 

seleniumbase/console_scripts/sb_mkdir.py

@@ -595,6 +595,7 @@ def main():
     data.append("class LoginPage():")
     data.append("")
     data.append("    def login_to_swag_labs(self, sb, username):")
+    data.append('        sb.open("https://www.saucedemo.com/")')
     data.append('        sb.type("#user-name", username)')
     data.append('        sb.type("#password", "secret_sauce")')
     data.append("        sb.click('input[type=\"submit\"]')")
@@ -603,7 +604,6 @@ def main():
     data.append("class MyTests(BaseCase):")
     data.append("")
     data.append("    def test_swag_labs_login(self):")
-    data.append('        self.open("https://www.saucedemo.com/")')
     data.append(
         '        LoginPage().login_to_swag_labs(self, "standard_user")')
     data.append('        self.assert_element("#inventory_container")')
@@ -614,6 +614,28 @@ def main():
     file.writelines("\r\n".join(data))
     file.close()
 
+    data = []
+    data.append("class LoginPage():")
+    data.append("")
+    data.append("    def login_to_swag_labs(self, sb, username):")
+    data.append('        sb.open("https://www.saucedemo.com/")')
+    data.append('        sb.type("#user-name", username)')
+    data.append('        sb.type("#password", "secret_sauce")')
+    data.append("        sb.click('input[type=\"submit\"]')")
+    data.append("")
+    data.append("")
+    data.append("class MyTests():")
+    data.append("")
+    data.append("    def test_swag_labs_login(self, sb):")
+    data.append('        LoginPage().login_to_swag_labs(sb, "standard_user")')
+    data.append('        sb.assert_element("#inventory_container")')
+    data.append('        sb.assert_text("Products", "div.product_label")')
+    data.append("")
+    file_path = "%s/%s" % (dir_name_3, "sb_swag_test.py")
+    file = codecs.open(file_path, "w+", "utf-8")
+    file.writelines("\r\n".join(data))
+    file.close()
+
     success = (
         '\n' + c1 + '* Directory "' + dir_name + '" was created '
         'with config files and sample tests! *' + cr + '\n')