add test to verify api is being copied in hash

1 files changed, 11 insertions, 0 deletions

diff --git a/tests/primitives/test_hashes.py b/tests/primitives/test_hashes.py
index d721f7af..07e2d8e7 100644
--- a/tests/primitives/test_hashes.py
+++ b/tests/primitives/test_hashes.py
@@ -13,6 +13,8 @@
 
 from __future__ import absolute_import, division, print_function
 
+import pretend
+
 import pytest
 
 import six
@@ -35,6 +37,15 @@ class TestBaseHash(object):
         assert isinstance(m.hexdigest(), str)
 
 
+class TestCopyHash(object):
+    def test_copy_api_object(self):
+        pretend_api = pretend.stub(copy_hash_context=lambda a: "copiedctx")
+        pretend_ctx = pretend.stub()
+        h = hashes.SHA1(api=pretend_api, ctx=pretend_ctx)
+        assert h._api is pretend_api
+        assert h.copy()._api is h._api
+
+
 class TestDefaultAPISHA1(object):
     def test_default_api_creation(self):
         """