Add resetdb command and create test users and utility

README.md

@@ -71,9 +71,10 @@ all scripts should be run from the host.
 | `console`      | Run interactive shell inside application container            |
 | `lint`         | Lint source code                                              |
 | `server`       | Run Docker Compose services                                   |
-| `setup`        | Provision Vagrant VM and run `update`                         |
+| `setup`        | Provision Vagrant VM and run `update` and `resetdb`           |
 | `test`         | Run unit tests                                                |
 | `update`       | Build Docker images                                           |
+| `resetdb`      | Restore development database to defaults (with test data)     |
 
 ### Adding NPM Packages
 

scripts/resetdb

@@ -0,0 +1,27 @@
+#!/bin/bash
+
+set -e
+
+if [[ -n "${IOW_BOUNDARY_TOOL_DEBUG}" ]]; then
+    set -x
+fi
+
+function usage() {
+    echo -n \
+        "Usage: $(basename "$0")
+Reset database and repopulate with test utilities and users.
+"
+}
+
+function resetdb() {
+    ./scripts/manage resetdb
+}
+
+if [ "${BASH_SOURCE[0]}" = "${0}" ]; then
+    if [ "${1:-}" = "--help" ]; then
+        usage
+    else
+        resetdb
+    fi
+fi
+

scripts/setup

@@ -25,6 +25,7 @@ if [[ "${BASH_SOURCE[0]}" == "$0" ]]; then
 
         ./scripts/bootstrap
         ./scripts/update
+        ./scripts/resetdb
     else
         usage
     fi

src/django/api/management/commands/resetdb.py

@@ -0,0 +1,33 @@
+from django.core.management import call_command
+from django.core.management.base import BaseCommand
+
+from api.models import Role, Utility, User
+
+
+class Command(BaseCommand):
+    help = "Reset the DB schema, run migrations, and create test data."
+
+    def handle(self, *args, **options):
+        call_command("reset_schema", "--noinput")
+        call_command("migrate")
+
+        # Create test utility located in Raleigh.
+        test_utility = Utility(pwsid="123456789", name="Azavea Test Utility")
+        test_utility.save()
+
+        # Create test users.
+        User.objects.create_superuser(
+            email="[email protected]",
+            password="password",
+        )
+        User.objects.create_user(
+            email="[email protected]",
+            password="password",
+            role=Role.objects.get(description="VALIDATOR"),
+        )
+        User.objects.create_user(
+            email="[email protected]",
+            password="password",
+            role=Role.objects.get(description="CONTRIBUTOR"),
+            utility=test_utility,
+        )