def test_get_paid_org_should_return_only_paid_org(self): TRIAL_ORGANIZATION_PARAMS = {'organization_name': 'myCompany', 'organization_sector': 'Public Health', 'organization_city': 'xian', 'organization_country': 'china', } self.organization = Organization.create_trial_organization(TRIAL_ORGANIZATION_PARAMS) self.organization.save() organizations = _get_paid_organization() self.assertNotIn(self.organization,organizations)
def create_respective_organization(self, kwargs): if self.is_subscription_registration(kwargs): organization = Organization.create_organization(kwargs) else: organization = Organization.create_trial_organization(kwargs) return organization