From 19c2c9c7f20d7a9f0ff6732b6986d688d9ee4bf1 Mon Sep 17 00:00:00 2001
From: Danny Hermes <daniel.j.hermes@gmail.com>
Date: Wed, 6 Jan 2016 20:59:27 -0800
Subject: [PATCH 1/2] Renaming dataset_id->project_id on PartitionId.

---
 gcloud/datastore/connection.py       |  4 ++--
 gcloud/datastore/helpers.py          | 20 ++++++++++----------
 gcloud/datastore/key.py              |  2 +-
 gcloud/datastore/test_batch.py       |  2 +-
 gcloud/datastore/test_client.py      |  4 ++--
 gcloud/datastore/test_connection.py  |  2 +-
 gcloud/datastore/test_helpers.py     | 20 ++++++++++----------
 gcloud/datastore/test_key.py         |  4 ++--
 gcloud/datastore/test_query.py       |  2 +-
 gcloud/datastore/test_transaction.py |  2 +-
 10 files changed, 31 insertions(+), 31 deletions(-)

diff --git a/gcloud/datastore/connection.py b/gcloud/datastore/connection.py
index cf81d5a19644..6a65f503c07e 100644
--- a/gcloud/datastore/connection.py
+++ b/gcloud/datastore/connection.py
@@ -398,10 +398,10 @@ def _prepare_key_for_request(key_pb):  # pragma: NO COVER copied from helpers
     :returns: A key which will be added to a request. It will be the
               original if nothing needs to be changed.
     """
-    if key_pb.partition_id.dataset_id:  # Simple field (string)
+    if key_pb.partition_id.project_id:  # Simple field (string)
         new_key_pb = _entity_pb2.Key()
         new_key_pb.CopyFrom(key_pb)
-        new_key_pb.partition_id.ClearField('dataset_id')
+        new_key_pb.partition_id.ClearField('project_id')
         key_pb = new_key_pb
     return key_pb
 
diff --git a/gcloud/datastore/helpers.py b/gcloud/datastore/helpers.py
index 9db96eb83705..cdb614c7225c 100644
--- a/gcloud/datastore/helpers.py
+++ b/gcloud/datastore/helpers.py
@@ -61,7 +61,7 @@ def find_true_project(project, connection):
     # the project so the backend won't complain.
     bogus_key_pb = Key('__MissingLookupKind', 1,
                        project=project).to_protobuf()
-    bogus_key_pb.partition_id.ClearField('dataset_id')
+    bogus_key_pb.partition_id.ClearField('project_id')
 
     found_pbs, missing_pbs, _ = connection.lookup(project, [bogus_key_pb])
     # By not passing in `deferred`, lookup will continue until
@@ -70,7 +70,7 @@ def find_true_project(project, connection):
     # We only asked for one, so should only receive one.
     returned_pb, = all_pbs
 
-    return returned_pb.key.partition_id.dataset_id
+    return returned_pb.key.partition_id.project_id
 
 
 def _get_meaning(value_pb, is_list=False):
@@ -267,8 +267,8 @@ def key_from_protobuf(pb):
             path_args.append(element.name)
 
     project = None
-    if pb.partition_id.dataset_id:  # Simple field (string)
-        project = pb.partition_id.dataset_id
+    if pb.partition_id.project_id:  # Simple field (string)
+        project = pb.partition_id.project_id
     namespace = None
     if pb.partition_id.namespace:  # Simple field (string)
         namespace = pb.partition_id.namespace
@@ -429,18 +429,18 @@ def _prepare_key_for_request(key_pb):
     :returns: A key which will be added to a request. It will be the
               original if nothing needs to be changed.
     """
-    if key_pb.partition_id.dataset_id:  # Simple field (string)
-        # We remove the dataset_id from the protobuf. This is because
+    if key_pb.partition_id.project_id:  # Simple field (string)
+        # We remove the project_id from the protobuf. This is because
         # the backend fails a request if the key contains un-prefixed
         # project. The backend fails because requests to
-        #     /datastore/.../datasets/foo/...
+        #     /v1beta3/projects/foo:...
         # and
-        #     /datastore/.../datasets/s~foo/...
+        #     /v1beta3/projects/s~foo:...
         # both go to the datastore given by 's~foo'. So if the key
