def testAggregatedTableOutput(self):
self.SetupMockGetRegionalResources([])
self.SetupMockGetZonalResources([])
self.list_json.side_effect = [
resource_projector.MakeSerializable(
test_resources.MakeInstanceGroupManagers(
api=API_VERSION, scope_name='zone-1', scope_type='zone') +
test_resources.MakeInstanceGroupManagers(api=API_VERSION,
scope_name='region-1',
scope_type='region'))
]
self.make_requests.side_effect = [[
test_resources.MakeAutoscalers(api=API_VERSION,
scope_name='region-1',
scope_type='region')[0],
test_resources.MakeAutoscalers(api=API_VERSION,
scope_name='zone-1',
scope_type='zone')[0]
]]
self.Run('compute instance-groups managed list')
self.mock_get_zonal_resources.assert_any_call(
service=self.compute.instanceGroups,
project='my-project',
requested_zones=set(['zone-1']),
filter_expr=None,
http=self.mock_http(),
batch_url=self.batch_url,
errors=[])
self.mock_get_regional_resources.assert_any_call(
service=self.compute.regionInstanceGroups,
project='my-project',
requested_regions=set(['region-1']),
filter_expr=None,
http=self.mock_http(),
batch_url=self.batch_url,
errors=[])
self.AssertOutputEquals(textwrap.dedent("""\
NAME LOCATION SCOPE BASE_INSTANCE_NAME SIZE TARGET_SIZE INSTANCE_TEMPLATE AUTOSCALED
group-1 zone-1 zone test-instance-name-1 0 1 template-1 yes
group-2 zone-1 zone test-instance-name-2 3 10 template-2 no
group-3 zone-1 zone test-instance-name-3 10 1 template-2 no
group-1 region-1 region test-instance-name-1 0 1 template-1 yes
group-2 region-1 region test-instance-name-2 3 10 template-2 no
group-3 region-1 region test-instance-name-3 10 1 template-2 no
"""),
normalize_space=True)
self.AssertErrEquals('')
def testScopePromptingWithDefaultZone(self):
self.StartPatch('googlecloudsdk.core.console.console_io.CanPrompt',
return_value=True)
properties.VALUES.compute.zone.Set('zone-1')
self.make_requests.side_effect = iter([
test_resources.MakeAutoscalers(
API_VERSION, scope_type='zone', scope_name='zone-1'),
[self.messages.Region(name='region-1')],
[self.messages.Zone(name='zone-2'),
self.messages.Zone(name='zone-1')],
test_resources.MakeInstanceGroupManagers(
API_VERSION, scope_type='region', scope_name='region-1'),
[], # Deleting 0 Autoscalers.
])
self.WriteInput('1\ny')
self.Run("""
compute instance-groups managed delete group-1
""")
self.CheckRequests(
self.autoscalers_list_request_zone_1,
[(self.compute.autoscalers,
'Delete',
self.messages.ComputeAutoscalersDeleteRequest(
autoscaler='autoscaler-1',
project='my-project',
zone='zone-1'))],
[(self.compute.instanceGroupManagers,
'Delete',
self.messages.ComputeInstanceGroupManagersDeleteRequest(
instanceGroupManager='group-1',
project='my-project',
zone='zone-1')
)],
)
def testScopePrompt(self):
self.mock_client.instanceGroupManagers.Get.Expect(
messages.ComputeInstanceGroupManagersGetRequest(
instanceGroupManager='group-1',
project='fake-project',
zone='zone-2'),
test_resources.MakeInstanceGroupManagers(API_VERSION)[0])
self.StartPatch('googlecloudsdk.core.console.console_io.CanPrompt',
return_value=True)
self.StartPatch(
'googlecloudsdk.api_lib.compute.zones.service.List',
return_value=[
messages.Zone(name='zone-1'),
messages.Zone(name='zone-2')
],
)
self.StartPatch(
'googlecloudsdk.api_lib.compute.regions.service.List',
return_value=[messages.Region(name='region-1')],
)
self._MockAutoscalerRequest(
test_resources.MakeAutoscalers(API_VERSION))
self.WriteInput('3\n')
self.Run("""
compute instance-groups managed describe group-1
""")
self.AssertErrContains('group-1')
self.AssertErrContains('zone-1')
self.AssertErrContains('zone-2')
self.AssertErrContains('region-1')
def SetUp(self):
self.SelectApi(API_VERSION)
self.make_requests.side_effect = iter([
[], # Listed 0 Autoscalers.
test_resources.MakeInstanceGroupManagers(
API_VERSION, scope_type='region', scope_name='region-1'),
[],
])
self.autoscalers_list_request_zone_1 = [(
self.compute.autoscalers, 'List',
self.messages.ComputeAutoscalersListRequest(
maxResults=500,
project='my-project',
zone='zone-1',
),
)]
self.autoscalers_list_request_region_1 = [(
self.compute.regionAutoscalers, 'List',
self.messages.ComputeRegionAutoscalersListRequest(
maxResults=500,
project='my-project',
region='region-1',
),
)]
self.autoscalers_list_request_region_2 = [(
self.compute.regionAutoscalers, 'List',
self.messages.ComputeRegionAutoscalersListRequest(
maxResults=500,
project='my-project',
region='region-2',
),
)]
def testMultipleZonesUri(self):
self.make_requests.side_effect = iter([
[], # No Autoscalers in zone-1 and zone-2
test_resources.MakeInstanceGroupManagers(API_VERSION),
])
self.WriteInput('y')
self.Run("""
compute instance-groups managed delete
https://compute.googleapis.com/compute/{0}/projects/my-project/zones/zone-2/instanceGroupManagers/group-1
https://compute.googleapis.com/compute/{0}/projects/my-project/zones/zone-1/instanceGroupManagers/group-2
""".format(API_VERSION))
self.CheckRequests(
self.autoscalers_list_request_zone_1
+ self.autoscalers_list_request_zone_2,
[(self.compute.instanceGroupManagers,
'Delete',
self.messages.ComputeInstanceGroupManagersDeleteRequest(
instanceGroupManager='group-1',
project='my-project',
zone='zone-2')
),
(self.compute.instanceGroupManagers,
'Delete',
self.messages.ComputeInstanceGroupManagersDeleteRequest(
instanceGroupManager='group-2',
project='my-project',
zone='zone-1')
)],
)
self.AssertErrContains(
r'The following instance group managers will be deleted:\n'
r' - [group-1] in [zone-2]\n'
r' - [group-2] in [zone-1]')
self.AssertErrContains('PROMPT_CONTINUE')
def testDescribeRegional(self):
self.mock_client.regionInstanceGroupManagers.Get.Expect(
messages.ComputeRegionInstanceGroupManagersGetRequest(
instanceGroupManager='group-1',
project='fake-project',
region='region-1'),
test_resources.MakeInstanceGroupManagers(API_VERSION,
scope_type='region',
scope_name='region-1')[0])
self._MockAutoscalerRequest([])
self.Run("""
compute instance-groups managed describe group-1 --region region-1
""")
self.assertMultiLineEqual(
encoding.Decode(self.stdout.getvalue()),
textwrap.dedent("""\
baseInstanceName: test-instance-name-1
creationTimestamp: '2013-09-06T17:54:10.636-07:00'
description: Test description.
fingerprint: MTIzNA==
instanceGroup: {0}/projects/my-project/regions/region-1/instanceGroups/group-1
instanceTemplate: {0}/projects/my-project/global/instanceTemplates/template-1
name: group-1
region: {0}/projects/my-project/regions/region-1
selfLink: {0}/projects/my-project/regions/region-1/instanceGroupManagers/group-1
targetSize: 1
""".format(self.compute_uri)))
def SetupMockGetRegionalResources(self, autoscalers):
def _MockGetRegionalResources(service, project, requested_regions,
filter_expr, http, batch_url, errors):
_ = project, requested_regions, filter_expr, http, batch_url, errors
if 'RegionInstanceGroupsService' in str(type(service)):
return test_resources.MakeInstanceGroups(msgs=self.messages,
api=API_VERSION,
scope_name='region-1',
scope_type='region')
return None
self.SelectApi(API_VERSION)
lister_patcher = mock.patch(
'googlecloudsdk.api_lib.compute.lister.GetRegionalResources',
autospec=True)
self.addCleanup(lister_patcher.stop)
self.mock_get_regional_resources = lister_patcher.start()
self.mock_get_regional_resources.side_effect = _MockGetRegionalResources
self.list_json.side_effect = [
resource_projector.MakeSerializable(
test_resources.MakeInstanceGroupManagers(api=API_VERSION,
scope_name='region-1',
scope_type='region'))
]
self.make_requests.side_effect = [autoscalers]
def _ExpectGetManagedInstanceGroup(self):
if self.scope is scope_util.ScopeEnum.REGION:
request = self._MakeGetManagedInstanceGroupRegionRequest()
if self.scope is scope_util.ScopeEnum.ZONE:
request = self._MakeGetManagedInstanceGroupZoneRequest()
instance_group_managers = test_resources.MakeInstanceGroupManagers(
self.api)
self._ExpectRequest(request, instance_group_managers[:1])
def testPromptingWithYes(self):
self.make_requests.side_effect = iter([
test_resources.MakeAutoscalers(
API_VERSION, scope_type='region', scope_name='region-1'),
test_resources.MakeInstanceGroupManagers(
API_VERSION, scope_type='region', scope_name='region-1'),
[],
])
self.WriteInput('y')
self.Run("""
compute instance-groups managed delete group-1 group-2 group-3
--region region-1
""")
self.CheckRequests(
self.autoscalers_list_request_region_1,
[(self.compute.regionAutoscalers,
'Delete',
self.messages.ComputeRegionAutoscalersDeleteRequest(
autoscaler='autoscaler-1',
project='my-project',
region='region-1')),
(self.compute.regionAutoscalers,
'Delete',
self.messages.ComputeRegionAutoscalersDeleteRequest(
autoscaler='autoscaler-2',
project='my-project',
region='region-1')),
(self.compute.regionAutoscalers,
'Delete',
self.messages.ComputeRegionAutoscalersDeleteRequest(
autoscaler='autoscaler-3',
project='my-project',
region='region-1'))],
[(self.compute.regionInstanceGroupManagers,
'Delete',
self.messages.ComputeRegionInstanceGroupManagersDeleteRequest(
instanceGroupManager='group-1',
project='my-project',
region='region-1')),
(self.compute.regionInstanceGroupManagers,
'Delete',
self.messages.ComputeRegionInstanceGroupManagersDeleteRequest(
instanceGroupManager='group-2',
project='my-project',
region='region-1')),
(self.compute.regionInstanceGroupManagers,
'Delete',
self.messages.ComputeRegionInstanceGroupManagersDeleteRequest(
instanceGroupManager='group-3',
project='my-project',
region='region-1'))],)
self.AssertErrContains(
r'The following region instance group managers will be deleted:\n'
r' - [group-1] in [region-1]\n'
r' - [group-2] in [region-1]\n'
r' - [group-3] in [region-1]')
self.AssertErrContains('PROMPT_CONTINUE')
def _MockMakeRequests(requests, batch_url, *unused_args,
**unused_kwargs):
_ = batch_url, unused_args, unused_kwargs
if len(requests) != 1:
self.fail('expected to send single request, sent: ' +
str(requests))
service = requests[0][0]
verb = requests[0][1]
if verb == 'AggregatedList':
if service == self.compute.instanceGroupManagers:
return (
test_resources.MakeInstanceGroupManagers(api_version) +
test_resources.MakeInstanceGroupManagers(
api=api_version,
scope_name='region-1',
scope_type='region'))
elif service == self.compute.instanceGroups:
return (test_resources.MakeInstanceGroups(
self.messages, api_version) +
test_resources.MakeInstanceGroups(
self.messages,
api=api_version,
scope_name='region-1',
scope_type='region'))
if verb == 'List':
if service == self.compute.instanceGroups:
return test_resources.MakeInstanceGroups(
self.messages, api_version)
elif service == self.compute.regionInstanceGroups:
return test_resources.MakeInstanceGroups(
self.messages,
api=api_version,
scope_name='region-1',
scope_type='region')
elif service == self.compute.instanceGroupManagers:
return test_resources.MakeInstanceGroupManagers(
api_version)
elif service == self.compute.regionInstanceGroupManagers:
return test_resources.MakeInstanceGroupManagers(
api=api_version,
scope_name='region-1',
scope_type='region')
else:
self.fail('unexpected request ' + str(requests[0]))
def testZoneAndRegion(self):
self.make_requests.side_effect = iter([
test_resources.MakeAutoscalers(
'beta', scope_type='zone', scope_name='zone-1'),
test_resources.MakeAutoscalers(
'beta', scope_type='region', scope_name='region-1'),
test_resources.MakeInstanceGroupManagers(
'beta', scope_type='zone', scope_name='zone-1'),
test_resources.MakeInstanceGroupManagers(
'beta', scope_type='region', scope_name='region-1'),
[], # Deleting 0 Autoscalers.
])
self.WriteInput('y')
self.Run("""
beta compute instance-groups managed delete
https://compute.googleapis.com/compute/{0}/projects/my-project/zones/zone-1/instanceGroupManagers/group-1
https://compute.googleapis.com/compute/{0}/projects/my-project/regions/region-1/instanceGroupManagers/group-2
""".format(API_VERSION))
self.AssertErrContains(
r'The following instance group managers will be deleted:\n'
r' - [group-1]\n'
r' - [group-2] in [region-1]')
self.AssertErrContains('PROMPT_CONTINUE')
self.CheckRequests(
self.autoscalers_list_request_zone_1 +
self.autoscalers_list_request_region_1,
[(self.compute.autoscalers, 'Delete',
self.messages.ComputeAutoscalersDeleteRequest(
autoscaler='autoscaler-1', project='my-project', zone='zone-1'))],
[(self.compute.instanceGroupManagers, 'Delete',
self.messages.ComputeInstanceGroupManagersDeleteRequest(
instanceGroupManager='group-1',
project='my-project',
zone='zone-1')),
(self.compute.regionInstanceGroupManagers, 'Delete',
self.messages.ComputeRegionInstanceGroupManagersDeleteRequest(
instanceGroupManager='group-2',
project='my-project',
region='region-1'))],)
def testDescribeAutoscaledRegional(self):
self.mock_client.regionInstanceGroupManagers.Get.Expect(
messages.ComputeRegionInstanceGroupManagersGetRequest(
instanceGroupManager='group-1',
project='fake-project',
region='region-1'),
test_resources.MakeInstanceGroupManagers(API_VERSION,
scope_type='region',
scope_name='region-1')[0])
self._MockAutoscalerRequest(
test_resources.MakeAutoscalers(API_VERSION,
scope_name='region-1',
scope_type='region'))
self.Run("""
compute instance-groups managed describe group-1 --region region-1
""")
self.assertMultiLineEqual(
encoding.Decode(self.stdout.getvalue()),
textwrap.dedent("""\
autoscaler:
autoscalingPolicy:
coolDownPeriodSec: 60
cpuUtilization:
utilizationTarget: 0.8
customMetricUtilizations:
- metric: custom.cloudmonitoring.googleapis.com/seconds
utilizationTarget: 60.0
utilizationTargetType: DELTA_PER_MINUTE
loadBalancingUtilization:
utilizationTarget: 0.9
maxNumReplicas: 10
minNumReplicas: 2
creationTimestamp: Two days ago
id: '1'
name: autoscaler-1
region: {0}/projects/my-project/regions/region-1
selfLink: {0}/projects/my-project/regions/region-1/autoscalers/autoscaler-1
target: {0}/projects/my-project/regions/region-1/instanceGroupManagers/group-1
baseInstanceName: test-instance-name-1
creationTimestamp: '2013-09-06T17:54:10.636-07:00'
description: Test description.
fingerprint: MTIzNA==
instanceGroup: {0}/projects/my-project/regions/region-1/instanceGroups/group-1
instanceTemplate: {0}/projects/my-project/global/instanceTemplates/template-1
name: group-1
region: {0}/projects/my-project/regions/region-1
selfLink: {0}/projects/my-project/regions/region-1/instanceGroupManagers/group-1
targetSize: 1
""".format(self.compute_uri)))
def _SetUpZonal(self, api_version):
self.SelectApi(api_version)
self.zone = 'us-central2-a'
self.scope_params = ('zone', self.zone)
self.make_requests.side_effect = iter([
[
self.messages.Zone(name=self.zone),
],
[
test_resources.MakeInstanceGroupManagers(
api_version, scope_name=self.zone)[0],
],
[
test_resources.MakeInstanceGroups(self.messages,
api_version,
scope_name=self.zone)[0],
],
[
test_resources.MakeAutoscalers(api_version,
scope_name=self.zone)[0],
],
])
self.zone_1_get_request = [
(self.compute.zones, 'Get',
self.messages.ComputeZonesGetRequest(project='my-project',
zone=self.zone))
]
self.zone_ig_list_request = [
(self.compute.instanceGroups, 'List',
self.messages.ComputeInstanceGroupsListRequest(
maxResults=500,
project='my-project',
zone=self.zone,
))
]
self.zone_as_list_request = [
(self.compute.autoscalers, 'List',
self.messages.ComputeAutoscalersListRequest(maxResults=500,
project='my-project',
zone=self.zone))
]
self.template_1_uri = (
'{0}/projects/my-project/global/instanceTemplates/template-1'.
format(self.compute_uri))
self.http_health_check_uri = (
'{0}/projects/my-project/global/httpHealthChecks/health-check-1'.
format(self.compute_uri))
self.https_health_check_uri = (
'{0}/projects/my-project/global/httpsHealthChecks/health-check-2'.
format(self.compute_uri))
def _SetUpRegional(self, api_version):
self.SelectApi(api_version)
self.region = 'us-central2'
self.scope_params = ('region', self.region)
self.make_requests.side_effect = iter([
[
test_resources.MakeInstanceGroupManagers(
api_version, scope_name=self.region,
scope_type='region')[0],
],
[
test_resources.MakeInstanceGroups(self.messages,
api_version,
scope_name=self.region,
scope_type='region')[0],
],
[
test_resources.MakeAutoscalers(api_version,
scope_name=self.region,
scope_type='region')[0],
],
])
self.region_ig_list_request = [
(self.compute.regionInstanceGroups, 'List',
self.messages.ComputeRegionInstanceGroupsListRequest(
maxResults=500,
project='my-project',
region=self.region,
))
]
self.region_as_list_request = [
(self.compute.regionAutoscalers, 'List',
self.messages.ComputeRegionAutoscalersListRequest(
maxResults=500, project='my-project', region=self.region))
]
self.template_1_uri = (
'{0}/projects/my-project/global/instanceTemplates/template-1'.
format(self.compute_uri))
self.http_health_check_uri = (
'{0}/projects/my-project/global/httpHealthChecks/health-check-1'.
format(self.compute_uri))
self.https_health_check_uri = (
'{0}/projects/my-project/global/httpsHealthChecks/health-check-2'.
format(self.compute_uri))
self.region_uri = ('{0}/projects/my-project/regions/{1}'.format(
self.compute_uri, self.region))
def _MockGetZonalResources(service, project, requested_zones,
filter_expr, http, batch_url, errors):
_ = project, requested_zones, filter_expr, http, batch_url, errors
if service == self.compute.instanceGroupManagers:
return test_resources.MakeInstanceGroupManagers(API_VERSION)
if service == self.compute.instanceGroups:
non_zonal_group = (self.messages.InstanceGroup(
name='regional-group',
selfLink=('https://compute.googleapis.com/compute/{0}/'
'projects/my-project/regions/region-1/'
'instanceGroups/group-1'.format(API_VERSION)),
creationTimestamp='2013-09-06T17:54:10.636-07:00',
description='Test regional instance group',
fingerprint=b'123',
namedPorts=[],
size=0,
))
return (test_resources.MakeInstanceGroups(
self.messages, API_VERSION) + [non_zonal_group])
return None
def testMultipleZonesUriAndFlag(self):
self.make_requests.side_effect = iter([
[], # No Autoscalers in any region
test_resources.MakeInstanceGroupManagers(
API_VERSION, scope_type='region', scope_name='region-1'),
])
self.WriteInput('y')
self.Run("""
compute instance-groups managed delete
https://compute.googleapis.com/compute/{0}/projects/my-project/regions/region-2/instanceGroupManagers/group-1
group-2
--region region-1
""".format(API_VERSION))
self.CheckRequests(
self.autoscalers_list_request_region_1
+ self.autoscalers_list_request_region_2,
[(self.compute.regionInstanceGroupManagers,
'Delete',
self.messages.ComputeRegionInstanceGroupManagersDeleteRequest(
instanceGroupManager='group-1',
project='my-project',
region='region-2')
),
(self.compute.regionInstanceGroupManagers,
'Delete',
self.messages.ComputeRegionInstanceGroupManagersDeleteRequest(
instanceGroupManager='group-2',
project='my-project',
region='region-1')
)],
)
self.AssertErrContains(
r'The following region instance group managers will be deleted:\n'
r' - [group-1] in [region-2]\n'
r' - [group-2] in [region-1]')
self.AssertErrContains('PROMPT_CONTINUE')
class InstanceGroupManagersStopAutoscalingZonalTest(test_base.BaseTest):
INSTANCE_GROUP_MANAGERS = test_resources.MakeInstanceGroupManagers(
api=API_VERSION)
AUTOSCALERS = test_resources.MakeAutoscalers(api=API_VERSION)
def SetUp(self):
self.SelectApi(API_VERSION)
self.autoscalers_list_request = [
(
self.compute.autoscalers,
'List',
self.messages.ComputeAutoscalersListRequest(
maxResults=500,
project='my-project',
zone='zone-1',
),
),
]
self.managed_instance_group_get_request = [(
self.compute.instanceGroupManagers,
'Get',
self.messages.ComputeInstanceGroupManagersGetRequest(
instanceGroupManager='group-1',
project='my-project',
zone='zone-1',
),
)]
def testStopAutoscalingNotAutoscaledManagedInstanceGroup(self):
self.make_requests.side_effect = iter([
[self.INSTANCE_GROUP_MANAGERS[0]],
self.AUTOSCALERS[1:],
])
with self.assertRaisesRegex(
managed_instance_groups_utils.ResourceNotFoundException,
r'The managed instance group is not autoscaled\.'):
self.Run(
'compute instance-groups managed stop-autoscaling group-1 '
'--zone zone-1')
self.CheckRequests(self.managed_instance_group_get_request,
self.autoscalers_list_request)
def testStopAutoscalingAutoscaledManagedInstanceGroupWithPrompt(self):
self.StartPatch('googlecloudsdk.core.console.console_io.CanPrompt',
return_value=True)
self.WriteInput('2\n')
self.make_requests.side_effect = iter([
[self.messages.Region(name='region-1')],
[
self.messages.Zone(name='zone-1'),
self.messages.Zone(name='zone-2')
],
[self.INSTANCE_GROUP_MANAGERS[0]],
self.AUTOSCALERS,
[] # Delete Autoscaler.
])
self.Run('compute instance-groups managed stop-autoscaling group-1')
delete_request = self.messages.ComputeAutoscalersDeleteRequest(
project='my-project', zone='zone-1', autoscaler='autoscaler-1')
self.CheckRequests(
self.regions_list_request,
self.zones_list_request,
self.managed_instance_group_get_request,
self.autoscalers_list_request,
[(self.compute.autoscalers, 'Delete', delete_request)],
)
def testAssertsIgmExists(self):
def MakeRequests(*_, **kwargs):
kwargs['errors'].append((404, 'Not Found'))
yield
self.make_requests.side_effect = MakeRequests
with self.assertRaisesRegex(
managed_instance_groups_utils.ResourceNotFoundException,
'Could not fetch resource:'):
self.Run(
'compute instance-groups managed stop-autoscaling group-1 '
'--zone zone-1')
self.CheckRequests(self.managed_instance_group_get_request)
def SetUp(self):
self.SelectApi(API_VERSION)
self.make_requests.side_effect = iter([
[test_resources.MakeInstanceGroups(self.messages, API_VERSION)[0]],
[test_resources.MakeInstanceGroupManagers(API_VERSION)[0]],
])
