diff options
Diffstat (limited to 'repoze/bfg/tests/test_authentication.py')
| -rw-r--r-- | repoze/bfg/tests/test_authentication.py | 13 |
1 files changed, 3 insertions, 10 deletions
diff --git a/repoze/bfg/tests/test_authentication.py b/repoze/bfg/tests/test_authentication.py index bce80ca20..d020a11a6 100644 --- a/repoze/bfg/tests/test_authentication.py +++ b/repoze/bfg/tests/test_authentication.py @@ -419,8 +419,8 @@ class TestAuthTktCookieHelper(unittest.TestCase): self.assertEqual(len(request.callbacks), 1) response = DummyResponse() request.callbacks[0](None, response) - self.assertEqual(len(response.headers.added), 3) - self.assertEqual(response.headers.added[0][0], 'Set-Cookie') + self.assertEqual(len(response.headerlist), 3) + self.assertEqual(response.headerlist[0][0], 'Set-Cookie') def test_remember(self): plugin = self._makeOne('secret') @@ -658,14 +658,7 @@ class DummyAuthTktModule(object): class BadTicket(Exception): pass -class DummyHeaders: - def __init__(self): - self.added = [] - - def add(self, k, v): - self.added.append((k, v)) - class DummyResponse: def __init__(self): - self.headers = DummyHeaders() + self.headerlist = [] |
