Support copy of UUIDLegacy

behackett · behackett · commit f812d1babb7c · 2012-11-19T06:27:59.000-08:00
diff --git a/bson/binary.py b/bson/binary.py
@@ -214,6 +214,10 @@ def __new__(cls, obj):
         self.__uuid = obj
         return self
 
+    def __getnewargs__(self):
+        # Support copy and deepcopy
+        return (self.__uuid,)
+
     @property
     def uuid(self):
         """UUID instance wrapped by this UUIDLegacy instance.
diff --git a/test/test_binary.py b/test/test_binary.py
@@ -15,6 +15,7 @@
 """Tests for the Binary wrapper."""
 
 import base64
+import copy
 import pickle
 import sys
 import unittest
@@ -23,6 +24,7 @@
     should_test_uuid = True
 except ImportError:
     should_test_uuid = False
+
 sys.path[0:0] = [""]
 
 import bson
@@ -292,6 +294,15 @@ def test_pickle(self):
         for proto in xrange(pickle.HIGHEST_PROTOCOL + 1):
             self.assertEqual(b1, pickle.loads(pickle.dumps(b1, proto)))
 
+        if should_test_uuid:
+            uu = uuid.uuid4()
+            uul = UUIDLegacy(uu)
+
+            self.assertEqual(uul, copy.copy(uul))
+            self.assertEqual(uul, copy.deepcopy(uul))
+
+            for proto in xrange(pickle.HIGHEST_PROTOCOL + 1):
+                self.assertEqual(uul, pickle.loads(pickle.dumps(uul, proto)))
 
 if __name__ == "__main__":
     unittest.main()
