fix mig module test

This commit is contained in:
Ludovico Magnocavallo 2020-11-18 08:40:41 +01:00
parent 4c24cf8c68
commit cb4abcafd6
1 changed files with 47 additions and 46 deletions

View File

@ -21,55 +21,56 @@ FIXTURES_DIR = os.path.join(os.path.dirname(__file__), 'fixture')
def test_defaults(plan_runner): def test_defaults(plan_runner):
"Test variable defaults." "Test variable defaults."
_, resources = plan_runner(FIXTURES_DIR) _, resources = plan_runner(FIXTURES_DIR)
assert len(resources) == 1 assert len(resources) == 1
mig = resources[0] mig = resources[0]
assert mig['type'] == 'google_compute_instance_group_manager' assert mig['type'] == 'google_compute_instance_group_manager'
assert mig['values']['target_size'] == 2 assert mig['values']['target_size'] == 2
assert mig['values']['zone'] assert mig['values']['zone']
_, resources = plan_runner(FIXTURES_DIR, regional='true') _, resources = plan_runner(FIXTURES_DIR, regional='true')
assert len(resources) == 1 assert len(resources) == 1
mig = resources[0] mig = resources[0]
assert mig['type'] == 'google_compute_region_instance_group_manager' assert mig['type'] == 'google_compute_region_instance_group_manager'
assert mig['values']['target_size'] == 2 assert mig['values']['target_size'] == 2
assert mig['values']['region'] assert mig['values']['region']
def test_health_check(plan_runner): def test_health_check(plan_runner):
"Test health check resource." "Test health check resource."
health_check_config = '{type="tcp", check={port=80}, config=null, logging=false}' health_check_config = '{type="tcp", check={port=80}, config=null, logging=false}'
_, resources = plan_runner( _, resources = plan_runner(
FIXTURES_DIR, health_check_config=health_check_config) FIXTURES_DIR, health_check_config=health_check_config)
assert len(resources) == 2 assert len(resources) == 2
assert any(r['type'] == 'google_compute_health_check' for r in resources) assert any(r['type'] == 'google_compute_health_check' for r in resources)
def test_autoscaler(plan_runner): def test_autoscaler(plan_runner):
"Test autoscaler resource." "Test autoscaler resource."
autoscaler_config = ( autoscaler_config = (
'{' '{'
'max_replicas=3, min_replicas=1, cooldown_period=60,' 'max_replicas=3, min_replicas=1, cooldown_period=60,'
'cpu_utilization_target=65, load_balancing_utilization_target=null,' 'cpu_utilization_target=65, load_balancing_utilization_target=null,'
'metric=null' 'metric=null'
'}' '}'
) )
_, resources = plan_runner( _, resources = plan_runner(
FIXTURES_DIR, autoscaler_config=autoscaler_config) FIXTURES_DIR, autoscaler_config=autoscaler_config)
assert len(resources) == 2 assert len(resources) == 2
autoscaler = resources[0] autoscaler = resources[0]
assert autoscaler['type'] == 'google_compute_autoscaler' assert autoscaler['type'] == 'google_compute_autoscaler'
assert autoscaler['values']['autoscaling_policy'] == [{ assert autoscaler['values']['autoscaling_policy'] == [{
'cooldown_period': 60, 'cooldown_period': 60,
'cpu_utilization': [{'target': 65}], 'cpu_utilization': [{'target': 65}],
'load_balancing_utilization': [], 'load_balancing_utilization': [],
'max_replicas': 3, 'max_replicas': 3,
'metric': [], 'metric': [],
'min_replicas': 1, 'min_replicas': 1,
'mode': 'ON' 'mode': 'ON',
}] 'scale_in_control': []
_, resources = plan_runner( }]
FIXTURES_DIR, autoscaler_config=autoscaler_config, regional='true') _, resources = plan_runner(
assert len(resources) == 2 FIXTURES_DIR, autoscaler_config=autoscaler_config, regional='true')
autoscaler = resources[0] assert len(resources) == 2
assert autoscaler['type'] == 'google_compute_region_autoscaler' autoscaler = resources[0]
assert autoscaler['type'] == 'google_compute_region_autoscaler'