Tests of our bytes and unicode custom contracts

1 files changed, 29 insertions, 1 deletions

diff --git a/tests/test_misc.py b/tests/test_misc.py
index 83a2aec..d3dede4 100644
--- a/tests/test_misc.py
+++ b/tests/test_misc.py
@@ -5,9 +5,11 @@
 
 import sys
 
+import pytest
+
 import coverage
 from coverage.version import _make_url, _make_version
-from coverage.misc import Hasher, file_be_gone
+from coverage.misc import contract, Hasher, file_be_gone
 
 from tests.coveragetest import CoverageTest
 
@@ -69,6 +71,32 @@ class RemoveFileTest(CoverageTest):
             file_be_gone(".")
 
 
+class ContractTest(CoverageTest):
+    """Tests of our contract decorators."""
+
+    run_in_temp_dir = False
+
+    def test_bytes(self):
+        @contract(text='bytes|None')
+        def need_bytes(text):
+            return text
+
+        assert need_bytes(b"Hey") == b"Hey"
+        assert need_bytes(None) == None
+        with pytest.raises(Exception):
+            need_bytes(u"Oops")
+
+    def test_unicode(self):
+        @contract(text='unicode|None')
+        def need_unicode(text):
+            return text
+
+        assert need_unicode(u"Hey") == u"Hey"
+        assert need_unicode(None) == None
+        with pytest.raises(Exception):
+            need_unicode(b"Oops")
+
+
 class VersionTest(CoverageTest):
     """Tests of version.py"""