def test_associate_with_user_id_collision_raises(self):
gae_auth_services.associate_auth_id_with_user_id(
auth_domain.AuthIdUserIdPair('aid', 'uid'))
with self.assertRaisesRegexp(Exception, 'already associated'):
gae_auth_services.associate_auth_id_with_user_id(
auth_domain.AuthIdUserIdPair('aid', 'uid'))
def test_get_association_that_is_present(self):
gae_auth_services.associate_auth_id_with_user_id(
auth_domain.AuthIdUserIdPair('aid', 'uid'))
self.assertEqual(gae_auth_services.get_user_id_from_auth_id('aid'),
'uid')
self.assertEqual(gae_auth_services.get_auth_id_from_user_id('uid'),
'aid')
def test_associate_without_collision(self):
gae_auth_services.associate_auth_id_with_user_id(
auth_domain.AuthIdUserIdPair('aid', 'uid'))
self.assertEqual(gae_auth_services.get_user_id_from_auth_id('aid'),
'uid')
self.assertEqual(gae_auth_services.get_auth_id_from_user_id('uid'),
'aid')
def test_associate_with_auth_id_collision_raises(self):
gae_auth_services.associate_auth_id_with_user_id(
auth_domain.AuthIdUserIdPair('aid', 'uid'))
# Erase the user_id collision, but leave the auth_id collision.
auth_models.UserIdentifiersModel.delete_by_id('aid')
with self.assertRaisesRegexp(Exception, 'already associated'):
gae_auth_services.associate_auth_id_with_user_id(
auth_domain.AuthIdUserIdPair('aid', 'uid'))
def test_get_multi_associations_with_one_missing(self):
gae_auth_services.associate_auth_id_with_user_id(
auth_domain.AuthIdUserIdPair('aid1', 'uid1'))
# The aid2 <-> uid2 association is missing.
gae_auth_services.associate_auth_id_with_user_id(
auth_domain.AuthIdUserIdPair('aid3', 'uid3'))
self.assertEqual(
gae_auth_services.get_multi_user_ids_from_auth_ids(
['aid1', 'aid2', 'aid3']), ['uid1', None, 'uid3'])
self.assertEqual(
gae_auth_services.get_multi_auth_ids_from_user_ids(
['uid1', 'uid2', 'uid3']), ['aid1', None, 'aid3'])
def test_get_association_that_is_present_and_marked_as_deleted(self):
gae_auth_services.associate_auth_id_with_user_id(
auth_domain.AuthIdUserIdPair('aid', 'uid'))
assoc_by_auth_id_model = auth_models.UserIdentifiersModel.get('aid')
assoc_by_auth_id_model.deleted = True
assoc_by_auth_id_model.update_timestamps()
assoc_by_auth_id_model.put()
self.assertEqual(gae_auth_services.get_user_id_from_auth_id('aid'),
'uid')
self.assertEqual(gae_auth_services.get_auth_id_from_user_id('uid'),
'aid')
def test_get_multi_associations_with_all_present(self):
gae_auth_services.associate_auth_id_with_user_id(
auth_domain.AuthIdUserIdPair('aid1', 'uid1'))
gae_auth_services.associate_auth_id_with_user_id(
auth_domain.AuthIdUserIdPair('aid2', 'uid2'))
gae_auth_services.associate_auth_id_with_user_id(
auth_domain.AuthIdUserIdPair('aid3', 'uid3'))
self.assertEqual(
gae_auth_services.get_multi_user_ids_from_auth_ids(
['aid1', 'aid2', 'aid3']), ['uid1', 'uid2', 'uid3'])
self.assertEqual(
gae_auth_services.get_multi_auth_ids_from_user_ids(
['uid1', 'uid2', 'uid3']), ['aid1', 'aid2', 'aid3'])
def test_disable_association_marks_model_for_deletion(self):
gae_auth_services.associate_auth_id_with_user_id(
auth_domain.AuthIdUserIdPair('aid', 'uid'))
gae_auth_services.mark_user_for_deletion('uid')
self.assertIsNone(
auth_models.UserIdentifiersModel.get('aid', strict=False))
