def setUp(): g._push_object(Globals()) c._push_object(mock.Mock()) request._push_object(Request.blank('/')) ThreadLocalORMSession.close_all() M.EmailAddress.query.remove({}) M.OpenIdNonce.query.remove({}) M.OpenIdAssociation.query.remove({})
def test_grab_user(self): class FakeRequest(object): pass class FakeUser(object): pass class SomeObject(object): pass req = FakeRequest() req.environ = {} req.environ['user'] = FakeUser() test_obj = SomeObject() req.environ['user'].id = test_obj request._push_object(req) assert h.grab(None, 'spender_id') == test_obj assert h.grab(None, 'creditor_id') == test_obj assert h.grab(None, 'debtor_id') == test_obj request._pop_object()