Implementing Bigtable ColumnFamily.create().

Author: dhermes

gcloud/bigtable/column_family.py

@@ -19,6 +19,8 @@
 
 from gcloud._helpers import _total_seconds
 from gcloud.bigtable._generated import bigtable_table_data_pb2 as data_pb2
+from gcloud.bigtable._generated import (
+    bigtable_table_service_messages_pb2 as messages_pb2)
 from gcloud.bigtable._generated import duration_pb2
 
 
@@ -193,6 +195,10 @@ def to_pb(self):
 class ColumnFamily(object):
     """Representation of a Google Cloud Bigtable Column Family.
 
+    We can use a :class:`ColumnFamily` to:
+
+    * :meth:`create` itself
+
     :type column_family_id: str
     :param column_family_id: The ID of the column family. Must be of the
                              form ``[_a-zA-Z0-9][-_.a-zA-Z0-9]*``.
@@ -238,6 +244,22 @@ def __eq__(self, other):
     def __ne__(self, other):
         return not self.__eq__(other)
 
+    def create(self):
+        """Create this column family."""
+        if self.gc_rule is None:
+            column_family = data_pb2.ColumnFamily()
+        else:
+            column_family = data_pb2.ColumnFamily(gc_rule=self.gc_rule.to_pb())
+        request_pb = messages_pb2.CreateColumnFamilyRequest(
+            name=self._table.name,
+            column_family_id=self.column_family_id,
+            column_family=column_family,
+        )
+        client = self._table._cluster._client
+        # We expect a `.data_pb2.ColumnFamily`
+        client._table_stub.CreateColumnFamily(request_pb,
+                                              client.timeout_seconds)
+
 
 def _gc_rule_from_pb(gc_rule_pb):
     """Convert a protobuf GC rule to a native object.

gcloud/bigtable/test_column_family.py

@@ -395,6 +395,65 @@ def test___ne__(self):
         column_family2 = self._makeOne('column_family_id2', None)
         self.assertNotEqual(column_family1, column_family2)
 
+    def _create_test_helper(self, gc_rule=None):
+        from gcloud.bigtable._generated import (
+            bigtable_table_data_pb2 as data_pb2)
+        from gcloud.bigtable._generated import (
+            bigtable_table_service_messages_pb2 as messages_pb2)
+        from gcloud.bigtable._testing import _FakeStub
+
+        project_id = 'project-id'
+        zone = 'zone'
+        cluster_id = 'cluster-id'
+        table_id = 'table-id'
+        column_family_id = 'column-family-id'
+        timeout_seconds = 4
+        table_name = ('projects/' + project_id + '/zones/' + zone +
+                      '/clusters/' + cluster_id + '/tables/' + table_id)
+
+        client = _Client(timeout_seconds=timeout_seconds)
+        table = _Table(table_name, client=client)
+        column_family = self._makeOne(column_family_id, table, gc_rule=gc_rule)
+
+        # Create request_pb
+        if gc_rule is None:
+            column_family_pb = data_pb2.ColumnFamily()
+        else:
+            column_family_pb = data_pb2.ColumnFamily(gc_rule=gc_rule.to_pb())
+        request_pb = messages_pb2.CreateColumnFamilyRequest(
+            name=table_name,
+            column_family_id=column_family_id,
+            column_family=column_family_pb,
+        )
+
+        # Create response_pb
+        response_pb = data_pb2.ColumnFamily()
+
+        # Patch the stub used by the API method.
+        client._table_stub = stub = _FakeStub(response_pb)
+
+        # Create expected_result.
+        expected_result = None  # create() has no return value.
+
+        # Perform the method and check the result.
+        self.assertEqual(stub.results, (response_pb,))
+        result = column_family.create()
+        self.assertEqual(stub.results, ())
+        self.assertEqual(result, expected_result)
+        self.assertEqual(stub.method_calls, [(
+            'CreateColumnFamily',
+            (request_pb, timeout_seconds),
+            {},
+        )])
+
+    def test_create(self):
+        self._create_test_helper(gc_rule=None)
+
+    def test_create_with_gc_rule(self):
+        from gcloud.bigtable.column_family import MaxVersionsGCRule
+        gc_rule = MaxVersionsGCRule(1337)
+        self._create_test_helper(gc_rule=gc_rule)
+
 
 class Test__gc_rule_from_pb(unittest2.TestCase):
 
@@ -471,7 +530,20 @@ def WhichOneof(cls, name):
         self.assertEqual(MockProto.names, ['rule'])
 
 
+class _Cluster(object):
+
+    def __init__(self, client=None):
+        self._client = client
+
+
+class _Client(object):
+
+    def __init__(self, timeout_seconds=None):
+        self.timeout_seconds = timeout_seconds
+
+
 class _Table(object):
 
-    def __init__(self, name):
+    def __init__(self, name, client=None):
         self.name = name
+        self._cluster = _Cluster(client)