DummySession: Correctly implement ISession by generating CSRF token if it's...

Robert Buchholz

2011-11-02 1f8a007f51cda0f4ebfc74d1a178cbab136f5ea0

DummySession: Correctly implement ISession by generating CSRF token if it's missing

 pyramid/testing.py

@@ -619,7 +619,11 @@
        return token

    def get_csrf_token(self):
        return self.get('_csrft_', None)
        token = self.get('_csrft_', None)
        if token is None:
            token = self.new_csrf_token()
        return token

        
class DummyRequest(DeprecatedRequestMethodsMixin, URLMethodsMixin,
                   CallbackMethodsMixin):

 pyramid/tests/test_testing.py

@@ -886,6 +886,11 @@
        self.assertEqual(token, 'token')
        self.assertTrue('_csrft_' in session)

    def test_get_csrf_token_generates_token(self):
        session = self._makeOne()
        token = session.get_csrf_token()
        self.assertNotEqual(token, None)
        self.assertTrue(len(token) >= 1)

from zope.interface import Interface
from zope.interface import implements