forked from DiamondLightSource/python-workflows
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathtest_logging.py
More file actions
66 lines (54 loc) · 2.02 KB
/
Copy pathtest_logging.py
File metadata and controls
66 lines (54 loc) · 2.02 KB
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
from __future__ import absolute_import, division
import logging
import os
import mock
import workflows.logging
def test_obtain_exception_source():
'''Check that exception information can be obtained.'''
def crashfunction():
'''A function that crashes in a predictable manner.'''
everything = 42
nothing = 0
print(everything / nothing)
try:
crashfunction()
assert False, "Test went past exception."
except Exception:
fp, fn, line_number, name, line = workflows.logging.get_exception_source()
assert fp == __file__
assert os.path.basename(fp) == fn
assert name == 'crashfunction'
assert line_number == 16 # this will break if line number in this file changes
assert line == "print(everything / nothing)"
def test_callback_handler_works_within_logging_framework():
'''Check that the callback handler can be used by Python logging
and works as expected.'''
cbmock = mock.Mock()
logmsg = 'Test message for callback'
log = logging.getLogger('workflows.tests.logging.callback')
log.setLevel(logging.INFO)
cbh = workflows.logging.CallbackHandler(cbmock)
cbh.handleError = mock.Mock()
log.addHandler(cbh)
log.info(logmsg)
cbmock.assert_called_once()
assert cbmock.call_args == ((mock.ANY,), {})
logrec = cbmock.call_args[0][0]
assert isinstance(logrec, logging.LogRecord)
assert logrec.name == 'workflows.tests.logging.callback'
assert logrec.levelname == 'INFO'
assert logrec.message == logmsg
assert logrec.funcName.startswith('test_')
assert not cbh.handleError.called
# Now check that the callback handler can handle errors in the
# callback function.
logmsg = 'Test message for error in logging'
cbmock.side_effect=AttributeError('Some failure')
log.info(logmsg)
assert cbmock.call_count == 2
assert cbmock.call_args == ((mock.ANY,), {})
logrec = cbmock.call_args[0][0]
assert isinstance(logrec, logging.LogRecord)
assert logrec.message == logmsg
cbh.handleError.assert_called_once()
assert cbh.handleError.call_args == cbmock.call_args