capture: recover from closed files

Author: xoviat

_pytest/capture.py

@@ -211,7 +211,8 @@ def readouterr(self):
 
     @contextlib.contextmanager
     def disabled(self):
-        capmanager = self.request.config.pluginmanager.getplugin('capturemanager')
+        capmanager = self.request.config.pluginmanager.getplugin(
+            'capturemanager')
         capmanager.suspendcapture_item(self.request.node, "call", in_=True)
         try:
             yield
@@ -248,12 +249,20 @@ def __init__(self, buffer, encoding):
     def write(self, obj):
         if isinstance(obj, unicode):
             obj = obj.encode(self.encoding, "replace")
-        self.buffer.write(obj)
+        try:
+            self.buffer.write(obj)
+        except OSError:
+            self.buffer = open(os.devnull, "wb+")
 
     def writelines(self, linelist):
         data = ''.join(linelist)
         self.write(data)
 
+    def fileno(self):
+        self.write('')
+
+        return self.buffer.fileno()
+
     @property
     def name(self):
         """Ensure that file.name is a string."""
@@ -371,9 +380,13 @@ def start(self):
         self.syscapture.start()
 
     def snap(self):
-        f = self.tmpfile
-        f.seek(0)
-        res = f.read()
+        try:
+            f = self.tmpfile
+            f.seek(0)
+            res = f.read()
+        except OSError:
+            self.tempfile = open(os.devnull, "r")
+            res = ''
         if res:
             enc = getattr(f, "encoding", None)
             if enc and isinstance(res, bytes):
@@ -398,7 +411,10 @@ def suspend(self):
 
     def resume(self):
         self.syscapture.resume()
-        os.dup2(self.tmpfile_fd, self.targetfd)
+        try:
+            os.dup2(self.tmpfile_fd, self.targetfd)
+        except OSError:
+            self.targetfd = open(os.devnull, "r").fileno()
 
     def writeorg(self, data):
         """ write to original file descriptor. """