def test_get_users(self): from lasco import api from lasco.models import User self.assertEqual(api.get_users(self.session), []) user = User(u'user1', u'fullname', 'password') self.session.add(user) self.assertEqual(api.get_users(self.session), [user])
def test_remove_user(self): from lasco import api from lasco.models import User user = User(u'user1', u'fullname', 'password') self.session.add(user) self.assertEqual(len(api.get_users(self.session)), 1) api.remove_user(self.session, u'user1') self.assertEqual(api.get_users(self.session), [])