Make sure to clean up even if an exception occurs (#50620)

This commit is contained in:
Matt Martz 2019-01-08 09:03:49 -06:00 committed by GitHub
parent 2dbade4adc
commit 9a520b26e3
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -61,8 +61,9 @@ def setup_env(filename):
setattr(module.AnsibleModule, '__init__', fake)
setattr(module, '_load_params', _fake_load_params)
try:
yield fake
finally:
setattr(module.AnsibleModule, '__init__', _original_init)
setattr(module, '_load_params', _original_load_params)