From a32361146ef19d73d84793b143aa9858702f53ce Mon Sep 17 00:00:00 2001 From: zandercymatics <141044360+zandercymatics@users.noreply.github.com> Date: Wed, 8 Nov 2023 14:22:32 -0700 Subject: [PATCH] Update test_transition_domain_migrations.py --- .../test_transition_domain_migrations.py | 42 ++++++++++++------- 1 file changed, 26 insertions(+), 16 deletions(-) diff --git a/src/registrar/tests/test_transition_domain_migrations.py b/src/registrar/tests/test_transition_domain_migrations.py index fdc5af5e0..841f847d8 100644 --- a/src/registrar/tests/test_transition_domain_migrations.py +++ b/src/registrar/tests/test_transition_domain_migrations.py @@ -355,27 +355,37 @@ class TestMigrations(TestCase): # Test created Domain Information objects domain = Domain.objects.filter(name="anomaly.gov").get() anomaly_domain_infos = DomainInformation.objects.filter(domain=domain) + self.assertEqual(anomaly_domain_infos.count(), 1) + + # This domain should be pretty barebones. Something isnt + # parsing right if we get a lot of data. anomaly = anomaly_domain_infos.get() + self.assertEqual(anomaly.organization_name, "Flashdog") + self.assertEqual(anomaly.organization_type, None) + self.assertEqual(anomaly.federal_agency, None) + self.assertEqual(anomaly.federal_type, None) + + # Check for the "system" creator user + Users = User.objects.filter(username="System") + self.assertEqual(Users.count(), 1) + self.assertEqual(anomaly.creator, Users.get()) - self.assertEqual(anomaly.expiration_date, datetime.date(2023, 3, 9)) - """ - self.assertEqual( - anomaly.created_at, datetime.datetime(2023, 11, 8, 17, 23, 46, 764663, tzinfo=datetime.timezone.utc) - ) - """ - self.assertEqual(anomaly.name, "anomaly.gov") - self.assertEqual(anomaly.state, "ready") + domain = Domain.objects.filter(name="fakewebsite2.gov") + fakewebsite_domain_infos = DomainInformation.objects.filter(domain=domain) + self.assertEqual(fakewebsite_domain_infos.count(), 1) - testdomain_domains = Domain.objects.filter(name="fakewebsite2.gov") - self.assertEqual(testdomain_domains.count(), 1) + fakewebsite = fakewebsite_domain_infos.get() + self.assertEqual(fakewebsite.organization_name, "Flashdog") + self.assertEqual(fakewebsite.organization_type, None) + self.assertEqual(fakewebsite.federal_agency, None) + self.assertEqual(fakewebsite.federal_type, None) + + # Check for the "system" creator user + Users = User.objects.filter(username="System") + self.assertEqual(Users.count(), 1) + self.assertEqual(anomaly.creator, Users.get()) - testdomain = testdomain_domains.get() - - self.assertEqual(testdomain.expiration_date, datetime.date(2023, 9, 30)) - #self.assertEqual(testdomain.created_at, "test") - self.assertEqual(testdomain.name, "fakewebsite2.gov") - self.assertEqual(testdomain.state, "on hold") def test_transfer_transition_domains_to_domains(self): self.run_load_domains()