Support deepcopy of DBRef PYTHON-169

Author: RedBeard0531

bson/dbref.py

@@ -15,6 +15,7 @@
 """Tools for manipulating DBRefs (references to MongoDB documents)."""
 
 from bson.son import SON
+from copy import deepcopy
 
 
 class DBRef(object):
@@ -115,3 +116,13 @@ def __hash__(self):
  """
  return hash((self.__collection, self.__id,
  self.__database, self.__kwargs))
+
+ def __deepcopy__(self, memo):
+ """Support function for `copy.deepcopy()`.
+
+ .. versionadded:: 1.10
+ """
+ return DBRef(deepcopy(self.__collection, memo),
+ deepcopy(self.__id, memo),
+ deepcopy(self.__database, memo),
+ deepcopy(self.__kwargs, memo))

test/test_dbref.py

@@ -21,6 +21,8 @@
 from bson.objectid import ObjectId
 from bson.dbref import DBRef
 
+from copy import deepcopy
+
 
 class TestDBRef(unittest.TestCase):
 
@@ -90,6 +92,21 @@ def test_kwargs(self):
  self.assertEqual("bar", DBRef("coll", 5, foo="bar").foo)
  self.assertRaises(KeyError, getattr, DBRef("coll", 5, foo="bar"), "bar")
 
+ def test_deepcopy(self):
+ a = DBRef('coll', 'asdf', 'db', x=[1])
+ b = deepcopy(a)
+
+ self.assertEqual(a, b)
+ self.assertNotEqual(id(a), id(b.x))
+ self.assertEqual(a.x, b.x)
+ self.assertNotEqual(id(a.x), id(b.x))
+
+ b.x[0] = 2
+ self.assertEqual(a.x, [1])
+ self.assertEqual(b.x, [2])
+
+
+
 
 if __name__ == "__main__":
  unittest.main()