def test_UserPhoneClaim_unicode(self):
"""
Test for verifying whether UserPhoneClaim instance
returns phone in unicode format
"""
haymitch = models.User(username=u'haymitch', fullname=u'Haymitch Abernathy')
phone = u'9191919191'
phone_claim = models.UserPhoneClaim(phone=phone, owner=haymitch)
db.session.add(phone_claim)
db.session.commit()
result = unicode(models.UserPhoneClaim(phone=phone))
self.assertIsInstance(result, unicode)
assert phone in result
def test_UserPhoneClaim_all(self):
"""
Test for retrieving all instances of UserPhoneClaim given a
phone number
"""
crusoe = self.fixtures.crusoe
oakley = self.fixtures.oakley
phone = u'9191919191'
claim_by_crusoe = models.UserPhoneClaim(phone=phone, owner=crusoe)
claim_by_oakley = models.UserPhoneClaim(phone=phone, owner=oakley)
db.session.add(claim_by_crusoe, claim_by_oakley)
db.session.commit()
result = models.UserPhoneClaim.all(phone)
self.assertIsInstance(result, list)
self.assertItemsEqual(result, [claim_by_crusoe, claim_by_oakley])
def test_ownermixin_owner(self):
"""Test to verify setting of owner property on any OwnerMixin inherited instance"""
crusoe = self.fixtures.crusoe
# scenario 1: check owner of existing UserEmail instance
crusoe_email = self.fixtures.crusoe_email
self.assertEqual(crusoe_email.owner, crusoe)
# scenario 2: check owner when owner is org
client = self.fixtures.client
batdog = self.fixtures.batdog
dachshunds = self.fixtures.dachshunds
self.assertEqual(client.owner, batdog)
# scenaio 3: check owner when new UserEmailClaim instance
spock = models.User(username=u'spock')
spock_email_claim = models.UserEmailClaim(
email=u'*****@*****.**', user=spock)
self.assertEqual(spock_email_claim.owner, spock)
# scenario 4: set user as owner on some OwnerMixin inherited instance
spock_phone = models.UserPhoneClaim(phone=u'+9112345678', user=spock)
spock_phone.owner = spock
self.assertEqual(spock_phone.owner, spock)
# scenario 5: set organization as owner on some OwnerMixin inherited instance
spock_phone.owner = batdog
self.assertEqual(spock_phone.owner, batdog)
# scenario 6: set Team as owner on some OwnerMixin inherited instance
spock_phone.owner = dachshunds
self.assertEqual(spock_phone.owner, dachshunds)
# scenario 7: when owner not insance of User, organization or Team
with self.assertRaises(ValueError):
spock_phone.owner = client
def test_UserPhoneClaim(self):
"""
Test for creation of UserPhoneClaim instance
"""
phone = u'9123456780'
result = models.UserPhoneClaim(phone)
self.assertIsInstance(result, models.UserPhoneClaim)
self.assertEqual(result.phone, phone)
def test_UserPhoneClaim_get(self):
"""
Test for retrieving UserPhoneClaim instances given phone
number and a user
"""
snow = models.User(username=u'', fullname=u'President Coriolanus Snow')
phone = u'9191919191'
phone_claim = models.UserPhoneClaim(phone=phone, owner=snow)
db.session.add(phone_claim)
db.session.commit()
result = models.UserPhoneClaim.get(phone, snow)
self.assertIsInstance(result, models.UserPhoneClaim)
self.assertEqual(result.phone, phone)
self.assertEqual(result.owner, snow)
def test_userphoneclaim_permissions(self):
"""
Test for verifying whether user has verify permission on a
UserPhoneClaim instance
"""
coin = models.User(username=u'coin', fullname=u'President Alma Coin')
phone = u'9191919191'
phone_claim = models.UserPhoneClaim(phone=phone, owner=coin)
permissions_expected = ['verify']
result = phone_claim.permissions(coin)
self.assertIsInstance(result, set)
permissions_received = []
for each in result:
permissions_received.append(each)
self.assertItemsEqual(permissions_expected, permissions_received)
