Merge "Implement DesignateObject.__deepcopy__"

designate/objects/base.py

@@ -12,6 +12,8 @@
 #    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
 #    License for the specific language governing permissions and limitations
 #    under the License.
+import copy
+
 import six
 
 from designate.openstack.common import importutils
@@ -215,6 +217,24 @@ class DesignateObject(DictObjectMixin):
         else:
             self._obj_changes.clear()
 
+    def __deepcopy__(self, memodict={}):
+        """
+        Efficiently make a deep copy of this object.
+
+        "Efficiently" is used here a relative term, this will be faster
+        than allowing python to naively deepcopy the object.
+        """
+        c_obj = self.__class__()
+
+        for field in self.FIELDS:
+            if self.obj_attr_is_set(field):
+                c_field = copy.deepcopy(getattr(self, field), memodict)
+                setattr(c_obj, field, c_field)
+
+        c_obj._obj_changes = set(self._obj_changes)
+
+        return c_obj
+
     def __iter__(self):
         # Redundant?
         self._i = iter(self.FIELDS)

designate/tests/test_objects/test_base.py

@@ -13,6 +13,8 @@
 # WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
 # License for the specific language governing permissions and limitations
 # under the License.
+import copy
+
 import testtools
 
 from designate.openstack.common import log as logging
@@ -204,3 +206,23 @@ class DesignateObjectTest(tests.TestCase):
 
         self.assertEqual(1, len(obj.obj_what_changed()))
         self.assertEqual({'name': "My Name"}, obj.obj_get_changes())
+
+    def test_deepcopy(self):
+        # Create the Original object
+        o_obj = TestObject()
+        o_obj.id = "My ID"
+        o_obj.name = "My Name"
+
+        # Clear the "changed" flag for one of the two fields we set
+        o_obj.obj_reset_changes(['name'])
+
+        # Deepcopy the object
+        c_obj = copy.deepcopy(o_obj)
+
+        # Ensure the copy was sucessful
+        self.assertEqual(o_obj.id, c_obj.id)
+        self.assertEqual(o_obj.name, c_obj.name)
+        self.assertEqual(o_obj.nested, c_obj.nested)
+
+        self.assertEqual(o_obj.obj_get_changes(), c_obj.obj_get_changes())
+        self.assertEqual(o_obj.to_primitive(), c_obj.to_primitive())