From e24d274cfd0a6618ec8ba112a7bcad609e92607c Mon Sep 17 00:00:00 2001 From: benjamyn Date: Thu, 22 Oct 2020 02:05:19 -0400 Subject: [PATCH] Added basic test --- migratorapi/api/tests.py | 24 +++++++++++++++++++++++- 1 file changed, 23 insertions(+), 1 deletion(-) diff --git a/migratorapi/api/tests.py b/migratorapi/api/tests.py index 7ce503c..ec05475 100644 --- a/migratorapi/api/tests.py +++ b/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")