Merge pull request #330 from jborean93/stderr-warning

Raise warning rather than print

CHANGELOG.md

@@ -2,6 +2,7 @@
 
 ### Version 0.4.3
 - Fix invalid regex escape sequences.
+- Decoding CLIXML failures for `run_ps` will create a `UserWarning` rather than printing the warning.
 
 ### Version 0.4.2
 - Dropped Python 3.5 from support matrix as it is EOL.

setup.py

@@ -1,6 +1,6 @@
 from setuptools import setup, find_packages
 
-__version__ = '0.4.2'
+__version__ = '0.4.3'
 project_name = 'pywinrm'
 
 # PyPi supports only reStructuredText, so pandoc should be installed

winrm/__init__.py

@@ -2,6 +2,7 @@
 import re
 from base64 import b64encode
 import xml.etree.ElementTree as ET
+import warnings
 
 from winrm.protocol import Protocol
 
@@ -79,9 +80,9 @@ def _clean_error_msg(self, msg):
             except Exception as e:
                 # if any of the above fails, the msg was not true xml
                 # print a warning and return the original string
-                # TODO do not print, raise user defined error instead
-                print("Warning: there was a problem converting the Powershell"
-                      " error message: %s" % (e))
+                warnings.warn(
+                    "There was a problem converting the Powershell error "
+                    "message: %s" % (e))
             else:
                 # if new_msg was populated, that's our error message
                 # otherwise the original error message will be used

winrm/tests/test_session.py

@@ -1,3 +1,5 @@
+import pytest
+
 from winrm import Session
 
 
@@ -78,3 +80,13 @@ def test_decode_clixml_no_errors():
     expected = msg
     actual = s._clean_error_msg(msg)
     assert actual == expected
+
+
+def test_decode_clixml_invalid_xml():
+    s = Session('windows-host.example.com', auth=('john.smith', 'secret'))
+    msg = b'#< CLIXML\r\n<in >dasf<?dsfij>'
+
+    with pytest.warns(UserWarning, match="There was a problem converting the Powershell error message"):
+        actual = s._clean_error_msg(msg)
+
+    assert actual == msg