-        # protobuf in the request body has dataset_id='foo', the
+        # protobuf in the request body has project_id='foo', the
         # backend will reject since 'foo' != 's~foo'.
         new_key_pb = _entity_pb2.Key()
         new_key_pb.CopyFrom(key_pb)
-        new_key_pb.partition_id.ClearField('dataset_id')
+        new_key_pb.partition_id.ClearField('project_id')
         key_pb = new_key_pb
     return key_pb
diff --git a/gcloud/datastore/key.py b/gcloud/datastore/key.py
index 7a8e2848b7f2..49a6c67d05b0 100644
--- a/gcloud/datastore/key.py
+++ b/gcloud/datastore/key.py
@@ -240,7 +240,7 @@ def to_protobuf(self):
         :returns: The protobuf representing the key.
         """
         key = _entity_pb2.Key()
-        key.partition_id.dataset_id = self.project
+        key.partition_id.project_id = self.project
 
         if self.namespace:
             key.partition_id.namespace = self.namespace
diff --git a/gcloud/datastore/test_batch.py b/gcloud/datastore/test_batch.py
index 316070bcfeb1..73ee4eda26d4 100644
--- a/gcloud/datastore/test_batch.py
+++ b/gcloud/datastore/test_batch.py
@@ -388,7 +388,7 @@ def to_protobuf(self):
         from gcloud.datastore._generated import entity_pb2
         key = self._key = entity_pb2.Key()
         # Don't assign it, because it will just get ripped out
-        # key.partition_id.dataset_id = self.project
+        # key.partition_id.project_id = self.project
 
         element = key.path_element.add()
         element.kind = self._kind
diff --git a/gcloud/datastore/test_client.py b/gcloud/datastore/test_client.py
index 90cbbad6183e..5549eb9de8c1 100644
--- a/gcloud/datastore/test_client.py
+++ b/gcloud/datastore/test_client.py
@@ -20,7 +20,7 @@ def _make_entity_pb(project, kind, integer_id, name=None, str_val=None):
     from gcloud.datastore.helpers import _new_value_pb
 
     entity_pb = entity_pb2.Entity()
-    entity_pb.key.partition_id.dataset_id = project
+    entity_pb.key.partition_id.project_id = project
     path_element = entity_pb.key.path_element.add()
     path_element.kind = kind
     path_element.id = integer_id
@@ -322,7 +322,7 @@ def test_get_multi_miss_w_missing(self):
 
         # Make a missing entity pb to be returned from mock backend.
         missed = entity_pb2.Entity()
-        missed.key.partition_id.dataset_id = self.PROJECT
+        missed.key.partition_id.project_id = self.PROJECT
         path_element = missed.key.path_element.add()
         path_element.kind = KIND
         path_element.id = ID
diff --git a/gcloud/datastore/test_connection.py b/gcloud/datastore/test_connection.py
index c04e397a2247..e5f86fe28846 100644
--- a/gcloud/datastore/test_connection.py
+++ b/gcloud/datastore/test_connection.py
@@ -857,7 +857,7 @@ def request(self, **kw):
 
 def _compare_key_pb_after_request(test, key_before, key_after):
     # Unset values are False-y.
-    test.assertEqual(key_after.partition_id.dataset_id, '')
+    test.assertEqual(key_after.partition_id.project_id, '')
     test.assertEqual(key_before.partition_id.namespace,
                      key_after.partition_id.namespace)
     test.assertEqual(len(key_before.path_element),
diff --git a/gcloud/datastore/test_helpers.py b/gcloud/datastore/test_helpers.py
index af746a154c9d..17e479b7530c 100644
--- a/gcloud/datastore/test_helpers.py
+++ b/gcloud/datastore/test_helpers.py
@@ -70,7 +70,7 @@ def test_it(self):
         _KIND = 'KIND'
         _ID = 1234
         entity_pb = entity_pb2.Entity()
-        entity_pb.key.partition_id.dataset_id = _PROJECT
+        entity_pb.key.partition_id.project_id = _PROJECT
         entity_pb.key.path_element.add(kind=_KIND, id=_ID)
 
         value_pb = _new_value_pb(entity_pb, 'foo')
@@ -117,7 +117,7 @@ def test_mismatched_value_indexed(self):
         _KIND = 'KIND'
         _ID = 1234
         entity_pb = entity_pb2.Entity()
-        entity_pb.key.partition_id.dataset_id = _PROJECT
+        entity_pb.key.partition_id.project_id = _PROJECT
         entity_pb.key.path_element.add(kind=_KIND, id=_ID)
 
         list_val_pb = _new_value_pb(entity_pb, 'baz')
@@ -173,7 +173,7 @@ def test_nested_entity_no_key(self):
         inside_val_pb.integer_value = INSIDE_VALUE
 
         entity_pb = entity_pb2.Entity()
-        entity_pb.key.partition_id.dataset_id = PROJECT
+        entity_pb.key.partition_id.project_id = PROJECT
         element = entity_pb.key.path_element.add()
         element.kind = KIND
 
@@ -235,7 +235,7 @@ def test_key_only(self):
         entity_pb = self._callFUT(entity)
 
         expected_pb = entity_pb2.Entity()
-        expected_pb.key.partition_id.dataset_id = project
+        expected_pb.key.partition_id.project_id = project
         path_elt = expected_pb.key.path_element.add()
         path_elt.kind = kind
         path_elt.name = name
@@ -279,7 +279,7 @@ def test_inverts_to_protobuf(self):
 
         original_pb = entity_pb2.Entity()
         # Add a key.
-        original_pb.key.partition_id.dataset_id = project = 'PROJECT'
+        original_pb.key.partition_id.project_id = project = 'PROJECT'
         elem1 = original_pb.key.path_element.add()
         elem1.kind = 'Family'
         elem1.id = 1234
@@ -322,7 +322,7 @@ def test_inverts_to_protobuf(self):
         new_pb = self._callFUT(entity)
 
         # NOTE: entity_to_protobuf() strips the project so we "cheat".
-        new_pb.key.partition_id.dataset_id = project
+        new_pb.key.partition_id.project_id = project
         self._compareEntityProto(original_pb, new_pb)
 
     def test_meaning_with_change(self):
@@ -355,7 +355,7 @@ def _makePB(self, project=None, namespace=None, path=()):
         from gcloud.datastore._generated import entity_pb2
         pb = entity_pb2.Key()
         if project is not None:
-            pb.partition_id.dataset_id = project
+            pb.partition_id.project_id = project
         if namespace is not None:
             pb.partition_id.namespace = namespace
         for elem in path:
@@ -563,7 +563,7 @@ def test_entity(self):
         pb = entity_pb2.Value()
         entity_pb = pb.entity_value
         entity_pb.key.path_element.add(kind='KIND')
-        entity_pb.key.partition_id.dataset_id = 'PROJECT'
+        entity_pb.key.partition_id.project_id = 'PROJECT'
 
         value_pb = _new_value_pb(entity_pb, 'foo')
         value_pb.string_value = 'Foo'
@@ -740,7 +740,7 @@ def _callFUT(self, key_pb):
     def test_prepare_project_valid(self):
         from gcloud.datastore._generated import entity_pb2
         key = entity_pb2.Key()
-        key.partition_id.dataset_id = 'foo'
+        key.partition_id.project_id = 'foo'
         new_key = self._callFUT(key)
         self.assertFalse(new_key is key)
 
@@ -909,7 +909,7 @@ def lookup(self, project, key_pbs):
 
         response = entity_pb2.Entity()
         response.key.CopyFrom(key_pb)
-        response.key.partition_id.dataset_id = self.prefix + project
+        response.key.partition_id.project_id = self.prefix + project
 
         missing = []
         deferred = []
diff --git a/gcloud/datastore/test_key.py b/gcloud/datastore/test_key.py
index 3b3d0a120849..5afdf37be7f4 100644
--- a/gcloud/datastore/test_key.py
+++ b/gcloud/datastore/test_key.py
@@ -341,7 +341,7 @@ def test_to_protobuf_defaults(self):
         self.assertTrue(isinstance(pb, entity_pb2.Key))
 
         # Check partition ID.
-        self.assertEqual(pb.partition_id.dataset_id, self._DEFAULT_PROJECT)
+        self.assertEqual(pb.partition_id.project_id, self._DEFAULT_PROJECT)
         # Unset values are False-y.
         self.assertEqual(pb.partition_id.namespace, '')
 
@@ -357,7 +357,7 @@ def test_to_protobuf_w_explicit_project(self):
         _PROJECT = 'PROJECT-ALT'
         key = self._makeOne('KIND', project=_PROJECT)
         pb = key.to_protobuf()
-        self.assertEqual(pb.partition_id.dataset_id, _PROJECT)
+        self.assertEqual(pb.partition_id.project_id, _PROJECT)
 
     def test_to_protobuf_w_explicit_namespace(self):
         _NAMESPACE = 'NAMESPACE'
diff --git a/gcloud/datastore/test_query.py b/gcloud/datastore/test_query.py
index a16140fefd12..4014c95c2936 100644
--- a/gcloud/datastore/test_query.py
+++ b/gcloud/datastore/test_query.py
@@ -354,7 +354,7 @@ def _addQueryResults(self, connection, cursor=_END, more=False):
         NO_MORE = query_pb2.QueryResultBatch.MORE_RESULTS_AFTER_LIMIT
         _ID = 123
         entity_pb = entity_pb2.Entity()
-        entity_pb.key.partition_id.dataset_id = self._PROJECT
+        entity_pb.key.partition_id.project_id = self._PROJECT
         path_element = entity_pb.key.path_element.add()
         path_element.kind = self._KIND
         path_element.id = _ID
diff --git a/gcloud/datastore/test_transaction.py b/gcloud/datastore/test_transaction.py
index 8260e1916f70..e98f2faca17f 100644
--- a/gcloud/datastore/test_transaction.py
+++ b/gcloud/datastore/test_transaction.py
@@ -166,7 +166,7 @@ def _make_key(kind, id_, project):
     from gcloud.datastore._generated import entity_pb2
 
     key = entity_pb2.Key()
-    key.partition_id.dataset_id = project
+    key.partition_id.project_id = project
     elem = key.path_element.add()
     elem.kind = kind
     elem.id = id_

From d55d58feca3bdd1d74c32298941f62ca8d69c02b Mon Sep 17 00:00:00 2001
From: Danny Hermes <daniel.j.hermes@gmail.com>
Date: Wed, 6 Jan 2016 21:00:46 -0800
Subject: [PATCH 2/2] Renaming namespace->namespace_id on PartitionId.

---
 gcloud/datastore/connection.py      |  2 +-
 gcloud/datastore/helpers.py         |  4 ++--
 gcloud/datastore/key.py             |  2 +-
 gcloud/datastore/test_connection.py | 12 ++++++------
 gcloud/datastore/test_helpers.py    |  2 +-
 gcloud/datastore/test_key.py        |  4 ++--
 6 files changed, 13 insertions(+), 13 deletions(-)

diff --git a/gcloud/datastore/connection.py b/gcloud/datastore/connection.py
index 6a65f503c07e..baa195ee4336 100644
--- a/gcloud/datastore/connection.py
+++ b/gcloud/datastore/connection.py
@@ -262,7 +262,7 @@ def run_query(self, project, query_pb, namespace=None,
         _set_read_options(request, eventual, transaction_id)
 
         if namespace:
-            request.partition_id.namespace = namespace
+            request.partition_id.namespace_id = namespace
 
         request.query.CopyFrom(query_pb)
         response = self._rpc(project, 'runQuery', request,
diff --git a/gcloud/datastore/helpers.py b/gcloud/datastore/helpers.py
index cdb614c7225c..0fa5321ba765 100644
--- a/gcloud/datastore/helpers.py
+++ b/gcloud/datastore/helpers.py
@@ -270,8 +270,8 @@ def key_from_protobuf(pb):
     if pb.partition_id.project_id:  # Simple field (string)
         project = pb.partition_id.project_id
     namespace = None
-    if pb.partition_id.namespace:  # Simple field (string)
-        namespace = pb.partition_id.namespace
+    if pb.partition_id.namespace_id:  # Simple field (string)
+        namespace = pb.partition_id.namespace_id
 
     return Key(*path_args, namespace=namespace, project=project)
 
diff --git a/gcloud/datastore/key.py b/gcloud/datastore/key.py
index 49a6c67d05b0..935f2ffa9935 100644
--- a/gcloud/datastore/key.py
+++ b/gcloud/datastore/key.py
@@ -243,7 +243,7 @@ def to_protobuf(self):
         key.partition_id.project_id = self.project
 
         if self.namespace:
-            key.partition_id.namespace = self.namespace
+            key.partition_id.namespace_id = self.namespace
 
         for item in self.path:
             element = key.path_element.add()
diff --git a/gcloud/datastore/test_connection.py b/gcloud/datastore/test_connection.py
index e5f86fe28846..34b3ca1c685c 100644
--- a/gcloud/datastore/test_connection.py
+++ b/gcloud/datastore/test_connection.py
@@ -483,7 +483,7 @@ def test_run_query_w_eventual_no_transaction(self):
         rq_class = datastore_pb2.RunQueryRequest
         request = rq_class()
         request.ParseFromString(cw['body'])
-        self.assertEqual(request.partition_id.namespace, '')
+        self.assertEqual(request.partition_id.namespace_id, '')
         self.assertEqual(request.query, q_pb)
         self.assertEqual(request.read_options.read_consistency,
                          datastore_pb2.ReadOptions.EVENTUAL)
@@ -522,7 +522,7 @@ def test_run_query_wo_eventual_w_transaction(self):
         rq_class = datastore_pb2.RunQueryRequest
         request = rq_class()
         request.ParseFromString(cw['body'])
-        self.assertEqual(request.partition_id.namespace, '')
+        self.assertEqual(request.partition_id.namespace_id, '')
         self.assertEqual(request.query, q_pb)
         self.assertEqual(request.read_options.read_consistency,
                          datastore_pb2.ReadOptions.DEFAULT)
@@ -577,7 +577,7 @@ def test_run_query_wo_namespace_empty_result(self):
         rq_class = datastore_pb2.RunQueryRequest
         request = rq_class()
         request.ParseFromString(cw['body'])
-        self.assertEqual(request.partition_id.namespace, '')
+        self.assertEqual(request.partition_id.namespace_id, '')
         self.assertEqual(request.query, q_pb)
 
     def test_run_query_w_namespace_nonempty_result(self):
@@ -607,7 +607,7 @@ def test_run_query_w_namespace_nonempty_result(self):
         rq_class = datastore_pb2.RunQueryRequest
         request = rq_class()
         request.ParseFromString(cw['body'])
-        self.assertEqual(request.partition_id.namespace, 'NS')
+        self.assertEqual(request.partition_id.namespace_id, 'NS')
         self.assertEqual(request.query, q_pb)
 
     def test_begin_transaction(self):
@@ -858,8 +858,8 @@ def request(self, **kw):
 def _compare_key_pb_after_request(test, key_before, key_after):
     # Unset values are False-y.
     test.assertEqual(key_after.partition_id.project_id, '')
-    test.assertEqual(key_before.partition_id.namespace,
-                     key_after.partition_id.namespace)
+    test.assertEqual(key_before.partition_id.namespace_id,
+                     key_after.partition_id.namespace_id)
     test.assertEqual(len(key_before.path_element),
                      len(key_after.path_element))
     for elt1, elt2 in zip(key_before.path_element, key_after.path_element):
diff --git a/gcloud/datastore/test_helpers.py b/gcloud/datastore/test_helpers.py
index 17e479b7530c..70aa50168907 100644
--- a/gcloud/datastore/test_helpers.py
+++ b/gcloud/datastore/test_helpers.py
@@ -357,7 +357,7 @@ def _makePB(self, project=None, namespace=None, path=()):
         if project is not None:
             pb.partition_id.project_id = project
         if namespace is not None:
-            pb.partition_id.namespace = namespace
+            pb.partition_id.namespace_id = namespace
         for elem in path:
             added = pb.path_element.add()
             added.kind = elem['kind']
diff --git a/gcloud/datastore/test_key.py b/gcloud/datastore/test_key.py
index 5afdf37be7f4..ea68c5bc3913 100644
--- a/gcloud/datastore/test_key.py
+++ b/gcloud/datastore/test_key.py
@@ -343,7 +343,7 @@ def test_to_protobuf_defaults(self):
         # Check partition ID.
         self.assertEqual(pb.partition_id.project_id, self._DEFAULT_PROJECT)
         # Unset values are False-y.
-        self.assertEqual(pb.partition_id.namespace, '')
+        self.assertEqual(pb.partition_id.namespace_id, '')
 
         # Check the element PB matches the partial key and kind.
         elem, = list(pb.path_element)
@@ -364,7 +364,7 @@ def test_to_protobuf_w_explicit_namespace(self):
         key = self._makeOne('KIND', namespace=_NAMESPACE,
                             project=self._DEFAULT_PROJECT)
         pb = key.to_protobuf()
-        self.assertEqual(pb.partition_id.namespace, _NAMESPACE)
+        self.assertEqual(pb.partition_id.namespace_id, _NAMESPACE)
 
     def test_to_protobuf_w_explicit_path(self):
         _PARENT = 'PARENT'