Mercurial > public > mercurial-scm > hg-stable
diff tests/test-extensions-wrapfunction.py @ 34032:47e52f079a57
extensions: add wrappedfunction() context manager
Several extensions exist that temporarily want to wrap a function (at
least narrowhg, any many of the extensions in hg-experimental). That's
why we have the unwrapfunction() that was introduced in 19578bb84731
(extensions: add unwrapfunction to undo wrapfunction, 2016-08-10).
This patch adds a simple wrappedfunction() that returns a context
manager.
Differential Revision: https://phab.mercurial-scm.org/D472
| author | Martin von Zweigbergk <martinvonz@google.com> |
|---|---|
| date | Mon, 21 Aug 2017 16:46:05 -0700 |
| parents | 19578bb84731 |
| children | 82bd4c5a81e5 |
line wrap: on
line diff
--- a/tests/test-extensions-wrapfunction.py Tue Aug 29 18:20:50 2017 -0700 +++ b/tests/test-extensions-wrapfunction.py Mon Aug 21 16:46:05 2017 -0700 @@ -37,3 +37,20 @@ batchwrap(wrappers + [wrappers[0]]) batchunwrap([(wrappers[i] if i >= 0 else None) for i in [3, None, 0, 4, 0, 2, 1, None]]) + +wrap0 = extensions.wrappedfunction(dummy, 'getstack', wrappers[0]) +wrap1 = extensions.wrappedfunction(dummy, 'getstack', wrappers[1]) + +# Use them in a different order from how they were created to check that +# the wrapping happens in __enter__, not in __init__ +print('context manager', dummy.getstack()) +with wrap1: + print('context manager', dummy.getstack()) + with wrap0: + print('context manager', dummy.getstack()) + # Bad programmer forgets to unwrap the function, but the context + # managers still unwrap their wrappings. + extensions.wrapfunction(dummy, 'getstack', wrappers[2]) + print('context manager', dummy.getstack()) + print('context manager', dummy.getstack()) +print('context manager', dummy.getstack())