def test_list_resource_descriptors(self):
RESOURCE_TYPE = 'pubsub_topic'
client = monitoring.Client()
descriptor = None
for item in client.list_resource_descriptors():
if item.type == RESOURCE_TYPE:
descriptor = item
self.assertIsNotNone(descriptor)
expected_name = 'projects/{project}/{what}/{type}'.format(
project=client.project,
what='monitoredResourceDescriptors',
type=RESOURCE_TYPE,
)
self.assertEqual(descriptor.name, expected_name)
self.assertEqual(descriptor.type, RESOURCE_TYPE)
self.assertTrue(descriptor.display_name)
self.assertTrue(descriptor.description)
self.assertTrue(descriptor.labels)
for label in descriptor.labels:
self.assertTrue(label.key)
self.assertTrue(label.value_type)
self.assertTrue(label.description)
def process(keyfile, config, project_id, list_resources, list_metrics, query,
metric_id, days, hours, minutes, resource_filter, metric_filter,
align, reduce, reduce_grouping, iloc00):
if not project_id:
error('--project not specified')
if not keyfile:
# --keyfile not specified, use interactive `gcloud auth login`
client = monitoring.Client(project=project_id)
else:
_file = keyfile
# file is relative to config (if present)
if config:
_file = os.path.join(os.path.split(config)[0], keyfile)
client = monitoring.Client.from_service_account_json(
_file, project=project_id)
if list_resources:
list_resource_descriptors(client)
elif list_metrics:
list_metric_descriptors(client)
elif query:
perform_query(client, metric_id, days, hours, minutes, resource_filter,
metric_filter, align, reduce, reduce_grouping, iloc00)
else:
error(
'No operation specified. Please choose one of --list-resources, --list-metrics, --query'
)
def test_list_metric_descriptors(self):
METRIC_TYPE = (
'pubsub.googleapis.com/topic/send_message_operation_count')
METRIC_KIND = monitoring.MetricKind.DELTA
VALUE_TYPE = monitoring.ValueType.INT64
client = monitoring.Client()
descriptor = None
for item in client.list_metric_descriptors():
if item.type == METRIC_TYPE:
descriptor = item
self.assertIsNotNone(descriptor)
expected_name = 'projects/{project}/{what}/{type}'.format(
project=client.project,
what='metricDescriptors',
type=METRIC_TYPE,
)
self.assertEqual(descriptor.name, expected_name)
self.assertEqual(descriptor.type, METRIC_TYPE)
self.assertEqual(descriptor.metric_kind, METRIC_KIND)
self.assertEqual(descriptor.value_type, VALUE_TYPE)
self.assertTrue(descriptor.description)
self.assertTrue(descriptor.labels)
for label in descriptor.labels:
self.assertTrue(label.key)
self.assertTrue(label.value_type)
self.assertTrue(label.description)
def test_list_metric_descriptors(self):
client = monitoring.Client()
descriptor = None
for item in client.list_metric_descriptors():
if item.type == METRIC_TYPE:
descriptor = item
self.assertIsNotNone(descriptor)
expected_name = 'projects/{project}/{what}/{type}'.format(
project=client.project,
what='metricDescriptors',
type=METRIC_TYPE,
)
self.assertEqual(descriptor.name, expected_name)
self.assertEqual(descriptor.type, METRIC_TYPE)
self.assertEqual(descriptor.metric_kind, METRIC_KIND)
self.assertEqual(descriptor.value_type, VALUE_TYPE)
self.assertTrue(descriptor.description)
self.assertTrue(descriptor.labels)
for label in descriptor.labels:
self.assertTrue(label.key)
self.assertTrue(label.value_type)
self.assertTrue(label.description)
def test_query(self):
METRIC_TYPE = (
'pubsub.googleapis.com/topic/send_message_operation_count')
client = monitoring.Client()
query = client.query(METRIC_TYPE, hours=1)
# There may be no data, but we can ask anyway.
for _ in query:
pass # Not necessarily reached.
def test_create_group(self):
client = monitoring.Client()
group = client.group(
display_name=self.DISPLAY_NAME,
filter_string=self.FILTER,
is_cluster=self.IS_CLUSTER,
)
group.create()
self.to_delete.append(group)
self.assertTrue(group.exists())
def test_list_metric_descriptors_filtered(self):
client = monitoring.Client()
PREFIX = 'compute.googleapis.com/'
descriptors = client.list_metric_descriptors(type_prefix=PREFIX)
# There are currently 18 types with this prefix, but that may change.
self.assertGreater(len(descriptors), 10)
for descriptor in descriptors:
self.assertTrue(descriptor.type.startswith(PREFIX))
def test_list_group_members(self):
client = monitoring.Client()
group = client.group(
display_name=self.DISPLAY_NAME,
filter_string=self.FILTER,
is_cluster=self.IS_CLUSTER,
)
group.create()
self.to_delete.append(group)
for member in group.list_members():
self.assertIsInstance(member, monitoring.Resource)
def test_reload_group(self):
client = monitoring.Client()
group = client.group(
display_name=self.DISPLAY_NAME,
filter_string=self.FILTER,
is_cluster=self.IS_CLUSTER,
)
group.create()
self.to_delete.append(group)
group.filter = 'resource.type = "aws_ec2_instance"'
group.display_name = 'locally changed name'
group.reload()
self.assertEqual(group.filter, self.FILTER)
self.assertEqual(group.display_name, self.DISPLAY_NAME)
def test_list_groups(self):
client = monitoring.Client()
new_group = client.group(
display_name=self.DISPLAY_NAME,
filter_string=self.FILTER,
is_cluster=self.IS_CLUSTER,
)
before_groups = client.list_groups()
before_names = set(group.name for group in before_groups)
new_group.create()
self.to_delete.append(new_group)
self.assertTrue(new_group.exists())
after_groups = client.list_groups()
after_names = set(group.name for group in after_groups)
self.assertEqual(after_names - before_names, set([new_group.name]))
def test_group_hierarchy(self):
client = monitoring.Client()
root_group = client.group(
display_name='Testing: Root group',
filter_string=self.FILTER,
)
retry_503(root_group.create)()
self.to_delete.insert(0, root_group)
middle_group = client.group(
display_name='Testing: Middle group',
filter_string=self.FILTER,
parent_id=root_group.id,
)
retry_503(middle_group.create)()
self.to_delete.insert(0, middle_group)
leaf_group = client.group(
display_name='Testing: Leaf group',
filter_string=self.FILTER,
parent_id=middle_group.id,
)
retry_503(leaf_group.create)()
self.to_delete.insert(0, leaf_group)
# Test for parent.
actual_parent = middle_group.fetch_parent()
self.assertTrue(actual_parent.name, root_group.name)
# Test for children.
actual_children = middle_group.list_children()
children_names = [group.name for group in actual_children]
self.assertEqual(children_names, [leaf_group.name])
# Test for descendants.
actual_descendants = root_group.list_descendants()
descendant_names = {group.name for group in actual_descendants}
self.assertEqual(descendant_names,
set([middle_group.name, leaf_group.name]))
# Test for ancestors.
actual_ancestors = leaf_group.list_ancestors()
ancestor_names = [group.name for group in actual_ancestors]
self.assertEqual(ancestor_names, [middle_group.name, root_group.name])
def test_create_and_delete_metric_descriptor(self):
METRIC_TYPE = ('custom.googleapis.com/tmp/system_test_example' +
unique_resource_id())
METRIC_KIND = monitoring.MetricKind.GAUGE
VALUE_TYPE = monitoring.ValueType.DOUBLE
DESCRIPTION = 'System test example -- DELETE ME!'
client = monitoring.Client()
descriptor = client.metric_descriptor(
METRIC_TYPE,
metric_kind=METRIC_KIND,
value_type=VALUE_TYPE,
description=DESCRIPTION,
)
descriptor.create()
descriptor.delete()
with self.assertRaises(NotFound):
descriptor.delete()
def test_fetch_resource_descriptor(self):
client = monitoring.Client()
descriptor = client.fetch_resource_descriptor(RESOURCE_TYPE)
expected_name = 'projects/{project}/{what}/{type}'.format(
project=client.project,
what='monitoredResourceDescriptors',
type=RESOURCE_TYPE,
)
self.assertEqual(descriptor.name, expected_name)
self.assertEqual(descriptor.type, RESOURCE_TYPE)
self.assertTrue(descriptor.display_name)
self.assertTrue(descriptor.description)
self.assertTrue(descriptor.labels)
for label in descriptor.labels:
self.assertTrue(label.key)
self.assertTrue(label.value_type)
self.assertTrue(label.description)
def test_fetch_metric_descriptor(self):
client = monitoring.Client()
descriptor = client.fetch_metric_descriptor(METRIC_TYPE)
expected_name = 'projects/{project}/metricDescriptors/{type}'.format(
project=client.project,
type=METRIC_TYPE,
)
self.assertEqual(descriptor.name, expected_name)
self.assertEqual(descriptor.type, METRIC_TYPE)
self.assertEqual(descriptor.metric_kind, METRIC_KIND)
self.assertEqual(descriptor.value_type, VALUE_TYPE)
self.assertTrue(descriptor.description)
self.assertTrue(descriptor.labels)
for label in descriptor.labels:
self.assertTrue(label.key)
self.assertTrue(label.value_type)
self.assertTrue(label.description)
def test_update_group(self):
NEW_FILTER = 'resource.type = "aws_ec2_instance"'
NEW_DISPLAY_NAME = 'updated'
client = monitoring.Client()
group = client.group(
display_name=self.DISPLAY_NAME,
filter_string=self.FILTER,
is_cluster=self.IS_CLUSTER,
)
group.create()
self.to_delete.append(group)
group.filter = NEW_FILTER
group.display_name = NEW_DISPLAY_NAME
group.update()
after = client.fetch_group(group.id)
self.assertEqual(after.filter, NEW_FILTER)
self.assertEqual(after.display_name, NEW_DISPLAY_NAME)
def test_fetch_metric_descriptor(self):
METRIC_TYPE = (
'pubsub.googleapis.com/topic/send_message_operation_count')
METRIC_KIND = monitoring.MetricKind.DELTA
VALUE_TYPE = monitoring.ValueType.INT64
client = monitoring.Client()
descriptor = client.fetch_metric_descriptor(METRIC_TYPE)
expected_name = 'projects/{project}/metricDescriptors/{type}'.format(
project=client.project,
type=METRIC_TYPE,
)
self.assertEqual(descriptor.name, expected_name)
self.assertEqual(descriptor.type, METRIC_TYPE)
self.assertEqual(descriptor.metric_kind, METRIC_KIND)
self.assertEqual(descriptor.value_type, VALUE_TYPE)
self.assertTrue(descriptor.description)
self.assertTrue(descriptor.labels)
for label in descriptor.labels:
self.assertTrue(label.key)
self.assertTrue(label.value_type)
self.assertTrue(label.description)
def test_query(self):
client = monitoring.Client()
query = client.query(METRIC_TYPE, hours=1)
# There may be no data, but we can ask anyway.
for _ in query:
pass # Not necessarily reached.
