added test for piccolo app new

Author: dantownsend

piccolo/apps/app/commands/new.py

@@ -16,14 +16,17 @@
 )
 
 
-def new_app(app_name: str):
+def new_app(app_name: str, root: str = ""):
     print(f"Creating {app_name} app ...")
-    if os.path.exists(app_name):
+
+    app_root = os.path.join(root, app_name)
+
+    if os.path.exists(app_root):
         print("Folder already exists - exiting.")
         sys.exit(1)
-    os.mkdir(app_name)
+    os.mkdir(app_root)
 
-    with open(os.path.join(app_name, "__init__.py"), "w"):
+    with open(os.path.join(app_root, "__init__.py"), "w"):
         pass
 
     templates: t.Dict[str, t.Any] = {
@@ -32,26 +35,30 @@ def new_app(app_name: str):
     }
 
     for filename, context in templates.items():
-        with open(os.path.join(app_name, filename), "w") as f:
+        with open(os.path.join(app_root, filename), "w") as f:
             template = JINJA_ENV.get_template(filename + ".jinja")
             file_contents = template.render(**context)
             file_contents = black.format_str(
                 file_contents, mode=black.FileMode(line_length=80)
             )
             f.write(file_contents)
 
-    migrations_folder_path = os.path.join(app_name, "piccolo_migrations")
+    migrations_folder_path = os.path.join(app_root, "piccolo_migrations")
     os.mkdir(migrations_folder_path)
 
     with open(os.path.join(migrations_folder_path, "__init__.py"), "w"):
         pass
 
 
-def new(app_name: str):
+def new(app_name: str, root: str = ""):
     """
     Creates a new Piccolo app.
 
     :param app_name:
         The name of the new app.
+    :param root:
+        Where to create the app e.g. /my/folder. By default it creates the
+        app in the current directory.
+
     """
-    new_app(app_name)
+    new_app(app_name=app_name, root=root)

tests/apps/app/__init__.py

@@ -0,0 +1,20 @@
+import os
+import shutil
+from unittest import TestCase
+
+from piccolo.apps.app.commands.new import new
+
+
+class TestNewApp(TestCase):
+    def test_new(self):
+        root = "/tmp"
+        app_name = "my_app"
+
+        app_path = os.path.join(root, app_name)
+
+        if os.path.exists(app_path):
+            shutil.rmtree(app_path)
+
+        new(app_name=app_name, root=root)
+
+        self.assertTrue(os.path.exists(app_path))