Added basic test

migratorapi/api/tests.py

@@ -1,3 +1,25 @@
+import datetime
 from django.test import TestCase
+from django.utils import timezone
+# from django.urls import reverse
 
-# Create your tests here.
+from .models import Migration
+
+
+def create_migration(submit_time, domain, booked_time, original_server,
+                     new_server, username, notes, brand, ticket_id, migration_status,
+                     agent_booked, additional_domains, migration_type, is_urgent):
+
+    return Migration.objects.create(submit_time=submit_time, domain=domain, booked_time=booked_time, original_server=original_server,
+                                    new_server=new_server, username=username, notes=notes, brand=brand, ticket_id=ticket_id,
+                                    migration_status=migration_status, agent_booked=agent_booked, additional_domains=additional_domains,
+                                    migration_type=migration_type, is_urgent=is_urgent)
+
+
+class MigrationModelTests(TestCase):
+    def test_migration_is_booked(self):
+        migration = create_migration(submit_time=timezone.now(), domain="test.com", booked_time=timezone.now() + datetime.timedelta(1),
+                                     original_server='1.2.3.4', new_server="4.3.2.1", username="ben", notes="", brand="VentraIP", ticket_id="VIP-1234567", migration_status="Booked",
+                                     agent_booked="BL", additional_domains="", migration_type="cPanel", is_urgent=False)
+        print(migration.migration_status)
+        self.assertTrue(migration.migration_status == "Booked")