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
