added test for piccolo user create

Author: dantownsend

piccolo/apps/user/commands/create.py

@@ -4,35 +4,53 @@
 from piccolo.apps.user.tables import BaseUser
 
 
-def create():
-    """
-    Create a new user.
-    """
+def get_username() -> str:
     default_username = getuser()
     username = input(f"Enter username ({default_username}):\n")
-    username = default_username if not username else username
+    return default_username if not username else username
 
-    email = input("Enter email:\n")
 
-    password = getpass("Enter password:\n")
-    confirmed_password = getpass("Confirm password:\n")
+def get_email() -> str:
+    return input("Enter email:\n")
 
-    if not password == confirmed_password:
-        print("Passwords don't match!")
-        sys.exit(1)
 
-    if len(password) < 4:
-        print("The password is too short")
-        sys.exit(1)
+def get_password() -> str:
+    return getpass("Enter password:\n")
+
 
+def get_confirmed_password() -> str:
+    return getpass("Confirm password:\n")
+
+
+def get_is_admin() -> bool:
     while True:
         admin = input("Admin user? Enter y or n:\n")
         if admin in ("y", "n"):
             break
         else:
             print("Unrecognised option")
 
-    is_admin = admin == "y"
+    return admin == "y"
+
+
+def create():
+    """
+    Create a new user.
+    """
+    username = get_username()
+    email = get_email()
+    password = get_password()
+    confirmed_password = get_confirmed_password()
+
+    if not password == confirmed_password:
+        print("Passwords don't match!")
+        sys.exit(1)
+
+    if len(password) < 4:
+        print("The password is too short")
+        sys.exit(1)
+
+    is_admin = get_is_admin()
 
     user = BaseUser(
         username=username,

tests/apps/user/__init__.py

@@ -0,0 +1,45 @@
+from unittest import TestCase
+from unittest.mock import patch
+
+from piccolo.apps.user.commands.create import create
+from piccolo.apps.user.tables import BaseUser
+
+
+class TestCreate(TestCase):
+    def setUp(self):
+        BaseUser.create_table(if_not_exists=True).run_sync()
+
+    def tearDown(self):
+        BaseUser.alter().drop_table().run_sync()
+
+    @patch(
+        "piccolo.apps.user.commands.create.get_username",
+        return_value="bob123",
+    )
+    @patch(
+        "piccolo.apps.user.commands.create.get_email",
+        return_value="[email protected]",
+    )
+    @patch(
+        "piccolo.apps.user.commands.create.get_password",
+        return_value="password123",
+    )
+    @patch(
+        "piccolo.apps.user.commands.create.get_confirmed_password",
+        return_value="password123",
+    )
+    @patch(
+        "piccolo.apps.user.commands.create.get_is_admin", return_value=True,
+    )
+    def test_create(self, *args, **kwargs):
+        create()
+
+        self.assertTrue(
+            BaseUser.exists()
+            .where(
+                (BaseUser.admin == True)
+                & (BaseUser.username == "bob123")
+                & (BaseUser.email == "[email protected]")  # noqa
+            )
+            .run_sync()
+        )