def test_to_dict(self):
"""Test output of to_dict() method"""
uid = UniqueIdentity(uuid='John Smith')
self.session.add(uid)
c = Country(code='US', name='United States of America', alpha3='USA')
self.session.add(c)
prf = Profile(uuid='John Smith', name='Smith, J.',
email='*****@*****.**', is_bot=True,
country_code='US')
self.session.add(prf)
self.session.commit()
# Tests
d = prf.to_dict()
self.assertIsInstance(d, dict)
self.assertEqual(d['uuid'], 'John Smith')
self.assertEqual(d['name'], 'Smith, J.')
self.assertEqual(d['email'], '*****@*****.**')
self.assertEqual(d['is_bot'], True)
self.assertEqual(d['country']['code'], 'US')
self.assertEqual(d['country']['name'], 'United States of America')
# No country set
prf = Profile(uuid='John Smith', name='Smith, J.',
email='*****@*****.**', is_bot=True,
country_code=None)
d = prf.to_dict()
self.assertEqual(d['country'], None)
def test_unique_profile(self):
"""Check if there is only one profile for each unique identity"""
uid = UniqueIdentity(uuid='John Smith')
self.session.add(uid)
prf1 = Profile(uuid='John Smith', name='John Smith')
prf2 = Profile(uuid='John Smith', name='Smith, J.')
with self.assertRaisesRegexp(IntegrityError, DUP_CHECK_ERROR):
self.session.add(prf1)
self.session.add(prf2)
self.session.commit()
def test_is_bot_invalid_type(self):
"""Check invalid values on is_bot bool column"""
with self.assertRaisesRegexp(StatementError, INVALID_DATATYPE_ERROR):
uid = UniqueIdentity(uuid='John Smith')
self.session.add(uid)
prf = Profile(uuid='John Smith', name='John Smith', is_bot='True')
self.session.add(prf)
self.session.commit()
def test_to_dict(self):
"""Test output of to_dict() method"""
uid = UniqueIdentity(uuid='John Smith')
self.session.add(uid)
c = Country(code='US', name='United States of America', alpha3='USA')
self.session.add(c)
prf = Profile(uuid='John Smith', name='Smith, J.',
email='*****@*****.**', is_bot=True,
country_code='US')
self.session.add(prf)
self.session.commit()
# Tests
d = prf.to_dict()
self.assertIsInstance(d, dict)
self.assertEqual(d['uuid'], 'John Smith')
self.assertEqual(d['name'], 'Smith, J.')
self.assertEqual(d['email'], '*****@*****.**')
self.assertEqual(d['is_bot'], True)
self.assertEqual(d['country']['code'], 'US')
self.assertEqual(d['country']['name'], 'United States of America')
# No country set
prf = Profile(uuid='John Smith', name='Smith, J.',
email='*****@*****.**', is_bot=True,
country_code=None)
d = prf.to_dict()
self.assertEqual(d['country'], None)
def test_get_profile_sh(self):
"""Test whether a profile from sortinghat model is correctly retrieved as a dict"""
p = Profile()
p.name = 'pepe'
p.email = '*****@*****.**'
p.gender = 'male'
p.gender_acc = 100
uidentity = UniqueIdentity()
uidentity.profile = p
vals = {'00000': uidentity}
def side_effect(uuid):
return vals[uuid]
self._enrich.get_unique_identity = MagicMock(side_effect=side_effect)
profile = self._enrich.get_profile_sh('00000')
self.assertEqual(profile['name'], uidentity.profile.name)
self.assertEqual(profile['email'], uidentity.profile.email)
self.assertEqual(profile['gender'], uidentity.profile.gender)
self.assertEqual(profile['gender_acc'], uidentity.profile.gender_acc)
def test_to_dict(self):
"""Test output of to_dict() method"""
c = Country(code='US', name='United States of America', alpha3='USA')
self.session.add(c)
uid = UniqueIdentity(uuid='John Smith')
self.session.add(uid)
id1 = Identity(id='A',
name='John Smith',
email='*****@*****.**',
username='******',
source='scm',
uuid='John Smith')
id2 = Identity(id='B',
name=None,
email='*****@*****.**',
username=None,
source='scm',
uuid='John Smith')
self.session.add(id1)
self.session.add(id2)
self.session.commit()
# Tests
d = uid.to_dict()
self.assertIsInstance(d, dict)
self.assertEqual(d['uuid'], 'John Smith')
self.assertEqual(d['profile'], None)
identities = d['identities']
self.assertEqual(len(identities), 2)
d0 = d['identities'][0]
self.assertEqual(d0['id'], 'A')
self.assertEqual(d0['name'], 'John Smith')
self.assertEqual(d0['email'], '*****@*****.**')
self.assertEqual(d0['username'], 'jsmith')
self.assertEqual(d0['source'], 'scm')
self.assertEqual(d0['uuid'], 'John Smith')
d1 = d['identities'][1]
self.assertEqual(d1['id'], 'B')
self.assertEqual(d1['name'], None)
self.assertEqual(d1['email'], '*****@*****.**')
self.assertEqual(d1['username'], None)
self.assertEqual(d1['source'], 'scm')
self.assertEqual(d1['uuid'], 'John Smith')
prf = Profile(uuid='John Smith',
name='Smith, J.',
email='*****@*****.**',
is_bot=True,
country_code='US')
# Add profile information
self.session.add(prf)
self.session.commit()
d = uid.to_dict()
dp = d['profile']
self.assertEqual(dp['uuid'], 'John Smith')
self.assertEqual(dp['name'], 'Smith, J.')
self.assertEqual(dp['email'], '*****@*****.**')
self.assertEqual(dp['is_bot'], True)
self.assertEqual(dp['country']['code'], 'US')
self.assertEqual(dp['country']['name'], 'United States of America')