Adding Bigtable Cluster location on create() request. (googleapis#3646)

landrito · Aug 21, 2017 · 7ee1d88 · 7ee1d88
1 parent c1eb49d
commit 7ee1d88
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/bigtable/google/cloud/bigtable/cluster.py b/bigtable/google/cloud/bigtable/cluster.py
@@ -49,6 +49,7 @@ def _prepare_create_request(cluster):
         parent=cluster._instance.name,
         cluster_id=cluster.cluster_id,
         cluster=data_v2_pb2.Cluster(
+            location=cluster.location,
             serve_nodes=cluster.serve_nodes,
         ),
     )

diff --git a/bigtable/tests/unit/test_cluster.py b/bigtable/tests/unit/test_cluster.py
@@ -393,12 +393,14 @@ def test_it(self):
         instance = _Instance(INSTANCE_ID, client)
         cluster = Cluster(CLUSTER_ID, instance,
                           serve_nodes=SERVE_NODES)
+        cluster.location = u'projects/prahj-ekt/locations/zona-tres'
 
         request_pb = self._call_fut(cluster)
 
         self.assertEqual(request_pb.cluster_id, CLUSTER_ID)
         self.assertEqual(request_pb.parent, instance.name)
         self.assertEqual(request_pb.cluster.serve_nodes, SERVE_NODES)
+        self.assertEqual(request_pb.cluster.location, cluster.location)
 
 
 def _ClusterPB(*args, **kw):