From b2b08d0e6cf55f05ea6e08d1c42b82f904cd7e06 Mon Sep 17 00:00:00 2001
From: zandercymatics <141044360+zandercymatics@users.noreply.github.com>
Date: Tue, 12 Dec 2023 12:16:53 -0700
Subject: [PATCH] Fix test cases

---
 .../tests/test_transition_domain_migrations.py    | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/src/registrar/tests/test_transition_domain_migrations.py b/src/registrar/tests/test_transition_domain_migrations.py
index 4efdee43e..eaa90bac3 100644
--- a/src/registrar/tests/test_transition_domain_migrations.py
+++ b/src/registrar/tests/test_transition_domain_migrations.py
@@ -29,14 +29,29 @@ class TestExtendExpirationDates(MockEppLib):
         Domain.objects.get_or_create(
             name="waterbutpurple.gov", state=Domain.State.READY, expiration_date=datetime.date(2023, 11, 15)
         )
+        TransitionDomain.objects.get_or_create(
+            username="testytester@mail.com",
+            domain_name="waterbutpurple.gov",
+            expiration_date=datetime.date(2023, 11, 15)
+        )
         # Create a domain with an invalid expiration date
         Domain.objects.get_or_create(
             name="fake.gov", state=Domain.State.READY, expiration_date=datetime.date(2022, 5, 25)
         )
+        TransitionDomain.objects.get_or_create(
+            username="themoonisactuallycheese@mail.com",
+            domain_name="fake.gov",
+            expiration_date=datetime.date(2022, 5, 25)
+        )
         # Create a domain with an invalid state
         Domain.objects.get_or_create(
             name="fakeneeded.gov", state=Domain.State.DNS_NEEDED, expiration_date=datetime.date(2023, 11, 15)
         )
+        TransitionDomain.objects.get_or_create(
+            username="fakeneeded@mail.com",
+            domain_name="fakeneeded.gov",
+            expiration_date=datetime.date(2023, 11, 15)
+        )
 
     def tearDown(self):
         """Deletes all DB objects related to migrations"""