26 lines
1.5 KiB
Python
26 lines
1.5 KiB
Python
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")
|