class TestInternalForwardingRuleMigration(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 testWithBackendServiceAttached(self):
""" A backend service is in use by this forwarding rule
Expectation: both the forwarding rule and the backend service will be migrated.
"""
### create test resources
forwarding_rule_name = 'end-to-end-test-forwarding-rule'
backend_service_name = 'end-to-end-test-backend-service'
group_name_1 = 'end-to-end-test-managed-instance-group-1'
operation = 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',
)
instance_group_1_selfLink = operation['targetLink'].replace(
'/instanceGroupManagers/', '/instanceGroups/')
original_instance_template_1_configs = self.google_api_interface.get_multi_zone_instance_template_configs(
group_name_1)
backend_service_selfLink = \
self.test_resource_creator.create_regional_backend_service(
'sample_internal_backend_service.json',
backend_service_name, [instance_group_1_selfLink])['targetLink']
original_backend_service_configs = self.google_api_interface.get_regional_backend_service_configs(
backend_service_name)
forwarding_rule_selfLink = \
self.test_resource_creator.create_regional_forwarding_rule_with_backend_service(
'sample_tcp_regional_forwarding_rule_internal.json',
forwarding_rule_name,
backend_service_selfLink)['targetLink']
original_forwarding_rule_config = self.google_api_interface.get_regional_forwarding_rule_config(
forwarding_rule_name)
### start migration
selfLink_executor = SelfLinkExecutor(self.compute,
forwarding_rule_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
# check internal forwarding rule
new_forwarding_rule_config = self.google_api_interface.get_regional_forwarding_rule_config(
forwarding_rule_name)
self.assertTrue(resource_config_is_unchanged_except_for_network(
original_forwarding_rule_config,
new_forwarding_rule_config))
self.assertTrue(
check_selfLink_equal(new_forwarding_rule_config['network'],
self.test_resource_creator.network_selfLink))
# check backend service config
new_backend_service_configs = self.google_api_interface.get_regional_backend_service_configs(
backend_service_name)
self.assertTrue(resource_config_is_unchanged_except_for_network(
original_backend_service_configs,
new_backend_service_configs))
self.assertEqual(new_backend_service_configs['network'],
self.test_resource_creator.network_selfLink)
# check its backends
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)
)
self.assertTrue(
check_instance_template_network(new_instance_template_1_configs,
self.test_resource_creator.network_selfLink,
self.test_resource_creator.subnetwork_selfLink))
print('Pass the current test')
def testWithTargetInstanceAttached(self):
""" A targetInstance is in use by the forwarding rule
Expectation: both the targetInstance and the forwarding rule will be migrated
"""
### create test resources
forwarding_rule_name = 'end-to-end-test-forwarding-rule'
target_instance_name = 'end-to-end-test-instance'
operation = self.test_resource_creator.create_instance_using_template(
target_instance_name,
self.test_resource_creator.legacy_instance_template_selfLink)
instance_selfLink = operation['targetLink']
target_instance_selfLink = \
self.test_resource_creator.create_a_target_instance(
target_instance_name, instance_selfLink)['targetLink']
original_instance_config = self.google_api_interface.get_instance_configs(
target_instance_name)
forwarding_rule_selfLink = \
self.test_resource_creator.create_regional_forwarding_rule_with_target(
'sample_tcp_regional_forwarding_rule_internal.json',
forwarding_rule_name,
target_instance_selfLink)['targetLink']
original_forwarding_rule_config = self.google_api_interface.get_regional_forwarding_rule_config(
forwarding_rule_name)
### start migration
selfLink_executor = SelfLinkExecutor(self.compute,
forwarding_rule_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
# check internal forwarding rule network
new_forwarding_rule_config = self.google_api_interface.get_regional_forwarding_rule_config(
forwarding_rule_name)
self.assertTrue(
resource_config_is_unchanged_except_for_network(
original_forwarding_rule_config,
new_forwarding_rule_config))
self.assertEqual(new_forwarding_rule_config['network'],
self.test_resource_creator.network_selfLink)
self.assertTrue(
check_selfLink_equal(new_forwarding_rule_config['network'],
self.test_resource_creator.network_selfLink))
# check instance network
new_instance_config = self.google_api_interface.get_instance_configs(
target_instance_name)
self.assertTrue(
resource_config_is_unchanged_except_for_network(new_instance_config,
original_instance_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 tearDown(self) -> None:
pass
def doCleanups(self) -> None:
self.google_api_interface.clean_all_resources()
class TestInternalBackendServiceMigration(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 testWithMultipleBackends(self):
""" A backend service has multiple backends
Expectation: all the backends will be migrated
"""
### create test resources
backend_service_name = 'end-to-end-test-backend-service'
group_name_1 = 'end-to-end-test-managed-instance-group-1'
operation = 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',
)
instance_group_1_selfLink = operation['targetLink'].replace(
'/instanceGroupManagers/', '/instanceGroups/')
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'
operation = 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',
)
instance_group_2_selfLink = operation['targetLink'].replace(
'/instanceGroupManagers/', '/instanceGroups/')
original_instance_template_2_configs = self.google_api_interface.get_multi_zone_instance_template_configs(
group_name_2)
original_backend_selfLinks = [
instance_group_2_selfLink, instance_group_1_selfLink
]
operation = self.test_resource_creator.create_regional_backend_service(
'sample_internal_backend_service.json', backend_service_name,
original_backend_selfLinks)
backend_service_selfLink = operation['targetLink']
### start migration
selfLink_executor = SelfLinkExecutor(
self.compute,
backend_service_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_backend_service_configs = self.google_api_interface.get_regional_backend_service_configs(
backend_service_name)
new_backend_selfLinks = self.google_api_interface.get_backends_links_from_backend_service_configs(
new_backend_service_configs)
# check backend service config
self.assertTrue(
check_selfLink_equal(new_backend_service_configs['network'],
self.test_resource_creator.network_selfLink))
self.assertTrue(
compare_two_list(original_backend_selfLinks,
new_backend_selfLinks))
# check its backends
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 testWithNoBackends(self):
""" A backend service has no backend
Expectation: the backend service itself will be migrated
"""
### create test resources
backend_service_name = 'end-to-end-test-backend-service'
operation = self.test_resource_creator.create_regional_backend_service(
'sample_internal_backend_service.json', backend_service_name, [])
original_backend_service_configs = self.google_api_interface.get_regional_backend_service_configs(
backend_service_name)
backend_service_selfLink = operation['targetLink']
### start migration
selfLink_executor = SelfLinkExecutor(
self.compute,
backend_service_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_backend_service_configs = self.google_api_interface.get_regional_backend_service_configs(
backend_service_name)
# check backend service config
self.assertTrue(
check_selfLink_equal(new_backend_service_configs['network'],
self.test_resource_creator.network_selfLink))
self.assertTrue(
resource_config_is_unchanged_except_for_network(
original_backend_service_configs, new_backend_service_configs))
print('Pass the current test')
def testWithForwardingRuleAttached(self):
""" A forwaring rule is using the external backend service.
Try to migrate the external backend service directly.
Expectation: still can migrate
"""
### create test resources
backend_service_name = 'end-to-end-test-backend-service'
original_backend_selfLinks = []
operation = self.test_resource_creator.create_regional_backend_service(
'sample_internal_backend_service.json', backend_service_name,
original_backend_selfLinks)
original_backend_service_configs = self.google_api_interface.get_regional_backend_service_configs(
backend_service_name)
backend_service_selfLink = operation['targetLink']
self.test_resource_creator.create_regional_forwarding_rule_with_backend_service(
'sample_tcp_regional_forwarding_rule_internal.json',
backend_service_name, backend_service_selfLink)
### start migration
selfLink_executor = SelfLinkExecutor(
self.compute,
backend_service_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
# migration didn't start
new_backend_service_configs = self.google_api_interface.get_regional_backend_service_configs(
backend_service_name)
self.assertTrue(
resource_config_is_unchanged_including_network(
original_backend_service_configs, new_backend_service_configs))
print('Pass the current test')
def testSharingBackendWithAnotherBackendService(self):
""" The internal backend service A shares the same backend with another backend service B
Expectation: rollback
"""
### create test resources
backend_service_1_name = 'end-to-end-test-backend-service-1'
backend_service_2_name = 'end-to-end-test-backend-service-2'
group_name_1 = 'end-to-end-test-managed-instance-group-1'
operation = 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',
)
instance_group_1_selfLink = operation['targetLink'].replace(
'/instanceGroupManagers/', '/instanceGroups/')
original_instance_group_config = self.google_api_interface.get_multi_zone_managed_instance_group_configs(
group_name_1)
backend_service_1_selfLink = \
self.test_resource_creator.create_regional_backend_service(
'sample_internal_backend_service.json',
backend_service_1_name, [instance_group_1_selfLink])[
'targetLink']
original_backend_service_1_configs = self.google_api_interface.get_regional_backend_service_configs(
backend_service_1_name)
self.test_resource_creator.create_regional_backend_service(
'sample_internal_backend_service.json', backend_service_2_name,
[instance_group_1_selfLink])
### start migration
selfLink_executor = SelfLinkExecutor(
self.compute,
backend_service_1_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
new_backend_service_1_configs = self.google_api_interface.get_regional_backend_service_configs(
backend_service_1_name)
# check backend service config
self.assertTrue(
resource_config_is_unchanged_including_network(
original_backend_service_1_configs,
new_backend_service_1_configs))
# check its backends
new_instance_group_config = self.google_api_interface.get_multi_zone_managed_instance_group_configs(
group_name_1)
self.assertTrue(
instance_template_config_is_unchanged(
original_instance_group_config, new_instance_group_config))
print('Pass the current test')
def testSharingBackendWithATargetPool(self):
""" backend_service_1 share the same backend with a target pool.
Try to migrate backend_service_1.
Expectation: rollback
"""
### create test resources
backend_service_1_name = 'end-to-end-test-backend-service-1'
group_name_1 = 'end-to-end-test-managed-instance-group-1'
operation = 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',
)
instance_group_1_selfLink = operation['targetLink'].replace(
'/instanceGroupManagers/', '/instanceGroups/')
original_instance_group_config = self.google_api_interface.get_multi_zone_managed_instance_group_configs(
group_name_1)
backend_service_1_selfLink = \
self.test_resource_creator.create_regional_backend_service(
'sample_internal_backend_service.json',
backend_service_1_name, [instance_group_1_selfLink])[
'targetLink']
original_backend_service_1_configs = self.google_api_interface.get_regional_backend_service_configs(
backend_service_1_name)
target_pool_name = 'end-to-end-test-target-pool'
target_pool_selfLink = \
self.test_resource_creator.create_target_pool_with_health_check(
'sample_target_pool_with_no_instance.json',
target_pool_name,
[group_name_1],
[],
health_check_selfLink=None)['targetLink']
### start migration
selfLink_executor = SelfLinkExecutor(
self.compute,
backend_service_1_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
new_backend_service_1_configs = self.google_api_interface.get_regional_backend_service_configs(
backend_service_1_name)
# check backend service config
self.assertTrue(
resource_config_is_unchanged_including_network(
original_backend_service_1_configs,
new_backend_service_1_configs))
# check its backends
new_instance_group_config = self.google_api_interface.get_multi_zone_managed_instance_group_configs(
group_name_1)
self.assertTrue(
instance_template_config_is_unchanged(
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()