diff options
-rw-r--r-- | src/lib/Bcfg2/Compat.py | 9 | ||||
-rw-r--r-- | testsuite/Testlib/TestClient/TestTools/TestPOSIX/TestFile.py | 1 |
2 files changed, 7 insertions, 3 deletions
diff --git a/src/lib/Bcfg2/Compat.py b/src/lib/Bcfg2/Compat.py index a245d3634..820dd9d1e 100644 --- a/src/lib/Bcfg2/Compat.py +++ b/src/lib/Bcfg2/Compat.py @@ -92,9 +92,12 @@ except: unicode = str # base64 compat -from base64 import b64encode as _b64encode, b64decode as _b64decode -b64encode = lambda s: _b64encode(s.encode('ascii')).decode('ascii') -b64decode = lambda s: _b64decode(s.encode('ascii')).decode('ascii') +if sys.hexversion >= 0x03000000: + from base64 import b64encode as _b64encode, b64decode as _b64decode + b64encode = lambda s: _b64encode(s.encode('UTF-8')).decode('UTF-8') + b64decode = lambda s: _b64decode(s.encode('UTF-8')).decode('UTF-8') +else: + from base64 import b64encode, b64decode try: input = raw_input diff --git a/testsuite/Testlib/TestClient/TestTools/TestPOSIX/TestFile.py b/testsuite/Testlib/TestClient/TestTools/TestPOSIX/TestFile.py index 4f073e443..cdf11ce5e 100644 --- a/testsuite/Testlib/TestClient/TestTools/TestPOSIX/TestFile.py +++ b/testsuite/Testlib/TestClient/TestTools/TestPOSIX/TestFile.py @@ -311,6 +311,7 @@ class TestPOSIXFile(TestPOSIXTool): entry = reset() entry.text = u("tëst") encoded = entry.text.encode(setup['encoding']) + mock_diff.return_value = ["-test2", "+tëst"] mock_get_data.return_value = (encoded, False) ptool._get_diffs(entry, interactive=True) mock_open.assert_called_with(entry.get("name")) |