aboutsummaryrefslogtreecommitdiffstats
path: root/test/test_utils.py
diff options
context:
space:
mode:
authorMaximilian Hils <git@maximilianhils.com>2015-09-12 17:03:09 +0200
committerMaximilian Hils <git@maximilianhils.com>2015-09-12 17:03:09 +0200
commit997fcde8ce94be9d8decddd4bc783106dbb41ab3 (patch)
treec668d1e49dcc9298bed2a85de136e7f053d72b17 /test/test_utils.py
parenta38142d5950a899c6e3f854841a45f4785515761 (diff)
downloadmitmproxy-997fcde8ce94be9d8decddd4bc783106dbb41ab3.tar.gz
mitmproxy-997fcde8ce94be9d8decddd4bc783106dbb41ab3.tar.bz2
mitmproxy-997fcde8ce94be9d8decddd4bc783106dbb41ab3.zip
make clean_bin unicode-aware
Diffstat (limited to 'test/test_utils.py')
-rw-r--r--test/test_utils.py15
1 files changed, 11 insertions, 4 deletions
diff --git a/test/test_utils.py b/test/test_utils.py
index 374d09ba..9dba5d35 100644
--- a/test/test_utils.py
+++ b/test/test_utils.py
@@ -15,10 +15,17 @@ def test_hexdump():
def test_cleanBin():
- assert utils.cleanBin("one") == "one"
- assert utils.cleanBin("\00ne") == ".ne"
- assert utils.cleanBin("\nne") == "\nne"
- assert utils.cleanBin("\nne", True) == ".ne"
+ assert utils.clean_bin(b"one") == b"one"
+ assert utils.clean_bin(b"\00ne") == b".ne"
+ assert utils.clean_bin(b"\nne") == b"\nne"
+ assert utils.clean_bin(b"\nne", False) == b".ne"
+ assert utils.clean_bin(u"\u2605".encode("utf8")) == b"..."
+
+ assert utils.clean_bin(u"one") == u"one"
+ assert utils.clean_bin(u"\00ne") == u".ne"
+ assert utils.clean_bin(u"\nne") == u"\nne"
+ assert utils.clean_bin(u"\nne", False) == u".ne"
+ assert utils.clean_bin(u"\u2605") == u"\u2605"
def test_pretty_size():