def test_fetch_ip_whitelists_non_empty(self): ent = model.AuthIPWhitelistAssignments( key=model.ip_whitelist_assignments_key()) def add(identity, **kwargs): kwargs['identity'] = model.Identity.from_bytes(identity) ent.assignments.append( model.AuthIPWhitelistAssignments.Assignment(**kwargs)) add('user:[email protected]', ip_whitelist='A') add('user:[email protected]', ip_whitelist='A') add('user:[email protected]', ip_whitelist='B') add('user:[email protected]', ip_whitelist='missing') ent.put() def store_whitelist(name): model.AuthIPWhitelist(key=model.ip_whitelist_key(name)).put() store_whitelist('A') store_whitelist('B') store_whitelist('bots') assignments, whitelists = model.fetch_ip_whitelists() self.assertEqual(ent.to_dict(), assignments.to_dict()) self.assertEqual(['A', 'B', 'bots'], [e.key.id() for e in whitelists])
def test_fetch_ip_whitelists_empty(self): assignments, whitelists = model.fetch_ip_whitelists() self.assertEqual(model.ip_whitelist_assignments_key(), assignments.key) self.assertEqual(0, len(assignments.assignments)) self.assertEqual([], whitelists)