def test_buffer_full(self, mock_create_demo_org):
for i in range(ORG_BUFFER_SIZE):
self.create_demo_org()
mock_create_demo_org.side_effect = self.create_demo_org
with self.tasks():
build_up_org_buffer()
assert mock_create_demo_org.call_count == 0
def test_add_two_fill_buffer(self, mock_create_demo_org):
for i in range(ORG_BUFFER_SIZE - 2):
self.create_demo_org()
mock_create_demo_org.side_effect = self.create_demo_org
with self.tasks():
build_up_org_buffer()
assert (
ORG_BUFFER_SIZE == DemoOrganization.objects.filter(status=DemoOrgStatus.PENDING).count()
)
assert mock_create_demo_org.call_count == 2
def test_add_two_fill_buffer(self, mock_create_demo_org):
for i in range(ORG_BUFFER_SIZE - 2):
# pending an initializing orgs both count
status = DemoOrgStatus.INITIALIZING if i % 1 == 0 else DemoOrgStatus.PENDING
self.create_demo_org(status=status)
mock_create_demo_org.side_effect = self.create_demo_org
with self.tasks():
build_up_org_buffer()
assert (ORG_BUFFER_SIZE == DemoOrganization.objects.filter(
status__in=[DemoOrgStatus.PENDING, DemoOrgStatus.INITIALIZING
]).count())
assert mock_create_demo_org.call_count == 2
def test_add_one_fill_buffer(self, mock_create_demo_org):
for i in range(ORG_BUFFER_SIZE - 1):
self.create_demo_org()
# active orgs shoudn't count
self.create_demo_org(status=DemoOrgStatus.ACTIVE)
mock_create_demo_org.side_effect = self.create_demo_org
with self.tasks():
build_up_org_buffer()
assert (
ORG_BUFFER_SIZE == DemoOrganization.objects.filter(status=DemoOrgStatus.PENDING).count()
)
mock_create_demo_org.assert_called_once_with()
def test_add_one_fill_buffer(self, mock_create_demo_org):
for i in range(ORG_BUFFER_SIZE - 1):
# pending an initializing orgs both count
status = DemoOrgStatus.INITIALIZING if i % 1 == 0 else DemoOrgStatus.PENDING
self.create_demo_org(status=status)
# active orgs shoudn't count
self.create_demo_org(status=DemoOrgStatus.ACTIVE)
mock_create_demo_org.side_effect = self.create_demo_org
with self.tasks():
build_up_org_buffer()
assert (ORG_BUFFER_SIZE == DemoOrganization.objects.filter(
status__in=[DemoOrgStatus.PENDING, DemoOrgStatus.INITIALIZING
]).count())
mock_create_demo_org.assert_called_once_with()