def test_get_multi_associations_with_one_missing(self) -> None:
auth_services.associate_auth_id_with_user_id(
auth_domain.AuthIdUserIdPair('aid1', 'uid1'))
# The aid2 <-> uid2 association is missing.
auth_services.associate_auth_id_with_user_id(
auth_domain.AuthIdUserIdPair('aid3', 'uid3'))
self.assertEqual(
auth_services.get_multi_user_ids_from_auth_ids(
['aid1', 'aid2', 'aid3']), ['uid1', None, 'uid3'])
self.assertEqual(
auth_services.get_multi_auth_ids_from_user_ids(
['uid1', 'uid2', 'uid3']), ['aid1', None, 'aid3'])
def test_get_multi_associations_with_all_present(self) -> None:
auth_services.associate_auth_id_with_user_id(
auth_domain.AuthIdUserIdPair('aid1', 'uid1'))
auth_services.associate_auth_id_with_user_id(
auth_domain.AuthIdUserIdPair('aid2', 'uid2'))
auth_services.associate_auth_id_with_user_id(
auth_domain.AuthIdUserIdPair('aid3', 'uid3'))
self.assertEqual(
auth_services.get_multi_user_ids_from_auth_ids(
['aid1', 'aid2', 'aid3']), ['uid1', 'uid2', 'uid3'])
self.assertEqual(
auth_services.get_multi_auth_ids_from_user_ids(
['uid1', 'uid2', 'uid3']), ['aid1', 'aid2', 'aid3'])
