class TestLogin(unittest.TestCase): def create_organization(self): organization = Organization('organization_name', 'organization_sector', 'organization_address', 'organization_city', 'organization_state', 'organization_country', 'organization_zipcode', 'organization_office_phone', 'organization_website', org_id=self.org_id, in_trial_mode=True, active_date=datetime(2011, 07, 11)) organization.save() self.organization_setting = OrganizationSetting() self.organization_setting.organization = organization self.organization_setting.document_store = slugify( "%s_%s_%s" % ("HNI", organization.name, self.org_id)) self.organization_setting.save() @raises(AccountExpiredException) def test_should_raise_a_trial_account_expired_exception_if_trial_account_is_expired( self): self.org_id = OrganizationIdCreator().generateId() self.create_organization() form = FakeForm(org_id=self.org_id) form.check_trial_account_expired() def tearDown(self): org = Organization.objects.get(org_id=self.org_id) self.organization_setting.delete() org.delete()
def test_should_get_organisation_sms_number(self): mock_org = Mock(spec=Organization) mock_org.in_trial_mode = False with patch.object(OrganizationSetting, "_get_organization") as func_mock: func_mock.return_value = mock_org numbers = OrganizationSetting(sms_tel_number='32323,4422').get_organisation_sms_number() self.assertEquals(['32323','4422'], numbers)
def create_organization(self): organization = Organization('organization_name', 'organization_sector', 'organization_address', 'organization_city', 'organization_state', 'organization_country', 'organization_zipcode', 'organization_office_phone', 'organization_website', org_id = self.org_id, account_type = 'Basic', active_date = datetime(2011,07,11)) organization.save() self.organization_setting = OrganizationSetting() self.organization_setting.organization = organization self.organization_setting.document_store = slugify("%s_%s_%s" % ("HNI", organization.name, self.org_id)) self.organization_setting.save()
class TestAdmin(unittest.TestCase): def setUp(self): self.user = self._prepare_test_user() self.org_id = OrganizationIdCreator().generateId() self.create_organization() self.admin = FakeUserAdmin(self.org_id) def tearDown(self): User.objects.get(username='******').delete() def create_organization(self): organization = Organization('organization_name', 'organization_sector', 'organization_address', 'organization_city', 'organization_state', 'organization_country', 'organization_zipcode', 'organization_office_phone', 'organization_website', org_id=self.org_id, account_type='Basic', active_date=datetime(2011, 07, 11)) organization.save() self.organization_setting = OrganizationSetting() self.organization_setting.organization = organization self.organization_setting.document_store = slugify( "%s_%s_%s" % ("HNI", organization.name, self.org_id)) self.organization_setting.save() def test_should_delete_partial_digests_if_email_edited(self): request = Mock(spec=WSGIRequest) obj = User.objects.get(username='******') obj.email = '*****@*****.**' form = Mock() changed_data = PropertyMock(return_value=['email']) type(form).changed_data = changed_data cleaned_data = PropertyMock( return_value={'organization_id': self.org_id}) type(form).cleaned_data = cleaned_data self.admin.save_model(request, obj, form, True) self.assertRaises(PartialDigest.DoesNotExist, PartialDigest.objects.get, user=self.user) def test_should_not_delete_partial_digests_if_anything_other_than_email_edited( self): request = Mock(spec=WSGIRequest) obj = User.objects.get(username='******') obj.first_name = 'b' form = Mock() changed_data = PropertyMock(return_value=['first_name']) type(form).changed_data = changed_data cleaned_data = PropertyMock( return_value={'organization_id': self.org_id}) type(form).cleaned_data = cleaned_data self.admin.save_model(request, obj, form, True) self.assertEquals(PartialDigest.objects.get(user=self.user), self.digest) def _prepare_test_user(self): user = User(username='******', first_name='a', last_name='a', email='*****@*****.**', password='******') user.save() (NGOUserProfile(user=user)).save() digest = PartialDigest(user=user, login='******', partial_digest='partial', confirmed=True) digest.save() self.digest = digest return user def test_should_check_whether_user_is_ngo_admin(self): group = Group.objects.filter(name="NGO Admins") self.user.groups.add(group[0]) self.user.save() self.assertTrue(self.user.is_ngo_admin(), 'User is a NGOAdmin') def test_should_check_whether_user_is_extended_user(self): group = Group.objects.filter(name="Extended Users") self.user.groups.add(group[0]) self.user.save() self.assertTrue(self.user.is_extended_user())