Adding back dataset's connection in Entity.reload.

Author: dhermes

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)

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