Merge pull request #9233 from readthedocs/fix-tests-com

readthedocs · May 16, 2022 · 3687c37 · 3687c37
2 parents 30f891f + 5117846
commit 3687c37
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/readthedocs/projects/tests/test_domain_views.py b/readthedocs/projects/tests/test_domain_views.py
@@ -5,6 +5,7 @@
 
 from readthedocs.organizations.models import Organization
 from readthedocs.projects.models import Domain, Project
+from readthedocs.subscriptions.models import Plan, PlanFeature, Subscription
 
 
 @override_settings(RTD_ALLOW_ORGANIZATIONS=False)
@@ -101,3 +102,17 @@ def setUp(self):
         self.org = get(
             Organization, owners=[self.user], projects=[self.project, self.subproject]
         )
+        self.plan = get(
+            Plan,
+            published=True,
+        )
+        self.subscription = get(
+            Subscription,
+            plan=self.plan,
+            organization=self.org,
+        )
+        self.feature = get(
+            PlanFeature,
+            plan=self.plan,
+            feature_type=PlanFeature.TYPE_CNAME,
+        )