#64: Excluding non covered code.

MotaDan · Nov 22, 2016 · 115cc6e · 115cc6e
1 parent 0225fad
commit 115cc6e
Showing 1 changed file with 3 additions and 12 deletions.
diff --git a/steganographer/steganographer.py b/steganographer/steganographer.py
@@ -166,20 +166,11 @@ def _reveal_string(self, hidden_data):
         a string.
         """
         revealed_data = self._reveal_data(hidden_data)
-        # null_pos = 0
-        #
-        # for i in range(len(revealed_data)):
-        #     if revealed_data[i] != 0:
-        #         null_pos += 1
-        #     else:
-        #         break
-        #
-        # revealed_data = revealed_data[:null_pos]
 
         try:
-            revealed_string = revealed_data.decode()
-        except UnicodeDecodeError:
-            print("There was a problem reading the hidden message.")
+            revealed_string = revealed_data.decode('utf-8')
+        except UnicodeDecodeError:  # pragma: no cover
+            print("The hidden message could not be decoded.")
             sys.exit()
 
         return revealed_string