Merge pull request #109 from jsfehler/master

Read and write binary for screenshots

Author: bubenkoff

CHANGES.rst

@@ -1,6 +1,12 @@
 Changelog
 =========
 
+1.9.1
+-----
+
+- Fix utf-8 decode warnings when taking screenshots with pytest-xdist active `#108 <https://github.com/pytest-dev/pytest-splinter/issues/108>`_ (jsfehler) 
+
+
 1.9.0
 -----
 

pytest_splinter/plugin.py

@@ -401,7 +401,7 @@ def _take_screenshot(
             reprtraceback.extraline = _screenshot_extraline(screenshot_png_path, screenshot_html_path)
         if slaveoutput is not None:
             with codecs.open(screenshot_html_path, encoding=splinter_screenshot_encoding) as html_fd:
-                with open(screenshot_png_path) as fd:
+                with open(screenshot_png_path, 'rb') as fd:
                     slaveoutput.setdefault('screenshots', []).append({
                         'class_name': classname,
                         'files': [
@@ -623,7 +623,7 @@ def pytest_testnodedown(self, node, error):
                 os.makedirs(screenshot_dir)
             for fil in screenshot['files']:
                 encoding = fil.get('encoding')
-                with codecs.open(os.path.join(screenshot_dir, fil['file_name']), 'w',
+                with codecs.open(os.path.join(screenshot_dir, fil['file_name']), 'wb',
                                  **dict(encoding=encoding) if encoding else {}) as fd:
                     fd.write(fil['content'])