def test_failing_view_DBAPIError(self): class StubDao: from sqlalchemy.exc import DBAPIError get_person = \ MagicMock(side_effect=DBAPIError('DB error', None, None)) request = testing.DummyRequest() request.db_session = None request.params['user_id'] = 'Ben' request.params['password'] = '******' login_view = LoginView(request, dao=StubDao) result = login_view.login() self.assertEqual(result.status_int, 500)
def test_passing_view(self): class StubDao: person = Person(id=123) get_person = \ MagicMock(spec=BaseDao, return_value=person) request = testing.DummyRequest() request.db_session = None request.params['user_id'] = 'Ben' request.params['password'] = '******' login_view = LoginView(request, dao=StubDao) result = login_view.login() self.assertEqual(result['user'], StubDao.person)