def setUp(self):
super().setUp()
self.instances = []
self.igms = []
self.templates = []
self.inst_client = InstancesClient(transport="rest")
self.name = self.get_unique_name("instance")
self.igm_client = InstanceGroupManagersClient()
self.template_client = InstanceTemplatesClient()
class TestInstanceGroups(TestBase):
def setUp(self):
super().setUp()
self.instances = []
self.igms = []
self.templates = []
self.inst_client = InstancesClient(transport="rest")
self.name = self.get_unique_name("instance")
self.igm_client = InstanceGroupManagersClient()
self.template_client = InstanceTemplatesClient()
def tearDown(self) -> None:
for igm in self.igms:
op = self.igm_client.delete_unary(
project=self.DEFAULT_PROJECT,
zone=self.DEFAULT_ZONE,
instance_group_manager=igm,
)
self.wait_for_zonal_operation(op.name)
for instance in self.instances:
op = self.inst_client.delete_unary(project=self.DEFAULT_PROJECT,
zone=self.DEFAULT_ZONE,
instance=instance)
for template in self.templates:
op = self.template_client.delete_unary(
project=self.DEFAULT_PROJECT, instance_template=template)
def test_instance_group_resize(self):
initialize_params = AttachedDiskInitializeParams(
source_image=self.DISK_IMAGE)
disk = AttachedDisk(
auto_delete=True,
boot=True,
type_="PERSISTENT",
initialize_params=initialize_params,
)
network_interface = NetworkInterface(name="default")
instance = Instance(
name=self.name,
description="test",
disks=[disk],
machine_type=self.MACHINE_TYPE,
network_interfaces=[network_interface],
)
request = InsertInstanceRequest(
zone=self.DEFAULT_ZONE,
project=self.DEFAULT_PROJECT,
instance_resource=instance,
)
operation = self.inst_client.insert_unary(request=request)
self.wait_for_zonal_operation(operation.name)
self.instances.append(self.name)
template_name = self.get_unique_name("template")
igm_name = self.get_unique_name("igm")
instance = operation.target_link
template_resource = InstanceTemplate(name=template_name,
source_instance=instance)
operation = self.template_client.insert_unary(
project=self.DEFAULT_PROJECT,
instance_template_resource=template_resource)
self.wait_for_global_operation(operation.name)
self.templates.append(template_name)
template = operation.target_link
igm_resource = InstanceGroupManager(
base_instance_name="gapicinst",
instance_template=template,
name=igm_name,
target_size=0,
)
operation = self.igm_client.insert_unary(
project=self.DEFAULT_PROJECT,
zone=self.DEFAULT_ZONE,
instance_group_manager_resource=igm_resource,
)
self.wait_for_zonal_operation(operation.name)
self.igms.append(igm_name)
instance_group = self.igm_client.get(
project=self.DEFAULT_PROJECT,
zone=self.DEFAULT_ZONE,
instance_group_manager=igm_name,
)
self.assertEqual(instance_group.target_size, 0)
resize_op = self.igm_client.resize_unary(
project=self.DEFAULT_PROJECT,
zone=self.DEFAULT_ZONE,
size=1,
instance_group_manager=igm_name,
)
self.wait_for_zonal_operation(resize_op.name)
instance_group = self.igm_client.get(
project=self.DEFAULT_PROJECT,
zone=self.DEFAULT_ZONE,
instance_group_manager=igm_name,
)
self.assertEqual(instance_group.target_size, 1)
resize_0_op = self.igm_client.resize_unary(
project=self.DEFAULT_PROJECT,
zone=self.DEFAULT_ZONE,
size=0,
instance_group_manager=igm_name,
)
self.wait_for_zonal_operation(resize_0_op.name)
igm = self.igm_client.get(
project=self.DEFAULT_PROJECT,
zone=self.DEFAULT_ZONE,
instance_group_manager=igm_name,
)
self.assertEqual(igm.target_size, 0)
class TestInstanceGroups(TestBase):
def setUp(self):
super().setUp()
self.instances = []
self.igms = []
self.templates = []
self.inst_client = InstancesClient(transport="rest")
self.name = self.get_unique_name("instance")
self.igm_client = InstanceGroupManagersClient()
self.template_client = InstanceTemplatesClient()
def tearDown(self) -> None:
for igm in self.igms:
op = self.igm_client.delete(
project=self.DEFAULT_PROJECT,
zone=self.DEFAULT_ZONE,
instance_group_manager=igm,
)
self.wait_for_zonal_operation(op.name)
for instance in self.instances:
op = self.inst_client.delete(project=self.DEFAULT_PROJECT,
zone=self.DEFAULT_ZONE,
instance=instance)
for template in self.templates:
op = self.template_client.delete(project=self.DEFAULT_PROJECT,
instance_template=template)
""" Resize fails due to
def test_instance_group_resize(self):
template_name = self.get_unique_name('template')
igm_name = self.get_unique_name('igm')
instance = self.insert_instance().target_link
template_resource = InstanceTemplate(
name=template_name,
source_instance=instance
)
operation = self.template_client.insert(project=self.DEFAULT_PROJECT,
instance_template_resource=template_resource)
self.wait_for_global_operation(operation.name)
self.templates.append(template_name)
template = operation.target_link
igm_resource = InstanceGroupManager(
base_instance_name="gapicinst",
instance_template=template,
name=igm_name,
target_size=1)
operation = self.igm_client.insert(project=self.DEFAULT_PROJECT, zone=self.DEFAULT_ZONE,
instance_group_manager_resource=igm_resource)
self.wait_for_zonal_operation(operation.name)
self.igms.append(igm_name)
instance_group = self.igm_client.get(project=self.DEFAULT_PROJECT,
zone=self.DEFAULT_ZONE, instance_group_manager=igm_name)
self.assertEqual(instance_group.target_size, 1)
resize_op = self.igm_client.resize(project=self.DEFAULT_PROJECT,
zone=self.DEFAULT_ZONE, size=0, instance_group_manager=igm_name)
self.wait_for_zonal_operation(resize_op.name)
igm = self.igm_client.get(project=self.DEFAULT_PROJECT, zone=self.DEFAULT_ZONE,
instance_group_manager=igm_name)
self.assertEqual(igm.target_size, 0)
"""
def insert_instance(self):
disk = AttachedDisk()
initialize_params = AttachedDiskInitializeParams()
initialize_params.source_image = self.DISK_IMAGE
disk.initialize_params = initialize_params
disk.auto_delete = True
disk.boot = True
disk.type_ = AttachedDisk.Type.PERSISTENT
network_interface = NetworkInterface()
network_interface.name = "default"
instance = Instance()
instance.name = self.name
instance.disks = [disk]
instance.machine_type = self.MACHINE_TYPE
instance.network_interfaces = [network_interface]
request = InsertInstanceRequest()
request.zone = self.DEFAULT_ZONE
request.project = self.DEFAULT_PROJECT
request.instance_resource = instance
operation = self.inst_client.insert(request=request)
self.wait_for_zonal_operation(operation.name)
self.instances.append(self.name)
return operation