This repository has been archived by the owner on Jul 24, 2023. It is now read-only.
/
support.py
69 lines (56 loc) · 2.35 KB
/
support.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
from openid import message
from logging.handlers import BufferingHandler
import logging
class TestHandler(BufferingHandler):
def __init__(self, messages):
BufferingHandler.__init__(self, 0)
self.messages = messages
def shouldFlush(self):
return False
def emit(self, record):
self.messages.append(record.__dict__)
class OpenIDTestMixin(object):
def failUnlessOpenIDValueEquals(self, msg, key, expected, ns=None):
if ns is None:
ns = message.OPENID_NS
actual = msg.getArg(ns, key)
error_format = 'Wrong value for openid.%s: expected=%s, actual=%s'
error_message = error_format % (key, expected, actual)
self.failUnlessEqual(expected, actual, error_message)
def failIfOpenIDKeyExists(self, msg, key, ns=None):
if ns is None:
ns = message.OPENID_NS
actual = msg.getArg(ns, key)
error_message = 'openid.%s unexpectedly present: %s' % (key, actual)
self.failIf(actual is not None, error_message)
class CatchLogs(object):
def setUp(self):
self.messages = []
root_logger = logging.getLogger()
self.old_log_level = root_logger.getEffectiveLevel()
root_logger.setLevel(logging.DEBUG)
self.handler = TestHandler(self.messages)
formatter = logging.Formatter("%(message)s [%(asctime)s - %(name)s - %(levelname)s]")
self.handler.setFormatter(formatter)
root_logger.addHandler(self.handler)
def tearDown(self):
root_logger = logging.getLogger()
root_logger.removeHandler(self.handler)
root_logger.setLevel(self.old_log_level)
def failUnlessLogMatches(self, *prefixes):
"""
Check that the log messages contained in self.messages have
prefixes in *prefixes. Raise AssertionError if not, or if the
number of prefixes is different than the number of log
messages.
"""
messages = [r['msg'] for r in self.messages]
assert len(prefixes) == len(messages), \
"Expected log prefixes %r, got %r" % (prefixes,
messages)
for prefix, message in zip(prefixes, messages):
assert message.startswith(prefix), \
"Expected log prefixes %r, got %r" % (prefixes,
messages)
def failUnlessLogEmpty(self):
self.failUnlessLogMatches()