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()