def __getitem__(self, key): mgr = OrganizationManager(self.request) org = mgr.get_by_pk(key) if org: org.__parent__ = self org.__name__ = key return org
def test_get_organization_get_all(self): from pyramid_signup.models import Organization from pyramid_signup.models import User from pyramid_signup.managers import OrganizationManager user = User(username='******', password='******') self.session.add(user) organization = Organization('test org', user) self.session.add(organization) self.session.commit() request = testing.DummyRequest() mgr = OrganizationManager(request) orgs = mgr.get_all() assert len(orgs) == 1
def test_get_organization_bad_pk(self): from pyramid_signup.models import Organization from pyramid_signup.models import User from pyramid_signup.managers import OrganizationManager user = User(username='******', password='******') self.session.add(user) organization = Organization('test org', user) self.session.add(organization) self.session.commit() request = testing.DummyRequest() mgr = OrganizationManager(request) new_org = mgr.get_by_pk(99999) assert new_org == None