diff options
Diffstat (limited to 'lib/fixtures/_fixtures/popen.py')
-rw-r--r-- | lib/fixtures/_fixtures/popen.py | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/lib/fixtures/_fixtures/popen.py b/lib/fixtures/_fixtures/popen.py index 708c5fd..80629fd 100644 --- a/lib/fixtures/_fixtures/popen.py +++ b/lib/fixtures/_fixtures/popen.py @@ -48,6 +48,12 @@ class FakeProcess(object): err = '' return out, err + def __enter__(self): + return self + + def __exit__(self, exc_type, exc_value, traceback): + self.wait() + def wait(self): if self.returncode is None: self.communicate() |