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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
|
# encoding=utf-8
# Copyright © 2017 Dylan Baker
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
"""Helpers for unittests themselves."""
from __future__ import absolute_import
import functools
import unittest
def _tear_down_class_wrapper(original, cls):
"""Ensure that doClassCleanups is called after tearDownClass."""
try:
original()
finally:
cls.doClassCleanups()
def _set_up_class_wrapper(original, cls):
"""If setUpClass fails, call doClassCleanups."""
try:
original()
except Exception:
cls.doClassCleanups()
raise
class TestCaseClassCleanup(unittest.TestCase):
"""A subclass of unittest.TestCase which adds classlevel clenups methods.
"""
__stack = []
def __new__(cls, _):
"""Wrap the tearDownClass method to esnure that doClassCleanups gets
called.
Because doCleanups (the test instance level version of this
functionality) is called in code we can't supclass we need to do some
hacking to ensure it's called. that hackery is in the form of wrapping
the call to tearDownClass and setupClass methods.
"""
original = cls.tearDownClass
# Get a unique object, otherwise functools.update_wrapper will always
# act on the same object. We're using functools.partial as a proxy to
# receive that information.
#
# We're also passing the original implementation to the wrapper
# function as an argument, because it is being passed an unbound class
# method the calling function will need to pass cls to it as an
# argument.
unique = functools.partial(_tear_down_class_wrapper, original, cls)
# the classmethod decorator hides the __module__ attribute, so don't
# try to set it. In python 3.x this is no longer true and the lat
# parameter of this call can be removed
functools.update_wrapper(unique, original, ['__name__', '__doc__'])
# Repalce the orinal tearDownClass method with our wrapper
cls.tearDownClass = unique
# Do essentially the same thing for setup, but to ensure that
# doClassCleanups is only called if there is an exception in the
# setUpClass method.
original = cls.setUpClass
unique = functools.partial(_set_up_class_wrapper, original, cls)
functools.update_wrapper(unique, original, ['__name__', '__doc__'])
cls.setUpClass = unique
return unittest.TestCase.__new__(cls)
@classmethod
def addClassCleanup(cls, function, *args, **kwargs): # pylint: disable=invalid-name
cls.__stack.append((function, args, kwargs))
@classmethod
def doClassCleanups(cls): # pylint: disable=invalid-name
while cls.__stack:
func, args, kwargs = cls.__stack.pop()
# TODO: Should exceptions be ignored from this?
# TODO: addCleanups success if part of the success of the test,
# what should we do here?
func(*args, **kwargs)
|