add unit tests

.github/workflows/ci.yaml

@@ -26,6 +26,8 @@ jobs:
         run: poetry install
       - name: Run lint
         run: poetry run ruff --output-format=github .
+      - name: Run unit tests
+        run: poetry run python manage.py test
 
   integration-tests:
     strategy:

atlas_provider_django/management/commands/atlas-provider-django.py

@@ -177,7 +177,7 @@ def handle(self, *args, **options):
         global current_dialect
         current_dialect = options.get("dialect", Dialect.sqlite)
         selected_apps = options.get("apps", None)
-        print(self.get_ddl(selected_apps))
+        return self.get_ddl(selected_apps)
 
     # Load migrations and get the sql statements describing the migrations.
     def get_ddl(self, selected_apps):

atlas_provider_django/settings.py

@@ -1 +1,9 @@
 INSTALLED_APPS = ["atlas_provider_django", "tests.app1", "tests.app2"]
+
+# if there are no databases defined, the tests tear down will fail
+DATABASES = {
+    "default": {
+        "ENGINE": "django.db.backends.sqlite3",
+        "NAME": "atlas_provider_django.db",
+    }
+}

tests/expected_all_apps.sql

@@ -0,0 +1,22 @@
+BEGIN;
+--
+-- Create model Musician
+--
+CREATE TABLE `app1_musician` (`id` bigint AUTO_INCREMENT NOT NULL PRIMARY KEY, `first_name` varchar(50) NOT NULL, `last_name` varchar(50) NOT NULL, `instrument` varchar(100) NOT NULL);
+--
+-- Create model Album
+--
+CREATE TABLE `app1_album` (`id` bigint AUTO_INCREMENT NOT NULL PRIMARY KEY, `name` varchar(100) NOT NULL, `release_date` date NOT NULL, `num_stars` integer NOT NULL, `artist_id` bigint NOT NULL);
+ALTER TABLE `app1_album` ADD CONSTRAINT `app1_album_artist_id_aed0987a_fk_app1_musician_id` FOREIGN KEY (`artist_id`) REFERENCES `app1_musician` (`id`);
+COMMIT;
+BEGIN;
+--
+-- Create model User
+--
+CREATE TABLE `app2_user` (`id` bigint AUTO_INCREMENT NOT NULL PRIMARY KEY, `first_name` varchar(50) NOT NULL, `last_name` varchar(50) NULL, `roll` varchar(100) NOT NULL);
+--
+-- Create model Blog
+--
+CREATE TABLE `app2_blog` (`id` bigint AUTO_INCREMENT NOT NULL PRIMARY KEY, `name` varchar(100) NOT NULL, `created_at` date NOT NULL, `num_stars` integer NOT NULL, `author_id` bigint NOT NULL);
+ALTER TABLE `app2_blog` ADD CONSTRAINT `app2_blog_author_id_1675e606_fk_app2_user_id` FOREIGN KEY (`author_id`) REFERENCES `app2_user` (`id`);
+COMMIT;

tests/expected_app1.sql

@@ -0,0 +1,11 @@
+BEGIN;
+--
+-- Create model Musician
+--
+CREATE TABLE `app1_musician` (`id` bigint AUTO_INCREMENT NOT NULL PRIMARY KEY, `first_name` varchar(50) NOT NULL, `last_name` varchar(50) NOT NULL, `instrument` varchar(100) NOT NULL);
+--
+-- Create model Album
+--
+CREATE TABLE `app1_album` (`id` bigint AUTO_INCREMENT NOT NULL PRIMARY KEY, `name` varchar(100) NOT NULL, `release_date` date NOT NULL, `num_stars` integer NOT NULL, `artist_id` bigint NOT NULL);
+ALTER TABLE `app1_album` ADD CONSTRAINT `app1_album_artist_id_aed0987a_fk_app1_musician_id` FOREIGN KEY (`artist_id`) REFERENCES `app1_musician` (`id`);
+COMMIT;

tests/tests_command.py

@@ -0,0 +1,18 @@
+from django.test import TestCase
+from django.core.management import call_command
+from io import StringIO
+
+
+# add unit test for the atlas_provider_django.py command
+class TestAtlasProviderDjango(TestCase):
+    def test_atlas_provider_django_all_apps(self):
+        out = StringIO()
+        call_command("atlas-provider-django", stdout=out)
+        with open("tests/expected_all_apps.sql", "r") as f:
+            self.assertEqual(out.getvalue(), f.read())
+
+    def test_atlas_provider_django_specific_app(self):
+        out = StringIO()
+        call_command("atlas-provider-django", "--app", "app1", stdout=out)
+        with open("tests/expected_app1.sql", "r") as f:
+            self.assertEqual(out.getvalue(), f.read())