def test_dimensions(self):
"""Ensures dimensions are returned."""
instance = models.Instance(key=instances.get_instance_key(
'base-name',
'revision',
'zone',
'instance-name',
), )
instance_template_revision = models.InstanceTemplateRevision(
dimensions=machine_provider.Dimensions(
os_family=machine_provider.OSFamily.LINUX, ),
disk_size_gb=300,
machine_type='n1-standard-8',
)
expected_dimensions = {
'backend': 'GCE',
'disk_size_gb': 300,
'hostname': 'instance-name',
'memory_gb': 30,
'num_cpus': 8,
'os_family': 'LINUX',
}
self.assertEqual(
catalog.extract_dimensions(instance, instance_template_revision),
expected_dimensions,
)
def test_no_dimensions(self):
"""Ensures basic dimensions are returned when there are no others."""
instance = models.Instance(key=instances.get_instance_key(
'base-name',
'revision',
'zone',
'instance-name',
), )
instance_template_revision = models.InstanceTemplateRevision()
expected_dimensions = {
'backend': 'GCE',
'hostname': 'instance-name',
}
self.assertEqual(
catalog.extract_dimensions(instance, instance_template_revision),
expected_dimensions,
)