Test add_request_method

fullapp.py

class TestFooAddRequest(unittest.TestCase):
    def setUp(self, environ=None):
        self.wsgi_app = self.config.make_wsgi_app()
        from webtest import TestApp
        self.test_app = TestApp(
            self.wsgi_app
         )

    def _makeRequest(self, environ=None):
        if environ is None:
            environ = {}
        from pyramid.testing import DummyRequest

        request = DummyRequest(environ)
        request.registry = self.config.registry
        return request

    def _makeBeforeRender(self, system, val=None):
        from pyramid.events import BeforeRender
        return BeforeRender(system, val)

    def _makeConfig(self):
        from pyramid import testing
        config = testing.setUp()
        return config

    def test_foo(self):
        from bar import foo

        self.config.add_request_method(foo, name='foo')





class DummyRegistry(object):
    #{
    #  'force_account_type': None,
    #  'disable_features': False
    #}

    def __init__(self):
        self.settings = {}

    def queryUtility(self, *arg, **kw):
        pass

class DummyResponse(object):
    def __init__(self):
        self.headerlist = []


class DummyUser(object):
    id = 2

    def __init__(self):
        pass

    def to_dict(self):
        return self.__dict__

usingconfig.py

class TestFooAddRequest(unittest.TestCase):

    def _makeRequest(self, environ=None):
        if environ is None:
            environ = {}
        from pyramid.testing import DummyRequest

        request = DummyRequest(environ)
        request.registry = self.config.registry
        return request

    def _makeBeforeRender(self, system, val=None):
        from pyramid.events import BeforeRender
        return BeforeRender(system, val)

    def _makeConfig(self):
        from pyramid import testing
        config = testing.setUp()
        return config

    def test_foo(self):
        from bar import foo
        config = self._makeConfig()
        
        config.set_request_factory(self._makeRequest())
        config.add_request_method(foo, name='foo')

        assert request.foo() == 'foobarbaz'



class DummyRegistry(object):
    #{
    #  'force_account_type': None,
    #  'disable_features': False
    #}

    def __init__(self):
        self.settings = {}

    def queryUtility(self, *arg, **kw):
        pass

class DummyResponse(object):
    def __init__(self):
        self.headerlist = []


class DummyUser(object):
    id = 2

    def __init__(self):
        pass

    def to_dict(self):
        return self.__dict__