diff --git a/readthedocs/projects/tests/test_domain_views.py b/readthedocs/projects/tests/test_domain_views.py index f5e0a7bbe23..fd3cd4faaa3 100644 --- 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, + )