def _list_column_families_helper(self, column_family_name=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' timeout_seconds = 502 cluster_name = ('projects/' + project_id + '/zones/' + zone + '/clusters/' + cluster_id) client = _Client(timeout_seconds=timeout_seconds) cluster = _Cluster(cluster_name, client=client) table = self._makeOne(table_id, cluster) # Create request_pb table_name = cluster_name + '/tables/' + table_id request_pb = messages_pb2.GetTableRequest(name=table_name) # Create response_pb column_family_id = 'foo' if column_family_name is None: column_family_name = (table_name + '/columnFamilies/' + column_family_id) column_family = data_pb2.ColumnFamily(name=column_family_name) response_pb = data_pb2.Table( column_families={column_family_id: column_family}, ) # Patch the stub used by the API method. client._table_stub = stub = _FakeStub(response_pb) # Create expected_result. expected_result = { column_family_id: table.column_family(column_family_id), } # Perform the method and check the result. result = table.list_column_families() self.assertEqual(result, expected_result) self.assertEqual(stub.method_calls, [( 'GetTable', (request_pb, timeout_seconds), {}, )])
def _list_tables_helper(self, table_id, table_name=None): from gcloud.bigtable._generated import ( bigtable_table_data_pb2 as table_data_pb2) from gcloud.bigtable._generated import ( bigtable_table_service_messages_pb2 as table_messages_pb2) from gcloud.bigtable._testing import _FakeStub project = 'PROJECT' zone = 'zone' cluster_id = 'cluster-id' timeout_seconds = 45 client = _Client(project, timeout_seconds=timeout_seconds) cluster = self._makeOne(zone, cluster_id, client) # Create request_ cluster_name = ('projects/' + project + '/zones/' + zone + '/clusters/' + cluster_id) request_pb = table_messages_pb2.ListTablesRequest(name=cluster_name) # Create response_pb table_name = table_name or (cluster_name + '/tables/' + table_id) response_pb = table_messages_pb2.ListTablesResponse( tables=[ table_data_pb2.Table(name=table_name), ], ) # Patch the stub used by the API method. client._table_stub = stub = _FakeStub(response_pb) # Create expected_result. expected_table = cluster.table(table_id) expected_result = [expected_table] # Perform the method and check the result. result = cluster.list_tables() self.assertEqual(result, expected_result) self.assertEqual(stub.method_calls, [( 'ListTables', (request_pb, timeout_seconds), {}, )])
def _create_test_helper(self, initial_split_keys): 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' timeout_seconds = 150 cluster_name = ('projects/' + project_id + '/zones/' + zone + '/clusters/' + cluster_id) client = _Client(timeout_seconds=timeout_seconds) cluster = _Cluster(cluster_name, client=client) table = self._makeOne(table_id, cluster) # Create request_pb request_pb = messages_pb2.CreateTableRequest( initial_split_keys=initial_split_keys, name=cluster_name, table_id=table_id, ) # Create response_pb response_pb = data_pb2.Table() # 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. result = table.create(initial_split_keys=initial_split_keys) self.assertEqual(result, expected_result) self.assertEqual(stub.method_calls, [( 'CreateTable', (request_pb, timeout_seconds), {}, )])