def test_generate_consistency_token(self):
# Setup Expected Response
consistency_token = 'consistencyToken-1090516718'
expected_response = {'consistency_token': consistency_token}
expected_response = bigtable_table_admin_pb2.GenerateConsistencyTokenResponse(
**expected_response)
# Mock the API response
channel = ChannelStub(responses=[expected_response])
patch = mock.patch('google.api_core.grpc_helpers.create_channel')
with patch as create_channel:
create_channel.return_value = channel
client = bigtable_admin_v2.BigtableTableAdminClient()
# Setup Request
name = client.table_path('[PROJECT]', '[INSTANCE]', '[TABLE]')
response = client.generate_consistency_token(name)
assert expected_response == response
assert len(channel.requests) == 1
expected_request = bigtable_table_admin_pb2.GenerateConsistencyTokenRequest(
name=name)
actual_request = channel.requests[0][1]
assert expected_request == actual_request
def test_test_iam_permissions(self):
# Setup Expected Response
expected_response = {}
expected_response = iam_policy_pb2.TestIamPermissionsResponse(
**expected_response)
# Mock the API response
channel = ChannelStub(responses=[expected_response])
patch = mock.patch("google.api_core.grpc_helpers.create_channel")
with patch as create_channel:
create_channel.return_value = channel
client = bigtable_admin_v2.BigtableTableAdminClient()
# Setup Request
resource = "resource-341064690"
permissions = []
response = client.test_iam_permissions(resource, permissions)
assert expected_response == response
assert len(channel.requests) == 1
expected_request = iam_policy_pb2.TestIamPermissionsRequest(
resource=resource, permissions=permissions)
actual_request = channel.requests[0][1]
assert expected_request == actual_request
def test_list_backups(self):
# Setup Expected Response
next_page_token = ""
backups_element = {}
backups = [backups_element]
expected_response = {
"next_page_token": next_page_token,
"backups": backups
}
expected_response = bigtable_table_admin_pb2.ListBackupsResponse(
**expected_response)
# Mock the API response
channel = ChannelStub(responses=[expected_response])
patch = mock.patch("google.api_core.grpc_helpers.create_channel")
with patch as create_channel:
create_channel.return_value = channel
client = bigtable_admin_v2.BigtableTableAdminClient()
# Setup Request
parent = client.cluster_path("[PROJECT]", "[INSTANCE]", "[CLUSTER]")
paged_list_response = client.list_backups(parent)
resources = list(paged_list_response)
assert len(resources) == 1
assert expected_response.backups[0] == resources[0]
assert len(channel.requests) == 1
expected_request = bigtable_table_admin_pb2.ListBackupsRequest(
parent=parent)
actual_request = channel.requests[0][1]
assert expected_request == actual_request
def test_list_tables(self):
# Setup Expected Response
next_page_token = ''
tables_element = {}
tables = [tables_element]
expected_response = {
'next_page_token': next_page_token,
'tables': tables
}
expected_response = bigtable_table_admin_pb2.ListTablesResponse(
**expected_response)
# Mock the API response
channel = ChannelStub(responses=[expected_response])
client = bigtable_admin_v2.BigtableTableAdminClient(channel=channel)
# Setup Request
parent = client.instance_path('[PROJECT]', '[INSTANCE]')
paged_list_response = client.list_tables(parent)
resources = list(paged_list_response)
assert len(resources) == 1
assert expected_response.tables[0] == resources[0]
assert len(channel.requests) == 1
expected_request = bigtable_table_admin_pb2.ListTablesRequest(
parent=parent)
actual_request = channel.requests[0][1]
assert expected_request == actual_request
def test_update_backup(self):
# Setup Expected Response
name = "name3373707"
source_table = "sourceTable1670858410"
size_bytes = 1796325715
expected_response = {
"name": name,
"source_table": source_table,
"size_bytes": size_bytes,
}
expected_response = table_pb2.Backup(**expected_response)
# Mock the API response
channel = ChannelStub(responses=[expected_response])
patch = mock.patch("google.api_core.grpc_helpers.create_channel")
with patch as create_channel:
create_channel.return_value = channel
client = bigtable_admin_v2.BigtableTableAdminClient()
# Setup Request
backup = {}
update_mask = {}
response = client.update_backup(backup, update_mask)
assert expected_response == response
assert len(channel.requests) == 1
expected_request = bigtable_table_admin_pb2.UpdateBackupRequest(
backup=backup, update_mask=update_mask)
actual_request = channel.requests[0][1]
assert expected_request == actual_request
def test_get_backup(self):
# Setup Expected Response
name_2 = "name2-1052831874"
source_table = "sourceTable1670858410"
size_bytes = 1796325715
expected_response = {
"name": name_2,
"source_table": source_table,
"size_bytes": size_bytes,
}
expected_response = table_pb2.Backup(**expected_response)
# Mock the API response
channel = ChannelStub(responses=[expected_response])
patch = mock.patch("google.api_core.grpc_helpers.create_channel")
with patch as create_channel:
create_channel.return_value = channel
client = bigtable_admin_v2.BigtableTableAdminClient()
# Setup Request
name = client.backup_path("[PROJECT]", "[INSTANCE]", "[CLUSTER]",
"[BACKUP]")
response = client.get_backup(name)
assert expected_response == response
assert len(channel.requests) == 1
expected_request = bigtable_table_admin_pb2.GetBackupRequest(name=name)
actual_request = channel.requests[0][1]
assert expected_request == actual_request
def test_get_snapshot(self):
# Setup Expected Response
name_2 = 'name2-1052831874'
data_size_bytes = 2110122398
description = 'description-1724546052'
expected_response = {
'name': name_2,
'data_size_bytes': data_size_bytes,
'description': description
}
expected_response = table_pb2.Snapshot(**expected_response)
# Mock the API response
channel = ChannelStub(responses=[expected_response])
client = bigtable_admin_v2.BigtableTableAdminClient(channel=channel)
# Setup Request
name = client.snapshot_path('[PROJECT]', '[INSTANCE]', '[CLUSTER]',
'[SNAPSHOT]')
response = client.get_snapshot(name)
assert expected_response == response
assert len(channel.requests) == 1
expected_request = bigtable_table_admin_pb2.GetSnapshotRequest(
name=name)
actual_request = channel.requests[0][1]
assert expected_request == actual_request
def test_create_table_from_snapshot(self):
# Setup Expected Response
name = "name3373707"
expected_response = {"name": name}
expected_response = table_pb2.Table(**expected_response)
operation = operations_pb2.Operation(
name="operations/test_create_table_from_snapshot", done=True)
operation.response.Pack(expected_response)
# Mock the API response
channel = ChannelStub(responses=[operation])
patch = mock.patch("google.api_core.grpc_helpers.create_channel")
with patch as create_channel:
create_channel.return_value = channel
client = bigtable_admin_v2.BigtableTableAdminClient()
# Setup Request
parent = client.instance_path("[PROJECT]", "[INSTANCE]")
table_id = "tableId-895419604"
source_snapshot = client.snapshot_path("[PROJECT]", "[INSTANCE]",
"[CLUSTER]", "[SNAPSHOT]")
response = client.create_table_from_snapshot(parent, table_id,
source_snapshot)
result = response.result()
assert expected_response == result
assert len(channel.requests) == 1
expected_request = bigtable_table_admin_pb2.CreateTableFromSnapshotRequest(
parent=parent, table_id=table_id, source_snapshot=source_snapshot)
actual_request = channel.requests[0][1]
assert expected_request == actual_request
def test_create_table(self):
# Setup Expected Response
name = "name3373707"
expected_response = {"name": name}
expected_response = table_pb2.Table(**expected_response)
# Mock the API response
channel = ChannelStub(responses=[expected_response])
patch = mock.patch("google.api_core.grpc_helpers.create_channel")
with patch as create_channel:
create_channel.return_value = channel
client = bigtable_admin_v2.BigtableTableAdminClient()
# Setup Request
parent = client.instance_path("[PROJECT]", "[INSTANCE]")
table_id = "tableId-895419604"
table = {}
response = client.create_table(parent, table_id, table)
assert expected_response == response
assert len(channel.requests) == 1
expected_request = bigtable_table_admin_pb2.CreateTableRequest(
parent=parent, table_id=table_id, table=table)
actual_request = channel.requests[0][1]
assert expected_request == actual_request
def test_set_iam_policy(self):
# Setup Expected Response
version = 351608024
etag = b"etag3123477"
expected_response = {"version": version, "etag": etag}
expected_response = policy_pb2.Policy(**expected_response)
# Mock the API response
channel = ChannelStub(responses=[expected_response])
patch = mock.patch("google.api_core.grpc_helpers.create_channel")
with patch as create_channel:
create_channel.return_value = channel
client = bigtable_admin_v2.BigtableTableAdminClient()
# Setup Request
resource = "resource-341064690"
policy = {}
response = client.set_iam_policy(resource, policy)
assert expected_response == response
assert len(channel.requests) == 1
expected_request = iam_policy_pb2.SetIamPolicyRequest(
resource=resource, policy=policy)
actual_request = channel.requests[0][1]
assert expected_request == actual_request
def test_check_consistency(self):
# Setup Expected Response
consistent = True
expected_response = {"consistent": consistent}
expected_response = bigtable_table_admin_pb2.CheckConsistencyResponse(
**expected_response)
# Mock the API response
channel = ChannelStub(responses=[expected_response])
patch = mock.patch("google.api_core.grpc_helpers.create_channel")
with patch as create_channel:
create_channel.return_value = channel
client = bigtable_admin_v2.BigtableTableAdminClient()
# Setup Request
name = client.table_path("[PROJECT]", "[INSTANCE]", "[TABLE]")
consistency_token = "consistencyToken-1090516718"
response = client.check_consistency(name, consistency_token)
assert expected_response == response
assert len(channel.requests) == 1
expected_request = bigtable_table_admin_pb2.CheckConsistencyRequest(
name=name, consistency_token=consistency_token)
actual_request = channel.requests[0][1]
assert expected_request == actual_request
def test_list_snapshots(self):
# Setup Expected Response
next_page_token = ''
snapshots_element = {}
snapshots = [snapshots_element]
expected_response = {
'next_page_token': next_page_token,
'snapshots': snapshots
}
expected_response = bigtable_table_admin_pb2.ListSnapshotsResponse(
**expected_response)
# Mock the API response
channel = ChannelStub(responses=[expected_response])
client = bigtable_admin_v2.BigtableTableAdminClient(channel=channel)
# Setup Request
parent = client.cluster_path('[PROJECT]', '[INSTANCE]', '[CLUSTER]')
paged_list_response = client.list_snapshots(parent)
resources = list(paged_list_response)
assert len(resources) == 1
assert expected_response.snapshots[0] == resources[0]
assert len(channel.requests) == 1
expected_request = bigtable_table_admin_pb2.ListSnapshotsRequest(
parent=parent)
actual_request = channel.requests[0][1]
assert expected_request == actual_request
def test_modify_column_families(self):
# Setup Expected Response
name_2 = "name2-1052831874"
expected_response = {"name": name_2}
expected_response = table_pb2.Table(**expected_response)
# Mock the API response
channel = ChannelStub(responses=[expected_response])
patch = mock.patch("google.api_core.grpc_helpers.create_channel")
with patch as create_channel:
create_channel.return_value = channel
client = bigtable_admin_v2.BigtableTableAdminClient()
# Setup Request
name = client.table_path("[PROJECT]", "[INSTANCE]", "[TABLE]")
modifications = []
response = client.modify_column_families(name, modifications)
assert expected_response == response
assert len(channel.requests) == 1
expected_request = bigtable_table_admin_pb2.ModifyColumnFamiliesRequest(
name=name, modifications=modifications)
actual_request = channel.requests[0][1]
assert expected_request == actual_request
def test_create_table_from_snapshot_exception(self):
# Setup Response
error = status_pb2.Status()
operation = operations_pb2.Operation(
name="operations/test_create_table_from_snapshot_exception",
done=True)
operation.error.CopyFrom(error)
# Mock the API response
channel = ChannelStub(responses=[operation])
patch = mock.patch("google.api_core.grpc_helpers.create_channel")
with patch as create_channel:
create_channel.return_value = channel
client = bigtable_admin_v2.BigtableTableAdminClient()
# Setup Request
parent = client.instance_path("[PROJECT]", "[INSTANCE]")
table_id = "tableId-895419604"
source_snapshot = client.snapshot_path("[PROJECT]", "[INSTANCE]",
"[CLUSTER]", "[SNAPSHOT]")
response = client.create_table_from_snapshot(parent, table_id,
source_snapshot)
exception = response.exception()
assert exception.errors[0] == error
def test_create_table_from_snapshot(self):
# Setup Expected Response
name = 'name3373707'
expected_response = {'name': name}
expected_response = table_pb2.Table(**expected_response)
operation = operations_pb2.Operation(
name='operations/test_create_table_from_snapshot', done=True)
operation.response.Pack(expected_response)
# Mock the API response
channel = ChannelStub(responses=[operation])
client = bigtable_admin_v2.BigtableTableAdminClient(channel=channel)
# Setup Request
parent = client.instance_path('[PROJECT]', '[INSTANCE]')
table_id = 'tableId-895419604'
source_snapshot = 'sourceSnapshot-947679896'
response = client.create_table_from_snapshot(parent, table_id,
source_snapshot)
result = response.result()
assert expected_response == result
assert len(channel.requests) == 1
expected_request = bigtable_table_admin_pb2.CreateTableFromSnapshotRequest(
parent=parent, table_id=table_id, source_snapshot=source_snapshot)
actual_request = channel.requests[0][1]
assert expected_request == actual_request
def test_get_snapshot(self):
# Setup Expected Response
name_2 = "name2-1052831874"
data_size_bytes = 2110122398
description = "description-1724546052"
expected_response = {
"name": name_2,
"data_size_bytes": data_size_bytes,
"description": description,
}
expected_response = table_pb2.Snapshot(**expected_response)
# Mock the API response
channel = ChannelStub(responses=[expected_response])
patch = mock.patch("google.api_core.grpc_helpers.create_channel")
with patch as create_channel:
create_channel.return_value = channel
client = bigtable_admin_v2.BigtableTableAdminClient()
# Setup Request
name = client.snapshot_path("[PROJECT]", "[INSTANCE]", "[CLUSTER]",
"[SNAPSHOT]")
response = client.get_snapshot(name)
assert expected_response == response
assert len(channel.requests) == 1
expected_request = bigtable_table_admin_pb2.GetSnapshotRequest(
name=name)
actual_request = channel.requests[0][1]
assert expected_request == actual_request
def test_drop_row_range_exception(self):
# Mock the API response
channel = ChannelStub(responses=[CustomException()])
client = bigtable_admin_v2.BigtableTableAdminClient(channel=channel)
# Setup request
name = client.table_path('[PROJECT]', '[INSTANCE]', '[TABLE]')
with pytest.raises(CustomException):
client.drop_row_range(name)
def test_list_snapshots_exception(self):
channel = ChannelStub(responses=[CustomException()])
client = bigtable_admin_v2.BigtableTableAdminClient(channel=channel)
# Setup request
parent = client.cluster_path('[PROJECT]', '[INSTANCE]', '[CLUSTER]')
paged_list_response = client.list_snapshots(parent)
with pytest.raises(CustomException):
list(paged_list_response)
def test_generate_consistency_token_exception(self):
# Mock the API response
channel = ChannelStub(responses=[CustomException()])
client = bigtable_admin_v2.BigtableTableAdminClient(channel=channel)
# Setup request
name = client.table_path('[PROJECT]', '[INSTANCE]', '[TABLE]')
with pytest.raises(CustomException):
client.generate_consistency_token(name)
def test_modify_column_families_exception(self):
# Mock the API response
channel = ChannelStub(responses=[CustomException()])
client = bigtable_admin_v2.BigtableTableAdminClient(channel=channel)
# Setup request
name = client.table_path('[PROJECT]', '[INSTANCE]', '[TABLE]')
modifications = []
with pytest.raises(CustomException):
client.modify_column_families(name, modifications)
def test_delete_snapshot_exception(self):
# Mock the API response
channel = ChannelStub(responses=[CustomException()])
client = bigtable_admin_v2.BigtableTableAdminClient(channel=channel)
# Setup request
name = client.snapshot_path('[PROJECT]', '[INSTANCE]', '[CLUSTER]',
'[SNAPSHOT]')
with pytest.raises(CustomException):
client.delete_snapshot(name)
def test_check_consistency_exception(self):
# Mock the API response
channel = ChannelStub(responses=[CustomException()])
client = bigtable_admin_v2.BigtableTableAdminClient(channel=channel)
# Setup request
name = client.table_path('[PROJECT]', '[INSTANCE]', '[TABLE]')
consistency_token = 'consistencyToken-1090516718'
with pytest.raises(CustomException):
client.check_consistency(name, consistency_token)
def test_create_table_exception(self):
# Mock the API response
channel = ChannelStub(responses=[CustomException()])
client = bigtable_admin_v2.BigtableTableAdminClient(channel=channel)
# Setup request
parent = client.instance_path('[PROJECT]', '[INSTANCE]')
table_id = 'tableId-895419604'
table = {}
with pytest.raises(CustomException):
client.create_table(parent, table_id, table)
def _table_admin_client(self):
"""Getter for the gRPC stub used for the Table Admin API.
:rtype: :class:`.bigtable_admin_pb2.BigtableTableAdmin`
:returns: A BigtableTableAdmin instance.
:raises: :class:`ValueError <exceptions.ValueError>` if the current
client is not an admin client or if it has not been
:meth:`start`-ed.
"""
if not self._admin:
raise ValueError('Client is not an admin client.')
return bigtable_admin_v2.BigtableTableAdminClient(
channel=self._channel, credentials=self._credentials)
def test_list_tables_exception(self):
channel = ChannelStub(responses=[CustomException()])
patch = mock.patch('google.api_core.grpc_helpers.create_channel')
with patch as create_channel:
create_channel.return_value = channel
client = bigtable_admin_v2.BigtableTableAdminClient()
# Setup request
parent = client.instance_path('[PROJECT]', '[INSTANCE]')
paged_list_response = client.list_tables(parent)
with pytest.raises(CustomException):
list(paged_list_response)
def test_generate_consistency_token_exception(self):
# Mock the API response
channel = ChannelStub(responses=[CustomException()])
patch = mock.patch("google.api_core.grpc_helpers.create_channel")
with patch as create_channel:
create_channel.return_value = channel
client = bigtable_admin_v2.BigtableTableAdminClient()
# Setup request
name = client.table_path("[PROJECT]", "[INSTANCE]", "[TABLE]")
with pytest.raises(CustomException):
client.generate_consistency_token(name)
def test_get_iam_policy_exception(self):
# Mock the API response
channel = ChannelStub(responses=[CustomException()])
patch = mock.patch("google.api_core.grpc_helpers.create_channel")
with patch as create_channel:
create_channel.return_value = channel
client = bigtable_admin_v2.BigtableTableAdminClient()
# Setup request
resource = "resource-341064690"
with pytest.raises(CustomException):
client.get_iam_policy(resource)
def test_drop_row_range_exception(self):
# Mock the API response
channel = ChannelStub(responses=[CustomException()])
patch = mock.patch('google.api_core.grpc_helpers.create_channel')
with patch as create_channel:
create_channel.return_value = channel
client = bigtable_admin_v2.BigtableTableAdminClient()
# Setup request
name = client.table_path('[PROJECT]', '[INSTANCE]', '[TABLE]')
with pytest.raises(CustomException):
client.drop_row_range(name)
def test_list_backups_exception(self):
channel = ChannelStub(responses=[CustomException()])
patch = mock.patch("google.api_core.grpc_helpers.create_channel")
with patch as create_channel:
create_channel.return_value = channel
client = bigtable_admin_v2.BigtableTableAdminClient()
# Setup request
parent = client.cluster_path("[PROJECT]", "[INSTANCE]", "[CLUSTER]")
paged_list_response = client.list_backups(parent)
with pytest.raises(CustomException):
list(paged_list_response)
def test_drop_row_range(self):
channel = ChannelStub()
client = bigtable_admin_v2.BigtableTableAdminClient(channel=channel)
# Setup Request
name = client.table_path('[PROJECT]', '[INSTANCE]', '[TABLE]')
client.drop_row_range(name)
assert len(channel.requests) == 1
expected_request = bigtable_table_admin_pb2.DropRowRangeRequest(
name=name)
actual_request = channel.requests[0][1]
assert expected_request == actual_request
