Merge pull request google#64 from google/reload

dcbrandao · web-flow · commit ea1850381bd4 · 2019-03-29T12:21:10.000-04:00
Set start_values on model reload
diff --git a/spanner_orm/model.py b/spanner_orm/model.py
@@ -569,9 +569,15 @@ def reload(self, transaction: spanner_transaction.Transaction = None
     updated_object = self._metaclass.find(transaction, **self.id())
     if updated_object is None:
       return None
+    start_values = {}
+
     for column in self._columns:
+      value = getattr(updated_object, column)
+      start_values[column] = copy.copy(value)
       if column not in self._primary_keys:
-        setattr(self, column, getattr(updated_object, column))
+        setattr(self, column, value)
+
+    self.start_values = start_values
     self._persisted = True
     return self
 
diff --git a/spanner_orm/tests/model_test.py b/spanner_orm/tests/model_test.py
@@ -148,6 +148,7 @@ def test_reload_reloads(self, find):
     find.return_value = models.SmallTestModel(updated_values)
     model.reload()
     self.assertEqual(model.value_1, updated_values['value_1'])
+    self.assertEqual(model.changes(), {})
 
   @mock.patch('spanner_orm.model.ModelApi.create')
   def test_save_creates(self, create):
