blob: 3bdf0b5578e744f603e634622bc3e45cea0de6fc (
plain)
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
|
# Copyright (c) 2011-2013 Simplistix Ltd
# See license.txt for license details.
from nose.plugins.skip import SkipTest
try:
from testfixtures.components import TestComponents
except ImportError: # pragma: no cover
raise SkipTest('zope.component is not available')
from mock import Mock, call
from testfixtures import Replacer, compare
from unittest import TestCase
from warnings import catch_warnings
class ComponentsTests(TestCase):
def test_atexit(self):
m = Mock()
with Replacer() as r:
r.replace('atexit.register', m.register)
c = TestComponents()
expected = [call.register(c.atexit)]
compare(expected, m.mock_calls)
with catch_warnings(record=True) as w:
c.atexit()
self.assertTrue(len(w), 1)
compare(str(w[0].message), ( # pragma: no branch
"TestComponents instances not uninstalled by shutdown!"
))
c.uninstall()
compare(expected, m.mock_calls)
# check re-running has no ill effects
c.atexit()
compare(expected, m.mock_calls)
|