def get_charity(self, name=None, id=None):
if id is not None:
try:
charity = Charity.get_by_id(id)
if charity is None:
raise MissingValueException(
'Could not find charity id=%s' % id)
return charity
except BadKeyError:
raise MissingValueException('Could not find charity id=%s' %
id)
query = Charity.all()
if name is not None:
query = query.filter('name =', name)
charities = [charity for charity in query.run()]
count = len(charities)
if count == 1:
return charities[0]
elif count == 0:
raise MissingValueException('Could not find charity name=%s' %
name)
elif count > 1:
raise MultipleValueException(
'Expected one charity to be returned, but got multiple',
charities)
def test_view_charity_urls(self):
charity = Charity(name='Oxfam')
CharityRepository().add_or_update_charity(charity)
view_charity_page = ViewCharityPage(self.driver, self.get_base_url(),
charity).load()
self.assertIn(charity.name, view_charity_page.title)
view_charity_page.without_charity_name_in_url().load()
self.assertIn(charity.name, view_charity_page.title)
def get_charity(self, name=None, id=None):
if id is not None:
try:
charity = Charity.get_by_id(id)
if charity is None:
raise MissingValueException('Could not find charity id=%s' % id)
return charity
except BadKeyError:
raise MissingValueException('Could not find charity id=%s' % id)
query = Charity.all()
if name is not None:
query = query.filter('name =', name)
charities = [charity for charity in query.run()]
count = len(charities)
if count == 1:
return charities[0]
elif count == 0:
raise MissingValueException('Could not find charity name=%s' % name)
elif count > 1:
raise MultipleValueException('Expected one charity to be returned, but got multiple', charities)
from givabit.backend.charity import Charity from givabit.backend.user import User c1 = Charity(name='Shelter') c2 = Charity(name='Oxfam') c3 = Charity(name='BHF') c4 = Charity(name='BHF') u1 = User(email='*****@*****.**') u2 = User(email='*****@*****.**') u3 = User(email='*****@*****.**') u4 = User(email='*****@*****.**') u5 = User(email='*****@*****.**')
def list_charities(self):
return [charity for charity in Charity.all().run()]
def list_charities(self):
return [charity for charity in Charity.all().run()]
