Merge pull request benoitc#1664 from hramezani/test_to_bytestring_util

Add some tests for `to_bytestring` util
mjjbell · Dec 11, 2017 · d7cc56e · d7cc56e
2 parents 123b33e + 360f71e
commit d7cc56e
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/tests/test_util.py b/tests/test_util.py
@@ -59,3 +59,13 @@ def test_import_app():
         util.import_app('support:wrong_app')
     msg = "Failed to find application object 'wrong_app' in 'support'"
     assert msg in str(err)
+
+
+def test_to_bytestring():
+    assert util.to_bytestring('test_str', 'ascii') == b'test_str'
+    assert util.to_bytestring('test_str®') == b'test_str\xc2\xae'
+    assert util.to_bytestring(b'byte_test_str') == b'byte_test_str'
+    with pytest.raises(TypeError) as err:
+        util.to_bytestring(100)
+    msg = '100 is not a string'
+    assert msg in str(err)