class TestTargetPoolMigration(unittest.TestCase):
def setUp(self):
print('Initialize test environment.')
project = os.environ["PROJECT_ID"]
credentials, default_project = google.auth.default()
self.compute = discovery.build('compute',
'v1',
credentials=credentials)
self.google_api_interface = GoogleApiInterface(self.compute, project,
'us-central1',
'us-central1-a')
self.test_resource_creator = TestResourceCreator(
self.google_api_interface)
def testOnlyInstancesAsBackend(self):
""" The target pool has single instances as its backends, no instance groups
"""
### create test resources
instance_name_1 = 'end-to-end-test-instance-1'
operation = self.test_resource_creator.create_instance_using_template(
instance_name_1,
self.test_resource_creator.legacy_instance_template_selfLink)
instance_selfLink_1 = operation['targetLink']
instance_name_2 = 'end-to-end-test-instance-2'
operation = self.test_resource_creator.create_instance_using_template(
instance_name_2,
self.test_resource_creator.legacy_instance_template_selfLink)
instance_selfLink_2 = operation['targetLink']
target_pool_name = 'end-to-end-test-target-pool'
operation = self.test_resource_creator.create_target_pool_with_health_check(
'sample_target_pool_with_no_instance.json',
target_pool_name, [], [instance_selfLink_1, instance_selfLink_2],
health_check_selfLink=None)
target_pool_selfLink = operation['targetLink']
original_target_pool_instance_list = \
self.google_api_interface.get_target_pool_config(target_pool_name)[
'instances']
### start migration
selfLink_executor = SelfLinkExecutor(
self.compute,
target_pool_selfLink,
self.test_resource_creator.network_name,
self.test_resource_creator.subnetwork_name,
)
migration_handler = selfLink_executor.build_migration_handler()
migration_handler.network_migration()
### check migration result
new_target_pool_instance_list = \
self.google_api_interface.get_target_pool_config(target_pool_name)[
'instances']
# target pool's instances unchanged
self.assertTrue(
compare_two_list(new_target_pool_instance_list,
original_target_pool_instance_list))
# instances' network changed
new_instance_1_config = self.google_api_interface.get_instance_configs(
instance_name_1)
self.assertTrue(
check_instance_network(
new_instance_1_config,
self.test_resource_creator.network_selfLink,
self.test_resource_creator.subnetwork_selfLink))
new_instance_2_config = self.google_api_interface.get_instance_configs(
instance_name_2)
self.assertTrue(
check_instance_network(
new_instance_2_config,
self.test_resource_creator.network_selfLink,
self.test_resource_creator.subnetwork_selfLink))
print('Pass the current test')
def testOnlyManagedInstanceGroupsAsBackend(self):
""" The target pool is served by managed instance groups
"""
### create resources
group_name_1 = 'end-to-end-test-managed-instance-group-1'
self.test_resource_creator.create_regional_managed_instance_group(
self.test_resource_creator.legacy_instance_template_selfLink,
group_name_1,
'sample_multi_zone_managed_instance_group.json',
)
original_instance_template_1_configs = self.google_api_interface.get_multi_zone_instance_template_configs(
group_name_1)
group_name_2 = 'end-to-end-test-managed-instance-group-2'
self.test_resource_creator.create_regional_managed_instance_group(
self.test_resource_creator.legacy_instance_template_selfLink,
group_name_2,
'sample_multi_zone_managed_instance_group.json',
)
original_instance_template_2_configs = self.google_api_interface.get_multi_zone_instance_template_configs(
group_name_2)
target_pool_name = 'end-to-end-test-target-pool'
operation = self.test_resource_creator.create_target_pool_with_health_check(
'sample_target_pool_with_no_instance.json',
target_pool_name, [group_name_1, group_name_2], [],
health_check_selfLink=None)
target_pool_selfLink = operation['targetLink']
# time allowance to let the instance groups create instances
time.sleep(20)
### start migration
selfLink_executor = SelfLinkExecutor(
self.compute,
target_pool_selfLink,
self.test_resource_creator.network_name,
self.test_resource_creator.subnetwork_name,
)
migration_handler = selfLink_executor.build_migration_handler()
migration_handler.network_migration()
### check migration result
new_instance_template_1_configs = self.google_api_interface.get_multi_zone_instance_template_configs(
group_name_1)
new_instance_template_2_configs = self.google_api_interface.get_multi_zone_instance_template_configs(
group_name_2)
self.assertTrue(
instance_template_config_is_unchanged_except_for_network_and_name(
original_instance_template_1_configs,
new_instance_template_1_configs))
self.assertTrue(
instance_template_config_is_unchanged_except_for_network_and_name(
original_instance_template_2_configs,
new_instance_template_2_configs))
# network changed
self.assertTrue(
check_instance_template_network(
new_instance_template_1_configs,
self.test_resource_creator.network_selfLink,
self.test_resource_creator.subnetwork_selfLink))
self.assertTrue(
check_instance_template_network(
new_instance_template_2_configs,
self.test_resource_creator.network_selfLink,
self.test_resource_creator.subnetwork_selfLink))
print('Pass the current test')
def testEmptyManagedInstanceGroupAsBackend(self):
""" The target pool is served by an empty managed instance groups (no instances in the group)
Expectation: the instance group which has no instance will not be migrated to the target subnet.
The instance group which has instances will be migrated to the target subnet.
"""
### create resources
group_name_1 = 'end-to-end-test-managed-instance-group-1'
self.test_resource_creator.create_regional_managed_instance_group(
self.test_resource_creator.legacy_instance_template_selfLink,
group_name_1,
'sample_multi_zone_managed_instance_group_with_zero_instance.json',
)
original_instance_template_1_configs = self.google_api_interface.get_multi_zone_instance_template_configs(
group_name_1)
group_name_2 = 'end-to-end-test-managed-instance-group-2'
self.test_resource_creator.create_regional_managed_instance_group(
self.test_resource_creator.legacy_instance_template_selfLink,
group_name_2,
'sample_multi_zone_managed_instance_group.json',
)
original_instance_template_2_configs = self.google_api_interface.get_multi_zone_instance_template_configs(
group_name_2)
target_pool_name = 'end-to-end-test-target-pool'
operation = self.test_resource_creator.create_target_pool_with_health_check(
'sample_target_pool_with_no_instance.json',
target_pool_name, [group_name_2, group_name_1], [],
health_check_selfLink=None)
target_pool_selfLink = operation['targetLink']
# time allowance to let the instance groups create instances
time.sleep(20)
### start migration
selfLink_executor = SelfLinkExecutor(
self.compute,
target_pool_selfLink,
self.test_resource_creator.network_name,
self.test_resource_creator.subnetwork_name,
)
migration_handler = selfLink_executor.build_migration_handler()
migration_handler.network_migration()
### check migration result
new_instance_template_1_configs = self.google_api_interface.get_multi_zone_instance_template_configs(
group_name_1)
new_instance_template_2_configs = self.google_api_interface.get_multi_zone_instance_template_configs(
group_name_2)
# group 1 didn't migrate
self.assertTrue(
instance_template_config_is_unchanged(
original_instance_template_1_configs,
new_instance_template_1_configs))
self.assertTrue(
instance_template_config_is_unchanged_except_for_network_and_name(
original_instance_template_2_configs,
new_instance_template_2_configs))
# group 2 network changed
self.assertTrue(
check_instance_template_network(
new_instance_template_2_configs,
self.test_resource_creator.network_selfLink,
self.test_resource_creator.subnetwork_selfLink))
print('Pass the current test')
def testInstancesAndManagedInstanceGroupsMixedBackends(self):
""" The target pool served by both instances and managed instance groups
"""
### create test resources
group_name_1 = 'end-to-end-test-managed-instance-group-1'
self.test_resource_creator.create_regional_managed_instance_group(
self.test_resource_creator.legacy_instance_template_selfLink,
group_name_1,
'sample_multi_zone_managed_instance_group.json',
)
original_instance_template_1_configs = self.google_api_interface.get_multi_zone_instance_template_configs(
group_name_1)
instance_name_1 = 'end-to-end-test-instance-1'
operation = self.test_resource_creator.create_instance_using_template(
instance_name_1,
self.test_resource_creator.legacy_instance_template_selfLink)
instance_1_selfLink = operation['targetLink']
original_instance_1_config = self.google_api_interface.get_instance_configs(
instance_name_1)
target_pool_name = 'end-to-end-test-target-pool'
operation = self.test_resource_creator.create_target_pool_with_health_check(
'sample_target_pool_with_no_instance.json',
target_pool_name, [group_name_1], [instance_1_selfLink],
health_check_selfLink=None)
target_pool_selfLink = operation['targetLink']
# time allowance to let the instance groups create instances
time.sleep(10)
### start migration
selfLink_executor = SelfLinkExecutor(
self.compute,
target_pool_selfLink,
self.test_resource_creator.network_name,
self.test_resource_creator.subnetwork_name,
)
migration_handler = selfLink_executor.build_migration_handler()
migration_handler.network_migration()
### check migration result
new_instance_template_1_configs = self.google_api_interface.get_multi_zone_instance_template_configs(
group_name_1)
self.assertTrue(
instance_template_config_is_unchanged_except_for_network_and_name(
original_instance_template_1_configs,
new_instance_template_1_configs))
new_instance_1_config = self.google_api_interface.get_instance_configs(
instance_name_1)
self.assertTrue(
resource_config_is_unchanged_except_for_network(
original_instance_1_config, new_instance_1_config))
# network changed
self.assertTrue(
check_instance_template_network(
new_instance_template_1_configs,
self.test_resource_creator.network_selfLink,
self.test_resource_creator.subnetwork_selfLink))
self.assertTrue(
check_instance_network(
new_instance_1_config,
self.test_resource_creator.network_selfLink,
self.test_resource_creator.subnetwork_selfLink))
print('Pass the current test')
def testWithInstancesFromAnUnmanagedInstanceGroup(self):
""" The target pool is served by an instance which is a member of an unmanaged instance group
Expectation: the migration will not start.
"""
### create resources
instance_name_1 = 'end-to-end-test-instance-1'
operation = self.test_resource_creator.create_instance_using_template(
instance_name_1,
self.test_resource_creator.legacy_instance_template_selfLink)
instance_selfLink_1 = operation['targetLink']
instance_name_2 = 'end-to-end-test-instance-2'
operation = self.test_resource_creator.create_instance_using_template(
instance_name_2,
self.test_resource_creator.legacy_instance_template_selfLink)
instance_selfLink_2 = operation['targetLink']
unmanaged_instance_group_name = 'end-to-end-test-unmanaged-instance-group-1'
original_instances_in_group = [instance_name_1, instance_name_2]
self.test_resource_creator.create_unmanaged_instance_group(
unmanaged_instance_group_name, original_instances_in_group)
original_group_config = self.google_api_interface.get_unmanaged_instance_group_configs(
unmanaged_instance_group_name)
target_pool_name = 'end-to-end-test-target-pool'
operation = self.test_resource_creator.create_target_pool_with_health_check(
'sample_target_pool_with_no_instance.json',
target_pool_name, [], [instance_selfLink_1, instance_selfLink_2],
health_check_selfLink=None)
target_pool_selfLink = operation['targetLink']
original_target_pool_config = self.google_api_interface.get_target_pool_config(
target_pool_name)
### start migration
selfLink_executor = SelfLinkExecutor(
self.compute,
target_pool_selfLink,
self.test_resource_creator.network_name,
self.test_resource_creator.subnetwork_name,
)
# the migration will not start, and raise an error
with self.assertRaises(AmbiguousTargetResource):
migration_handler = selfLink_executor.build_migration_handler()
migration_handler.network_migration()
### check migration result
# unmanaged instance group doesn't change
new_group_config = self.google_api_interface.get_unmanaged_instance_group_configs(
unmanaged_instance_group_name)
self.assertEqual(original_group_config, new_group_config)
# target pool is unchanged
new_target_pool_config = self.google_api_interface.get_target_pool_config(
target_pool_name)
self.assertEqual(original_target_pool_config, new_target_pool_config)
print('Pass the current test')
def tearDown(self) -> None:
pass
def doCleanups(self) -> None:
self.google_api_interface.clean_all_resources()
class TestUnmanagedInstanceGroupMigration(unittest.TestCase):
def setUp(self):
print('Initialize test environment.')
project = os.environ["PROJECT_ID"]
credentials, default_project = google.auth.default()
self.compute = discovery.build('compute',
'v1',
credentials=credentials)
self.google_api_interface = GoogleApiInterface(self.compute, project,
'us-central1',
'us-central1-a')
self.test_resource_creator = TestResourceCreator(
self.google_api_interface)
def testWithMultipleInstancesInTheGroup(self):
### create test resources
instance_name_1 = 'end-to-end-test-instance-1'
self.test_resource_creator.create_instance_using_template(
instance_name_1,
self.test_resource_creator.legacy_instance_template_selfLink)
instance_name_2 = 'end-to-end-test-instance-2'
self.test_resource_creator.create_instance_using_template(
instance_name_2,
self.test_resource_creator.legacy_instance_template_selfLink)
unmanaged_instance_group_name = 'end-to-end-test-unmanaged-instance-group-1'
original_instances_in_group = [instance_name_1, instance_name_2]
self.test_resource_creator.create_unmanaged_instance_group(
unmanaged_instance_group_name, original_instances_in_group)
original_config = self.google_api_interface.get_unmanaged_instance_group_configs(
unmanaged_instance_group_name)
### start migration
selfLink_executor = SelfLinkExecutor(
self.compute, original_config['selfLink'],
self.test_resource_creator.network_name,
self.test_resource_creator.subnetwork_name, False)
migration_handler = selfLink_executor.build_migration_handler()
migration_handler.network_migration()
### check migration result
new_config = self.google_api_interface.get_unmanaged_instance_group_configs(
unmanaged_instance_group_name)
self.assertTrue(
resource_config_is_unchanged_except_for_network(
original_config, new_config))
# network changed
self.assertTrue(
check_selfLink_equal(new_config['network'],
self.test_resource_creator.network_selfLink))
self.assertTrue(
check_selfLink_equal(
new_config['subnetwork'],
self.test_resource_creator.subnetwork_selfLink))
print('Pass the current test')
def testWithNoInstanceInTheGroup(self):
### create test resources
unmanaged_instance_group_name = 'end-to-end-test-unmanaged-instance-group-1'
self.test_resource_creator.create_unmanaged_instance_group(
unmanaged_instance_group_name, [])
original_config = self.google_api_interface.get_unmanaged_instance_group_configs(
unmanaged_instance_group_name)
### start migration
selfLink_executor = SelfLinkExecutor(
self.compute, original_config['selfLink'],
self.test_resource_creator.network_name,
self.test_resource_creator.subnetwork_name, False)
migration_handler = selfLink_executor.build_migration_handler()
migration_handler.network_migration()
### check migration result
new_config = self.google_api_interface.get_unmanaged_instance_group_configs(
unmanaged_instance_group_name)
self.assertTrue(
resource_config_is_unchanged_except_for_network(
original_config, new_config))
# network changed
self.assertTrue(
check_selfLink_equal(new_config['network'],
self.test_resource_creator.network_selfLink))
print('Pass the current test')
def testAsBackendOfTargetPool(self):
""" The unmanaged instance group serves a target pool, which means the
instances in this instance group serve the target pool.
Expectation: The migration can still succeed.
But the instances might be detached from the target pool after the migration.
"""
### create test resources
instance_name_1 = 'end-to-end-test-instance-1'
operation = self.test_resource_creator.create_instance_using_template(
instance_name_1,
self.test_resource_creator.legacy_instance_template_selfLink)
instance_selfLink_1 = operation['targetLink']
original_instance_config = self.google_api_interface.get_instance_configs(
instance_name_1)
unmanaged_instance_group_name = 'end-to-end-test-unmanaged-instance-group-1'
original_instances_in_group = [instance_name_1]
unmanaged_instance_group_selfLink = \
self.test_resource_creator.create_unmanaged_instance_group(
unmanaged_instance_group_name,
original_instances_in_group)['targetLink']
original_instance_group_config = self.google_api_interface.get_unmanaged_instance_group_configs(
unmanaged_instance_group_name)
target_pool_name = 'end-to-end-test-target-pool'
self.test_resource_creator.create_target_pool_with_health_check(
'sample_target_pool_with_no_instance.json',
target_pool_name, [], [instance_selfLink_1],
health_check_selfLink=None)
### start migration
selfLink_executor = SelfLinkExecutor(
self.compute,
unmanaged_instance_group_selfLink,
self.test_resource_creator.network_name,
self.test_resource_creator.subnetwork_name,
)
migration_handler = selfLink_executor.build_migration_handler()
migration_handler.network_migration()
### check migration result
# the migration is successful
new_instance_config = self.google_api_interface.get_instance_configs(
instance_name_1)
new_instance_group_config = self.google_api_interface.get_unmanaged_instance_group_configs(
unmanaged_instance_group_name)
self.assertTrue(
resource_config_is_unchanged_except_for_network(
original_instance_config, new_instance_config))
self.assertTrue(
resource_config_is_unchanged_except_for_network(
original_instance_group_config, new_instance_group_config))
# network changed
self.assertTrue(
check_instance_network(
new_instance_config,
self.test_resource_creator.network_selfLink,
self.test_resource_creator.subnetwork_selfLink))
print('Pass the current test')
def testAsBackendOfBackendService(self):
""" The instance group is serving a backend service
Expectation: The migration will fail and rollback to the legacy network.
"""
### create test resources
instance_name_1 = 'end-to-end-test-instance-1'
operation = self.test_resource_creator.create_instance_using_template(
instance_name_1,
self.test_resource_creator.legacy_instance_template_selfLink)
original_instance_config = self.google_api_interface.get_instance_configs(
instance_name_1)
unmanaged_instance_group_name = 'end-to-end-test-unmanaged-instance-group-1'
original_instances_in_group = [instance_name_1]
unmanaged_instance_group_selfLink = \
self.test_resource_creator.create_unmanaged_instance_group(
unmanaged_instance_group_name,
original_instances_in_group)['targetLink']
original_instance_group_config = self.google_api_interface.get_unmanaged_instance_group_configs(
unmanaged_instance_group_name)
backend_service_name = 'end-to-end-test-backend-service'
original_backend_selfLinks = [unmanaged_instance_group_selfLink]
self.test_resource_creator.create_regional_backend_service(
'sample_internal_backend_service.json', backend_service_name,
original_backend_selfLinks)
### start migration
selfLink_executor = SelfLinkExecutor(
self.compute,
unmanaged_instance_group_selfLink,
self.test_resource_creator.network_name,
self.test_resource_creator.subnetwork_name,
)
with self.assertRaises(MigrationFailed):
migration_handler = selfLink_executor.build_migration_handler()
migration_handler.network_migration()
### check migration result
# the migration is failed and the resources are rolled back
new_instance_config = self.google_api_interface.get_instance_configs(
instance_name_1)
new_instance_group_config = self.google_api_interface.get_unmanaged_instance_group_configs(
unmanaged_instance_group_name)
self.assertTrue(
resource_config_is_unchanged_including_network(
original_instance_config, new_instance_config))
self.assertTrue(
resource_config_is_unchanged_including_network(
original_instance_group_config, new_instance_group_config))
print('Pass the current test')
def tearDown(self) -> None:
pass
def doCleanups(self) -> None:
self.google_api_interface.clean_all_resources()