Пример #1
0
    def test_it_does_nothing_usually(self, handler, pyramid_request):
        tween = tweens.rollback_db_session_on_exception_factory(
            handler, pyramid_request.registry)

        tween(pyramid_request)

        handler.assert_called_once_with(pyramid_request)
        pyramid_request.db.rollback.assert_not_called()
Пример #2
0
    def test_it_calls_db_rollback_on_exception(self, handler, pyramid_request):
        handler.side_effect = IOError

        tween = tweens.rollback_db_session_on_exception_factory(
            handler, pyramid_request.registry)

        with pytest.raises(IOError):
            tween(pyramid_request)

        handler.assert_called_once_with(pyramid_request)
        pyramid_request.db.rollback.assert_called_once_with()