def test_update_jobs_fleet_containers__increase(mocker):
max_cnts = 3
min_cnts = 0
mocker.patch.object(fleet_manager, 'ecs')
mocker.patch.object(fleet_manager, 'calculate_optimal_job_container_capacity')
fleet_manager.ecs.get_service_capacity.return_value = 0
fleet_manager.calculate_optimal_job_container_capacity.return_value = 2
fleet_manager.update_jobs_fleet_containers('TestCluster', 'TestService', max_cnts, min_cnts, 30)
fleet_manager.ecs.set_service_capacity.assert_called_with('TestCluster', 'TestService', 1)
def test_update_jobs_fleet_containers__decrease_under_min(mocker):
max_cnts = 2
min_cnts = 1
mocker.patch.object(fleet_manager, 'ecs')
mocker.patch.object(fleet_manager, 'calculate_optimal_job_container_capacity')
fleet_manager.ecs.get_service_capacity.return_value = 1
fleet_manager.calculate_optimal_job_container_capacity.return_value = 0
fleet_manager.update_jobs_fleet_containers('TestCluster', 'TestService', max_cnts,
min_cnts, mock.ANY)
fleet_manager.ecs.set_service_capacity.assert_not_called()
def test_update_jobs_fleet_containers(mocker, test_type, cur_cnts, opt_cnts, expected):
max_cnts = 2
min_cnts = 1
mocker.patch.object(fleet_manager, 'ecs')
mocker.patch.object(fleet_manager, 'calculate_optimal_job_container_capacity')
fleet_manager.ecs.get_service_capacity.return_value = cur_cnts
fleet_manager.calculate_optimal_job_container_capacity.return_value = opt_cnts
assert fleet_manager.update_jobs_fleet_containers('TestCluster', 'TestService', max_cnts,
min_cnts, mock.ANY) == expected
