Adding back dataset's connection in Entity.reload.

googleapis · Jan 6, 2015 · c8e145f · c8e145f
1 parent 62204e7
commit c8e145f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/gcloud/datastore/entity.py b/gcloud/datastore/entity.py
@@ -213,7 +213,8 @@ def reload(self):
           exist only locally.
         """
         key = self._must_key
-        entity = key.get()
+        connection = self._must_dataset.connection()
+        entity = key.get(connection=connection)
 
         if entity:
             self.update(entity)

diff --git a/gcloud/datastore/test_entity.py b/gcloud/datastore/test_entity.py
@@ -264,7 +264,8 @@ def is_partial(self):
     def path(self):
         return self._path
 
-    def get(self):
+    def get(self, connection=None):
+        self._connection_used = connection
         return self._stored