diff options
| author | Barry Warsaw | 2017-01-01 17:25:29 -0500 |
|---|---|---|
| committer | Barry Warsaw | 2017-01-01 17:28:37 -0500 |
| commit | e3d6c34b7d02b925cb8c59fa9e1df24741fc46ee (patch) | |
| tree | e6e27c9dec4557dc46c4b8ffd99bf556da4ed4ed /src/mailman/testing | |
| parent | c786a77c3b82932facd63fabe3144f9099f1ea51 (diff) | |
| download | mailman-e3d6c34b7d02b925cb8c59fa9e1df24741fc46ee.tar.gz mailman-e3d6c34b7d02b925cb8c59fa9e1df24741fc46ee.tar.zst mailman-e3d6c34b7d02b925cb8c59fa9e1df24741fc46ee.zip | |
Diffstat (limited to 'src/mailman/testing')
| -rw-r--r-- | src/mailman/testing/documentation.py | 10 | ||||
| -rw-r--r-- | src/mailman/testing/helpers.py | 7 |
2 files changed, 9 insertions, 8 deletions
diff --git a/src/mailman/testing/documentation.py b/src/mailman/testing/documentation.py index 97e30e900..8670f05ef 100644 --- a/src/mailman/testing/documentation.py +++ b/src/mailman/testing/documentation.py @@ -21,7 +21,7 @@ Note that doctest extraction does not currently work for zip file distributions. doctest discovery currently requires file system traversal. """ -from inspect import isfunction, ismethod +from contextlib import ExitStack from mailman.app.lifecycle import create_list from mailman.config import config from mailman.testing.helpers import ( @@ -153,13 +153,9 @@ def setup(testobj): testobj.globs['subscribe'] = subscribe testobj.globs['transaction'] = config.db # Add this so that cleanups can be automatically added by the doctest. - testobj.globs['cleanups'] = [] + testobj.globs['cleanups'] = ExitStack() @public def teardown(testobj): - for cleanup in testobj.globs['cleanups']: - if isfunction(cleanup) or ismethod(cleanup): - cleanup() - else: - cleanup[0](*cleanup[1:]) + testobj.globs['cleanups'].close() diff --git a/src/mailman/testing/helpers.py b/src/mailman/testing/helpers.py index 233b03cf8..290f7e026 100644 --- a/src/mailman/testing/helpers.py +++ b/src/mailman/testing/helpers.py @@ -245,7 +245,7 @@ def get_nntp_server(cleanups): """ patcher = mock.patch('nntplib.NNTP') server_class = patcher.start() - cleanups.append(patcher.stop) + cleanups.callback(patcher.stop) nntpd = server_class() # A class for more convenient access to the posted message. class NNTPProxy: # noqa: E306 @@ -480,6 +480,11 @@ def reset_the_world(): getUtility(IStyleManager).populate() # Remove all dynamic header-match rules. config.chains['header-match'].flush() + # Remove cached organizational domain suffix file. + from mailman.rules.dmarc import LOCAL_FILE_NAME + suffix_file = os.path.join(config.VAR_DIR, LOCAL_FILE_NAME) + with suppress(FileNotFoundError): + os.remove(suffix_file) @public |
