class ManagerTestCase(TestCase):
def setUp(self):
from repose.managers import Manager
super(ManagerTestCase, self).setUp()
self.manager = Manager()
self.manager.contribute_api(self.api)
self.manager.contribute_to_class(User)
def test_get(self):
self.api.add_response('GET', '/user/1', USER_DATA)
user = self.manager.get(user_id=1)
self.assertEqual(user.id, 1)
self.assertEqual(user.name, 'Test User')
def test_all(self):
self.manager.contribute_to_class(User)
self.api.add_response('GET', '/user', [USER_DATA])
users = self.manager.all()
self.assertEqual(len(users), 1)
self.assertEqual(users[0]['id'], 1)
def test_count(self):
self.manager.contribute_to_class(User)
self.api.add_response('GET', '/user', [USER_DATA])
self.assertEqual(self.manager.count(), 1)
def test_iter(self):
self.manager.contribute_to_class(User)
self.api.add_response('GET', '/user', [USER_DATA])
self.assertEqual(len([u for u in self.manager]), 1)
def test_filtering(self):
from repose.managers import Manager
self.manager = Manager(filter=lambda u: u.id == 2)
self.manager.contribute_api(self.api)
self.manager.contribute_to_class(User)
self.manager.contribute_to_class(User)
user2 = USER_DATA.copy()
user2['id'] = 2
self.api.add_response('GET', '/user', [USER_DATA, user2])
self.assertEqual(self.manager.count(), 1)
self.assertEqual(len(self.manager.all()), 1)
