def assert_multi_auth_mappings_do_not_exist(self, auth_assocs):
"""Asserts that the given auth associations exist.
Args:
auth_assocs: list(AuthIdUserIdPair). The association to check.
"""
auth_ids, user_ids = (list(a) for a in python_utils.ZIP(*auth_assocs))
self.assertEqual(
firebase_auth_services.get_multi_user_ids_from_auth_ids(auth_ids),
[None] * len(auth_ids))
self.assertEqual(
firebase_auth_services.get_multi_auth_ids_from_user_ids(user_ids),
[None] * len(user_ids))
def test_get_multi_associations_with_one_missing(self):
firebase_auth_services.associate_auth_id_with_user_id(
auth_domain.AuthIdUserIdPair('aid1', 'uid1'))
# The aid2 <-> uid2 association is missing.
firebase_auth_services.associate_auth_id_with_user_id(
auth_domain.AuthIdUserIdPair('aid3', 'uid3'))
self.assertEqual(
firebase_auth_services.get_multi_user_ids_from_auth_ids(
['aid1', 'aid2', 'aid3']), ['uid1', None, 'uid3'])
self.assertEqual(
firebase_auth_services.get_multi_auth_ids_from_user_ids(
['uid1', 'uid2', 'uid3']), ['aid1', None, 'aid3'])
def assert_firebase_assoc_does_not_exist_multi(self, assocs):
"""Asserts that the given users' Firebase association doesn't exist.
Args:
assocs: list(AuthIdUserIdPair). The associations to check.
"""
auth_ids, user_ids = (list(a) for a in python_utils.ZIP(*assocs))
self.assertEqual(
firebase_auth_services.get_multi_user_ids_from_auth_ids(auth_ids),
[None] * len(auth_ids))
self.assertEqual(
firebase_auth_services.get_multi_auth_ids_from_user_ids(user_ids),
[None] * len(user_ids))
def test_get_multi_associations_with_all_present(self):
firebase_auth_services.associate_auth_id_with_user_id(
auth_domain.AuthIdUserIdPair('aid1', 'uid1'))
firebase_auth_services.associate_auth_id_with_user_id(
auth_domain.AuthIdUserIdPair('aid2', 'uid2'))
firebase_auth_services.associate_auth_id_with_user_id(
auth_domain.AuthIdUserIdPair('aid3', 'uid3'))
self.assertEqual(
firebase_auth_services.get_multi_user_ids_from_auth_ids(
['aid1', 'aid2', 'aid3']), ['uid1', 'uid2', 'uid3'])
self.assertEqual(
firebase_auth_services.get_multi_auth_ids_from_user_ids(
['uid1', 'uid2', 'uid3']), ['aid1', 'aid2', 'aid3'])
