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