From 0639dbd47c1284c4d02a321f9158fd746ace8853 Mon Sep 17 00:00:00 2001
From: Johannes 'josch' Schauer <josch@mister-muffin.de>
Date: Mon, 24 Dec 2018 18:04:40 +0100
Subject: [PATCH] src/tests/__init__.py: fix CommandLineTests on python3

---
 src/tests/__init__.py | 30 +++++++++++++++++++-----------
 1 file changed, 19 insertions(+), 11 deletions(-)

diff --git a/src/tests/__init__.py b/src/tests/__init__.py
index 6c9fdd1..a64890f 100644
--- a/src/tests/__init__.py
+++ b/src/tests/__init__.py
@@ -435,21 +435,29 @@ def tiff_header_for_ccitt(width, height, img_size, ccitt_group=4):
 
 class CommandLineTests(unittest.TestCase):
     def test_main_help(self):
-        # silence output
-        sys_stdout = sys.stdout
         if PY3:
-            sys.stdout = TextIOWrapper(BytesIO())
+            from contextlib import redirect_stdout
+            f = StringIO()
+            with redirect_stdout(f):
+                try:
+                    img2pdf.main(['img2pdf', '--help'])
+                except SystemExit:
+                    pass
+            res = f.getvalue()
+            self.assertIn('img2pdf', res)
         else:
+            # silence output
+            sys_stdout = sys.stdout
             sys.stdout = BytesIO()
 
-        try:
-            img2pdf.main(['img2pdf', '--help'])
-        except SystemExit:
-            # argparse does sys.exit(0) on --help
-            res = sys.stdout.getvalue()
-            self.assertIn('img2pdf', res)
-        finally:
-            sys.stdout = sys_stdout
+            try:
+                img2pdf.main(['img2pdf', '--help'])
+            except SystemExit:
+                # argparse does sys.exit(0) on --help
+                res = sys.stdout.getvalue()
+                self.assertIn('img2pdf', res)
+            finally:
+                sys.stdout = sys_stdout
 
 
 def test_suite():