diff options
| author | Theron Luhn <theron@luhn.com> | 2019-04-15 19:14:42 -0700 |
|---|---|---|
| committer | Theron Luhn <theron@luhn.com> | 2019-04-15 19:14:42 -0700 |
| commit | d6e543bc01d2f1aa3bb29f005171911f6f09da02 (patch) | |
| tree | 06bb8c9ac7089de2a20287ef703229b22ce777d1 /tests | |
| parent | f4c6c993ded900b4e32d1dd49207ca1e18b11336 (diff) | |
| download | pyramid-d6e543bc01d2f1aa3bb29f005171911f6f09da02.tar.gz pyramid-d6e543bc01d2f1aa3bb29f005171911f6f09da02.tar.bz2 pyramid-d6e543bc01d2f1aa3bb29f005171911f6f09da02.zip | |
Move SessionAuthenticationHelper to pyramid.authentication.
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/test_authentication.py | 50 | ||||
| -rw-r--r-- | tests/test_security.py | 50 |
2 files changed, 50 insertions, 50 deletions
diff --git a/tests/test_authentication.py b/tests/test_authentication.py index 8671eba05..710e87423 100644 --- a/tests/test_authentication.py +++ b/tests/test_authentication.py @@ -1693,6 +1693,56 @@ class TestSessionAuthenticationPolicy(unittest.TestCase): self.assertEqual(result, []) +class TestSessionAuthenticationHelper(unittest.TestCase): + def _makeRequest(self, session=None): + from types import SimpleNamespace + + if session is None: + session = dict() + return SimpleNamespace(session=session) + + def _makeOne(self, prefix=''): + from pyramid.authentication import SessionAuthenticationHelper + + return SessionAuthenticationHelper(prefix=prefix) + + def test_identify(self): + request = self._makeRequest({'userid': 'fred'}) + helper = self._makeOne() + self.assertEqual(helper.identify(request), 'fred') + + def test_identify_with_prefix(self): + request = self._makeRequest({'foo.userid': 'fred'}) + helper = self._makeOne(prefix='foo.') + self.assertEqual(helper.identify(request), 'fred') + + def test_identify_none(self): + request = self._makeRequest() + helper = self._makeOne() + self.assertEqual(helper.identify(request), None) + + def test_remember(self): + request = self._makeRequest() + helper = self._makeOne() + result = helper.remember(request, 'fred') + self.assertEqual(request.session.get('userid'), 'fred') + self.assertEqual(result, []) + + def test_forget(self): + request = self._makeRequest({'userid': 'fred'}) + helper = self._makeOne() + result = helper.forget(request) + self.assertEqual(request.session.get('userid'), None) + self.assertEqual(result, []) + + def test_forget_no_identity(self): + request = self._makeRequest() + helper = self._makeOne() + result = helper.forget(request) + self.assertEqual(request.session.get('userid'), None) + self.assertEqual(result, []) + + class TestBasicAuthAuthenticationPolicy(unittest.TestCase): def _getTargetClass(self): from pyramid.authentication import BasicAuthAuthenticationPolicy as cls diff --git a/tests/test_security.py b/tests/test_security.py index f14159156..ecd6a088b 100644 --- a/tests/test_security.py +++ b/tests/test_security.py @@ -887,53 +887,3 @@ GUEST_PERMS = (VIEW, COMMENT) MEMBER_PERMS = GUEST_PERMS + (EDIT, CREATE, DELETE) MODERATOR_PERMS = MEMBER_PERMS + (MODERATE,) ADMINISTRATOR_PERMS = MODERATOR_PERMS + (ADMINISTER,) - - -class TestSessionAuthenticationHelper(unittest.TestCase): - def _makeRequest(self, session=None): - from types import SimpleNamespace - - if session is None: - session = dict() - return SimpleNamespace(session=session) - - def _makeOne(self, prefix=''): - from pyramid.security import SessionAuthenticationHelper - - return SessionAuthenticationHelper(prefix=prefix) - - def test_identify(self): - request = self._makeRequest({'userid': 'fred'}) - helper = self._makeOne() - self.assertEqual(helper.identify(request), 'fred') - - def test_identify_with_prefix(self): - request = self._makeRequest({'foo.userid': 'fred'}) - helper = self._makeOne(prefix='foo.') - self.assertEqual(helper.identify(request), 'fred') - - def test_identify_none(self): - request = self._makeRequest() - helper = self._makeOne() - self.assertEqual(helper.identify(request), None) - - def test_remember(self): - request = self._makeRequest() - helper = self._makeOne() - result = helper.remember(request, 'fred') - self.assertEqual(request.session.get('userid'), 'fred') - self.assertEqual(result, []) - - def test_forget(self): - request = self._makeRequest({'userid': 'fred'}) - helper = self._makeOne() - result = helper.forget(request) - self.assertEqual(request.session.get('userid'), None) - self.assertEqual(result, []) - - def test_forget_no_identity(self): - request = self._makeRequest() - helper = self._makeOne() - result = helper.forget(request) - self.assertEqual(request.session.get('userid'), None) - self.assertEqual(result, []) |
