From adb888319cac7931729bb22a2a581b22709ad571 Mon Sep 17 00:00:00 2001 From: Santos Gallegos Date: Tue, 23 Aug 2022 16:08:07 -0500 Subject: [PATCH] Domains: test tasks with organizations --- readthedocs/domains/tests/test_tasks.py | 26 +++++++++++++++++++++++++ 1 file changed, 26 insertions(+) diff --git a/readthedocs/domains/tests/test_tasks.py b/readthedocs/domains/tests/test_tasks.py index b649b5d73de..ec51aff031f 100644 --- a/readthedocs/domains/tests/test_tasks.py +++ b/readthedocs/domains/tests/test_tasks.py @@ -7,6 +7,7 @@ from django_dynamic_fixture import get from readthedocs.domains.tasks import email_pending_custom_domains +from readthedocs.organizations.models import Organization, Team from readthedocs.projects.constants import ( SSL_STATUS_INVALID, SSL_STATUS_PENDING, @@ -119,3 +120,28 @@ def test_send_email_on_given_days(self, send_email): kwargs = send_email.call_args_list[2][1] self.assertEqual(kwargs["recipient"], self.another_user.email) self.assertIn(self.domain_invalid.domain, kwargs["context"]["content"]) + + +@override_settings(RTD_ALLOW_ORGANIZATIONS=True) +class TestTasksWithOrganizations(TestTasks): + def setUp(self): + super().setUp() + self.organization = get( + Organization, + owners=[self.user], + projects=[self.project, self.another_project], + ) + self.team_a = get( + Team, + organization=self.organization, + members=[self.user], + projects=[self.project, self.another_project], + access="admin", + ) + self.team_b = get( + Team, + organization=self.organization, + members=[self.user, self.another_user], + projects=[self.another_project], + access="admin", + )