import datetime
from django.test import TestCase
from django.utils import timezone
# from django.urls import reverse

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")