def test_update_job_items_fleet_containers(mocker, test_type, cur_cntrs, opt_cntrs, expected):
max_cntrs = 2
min_cntrs = 1
mocker.patch.object(fleet_manager, 'ecs')
mocker.patch.object(fleet_manager, 'sqs')
mocker.patch.object(fleet_manager, 'calculate_optimal_job_item_container_capacity')
fleet_manager.ecs.get_total_tasks_in_cluster.return_value = cur_cntrs
fleet_manager.calculate_optimal_job_item_container_capacity.return_value = opt_cntrs
result = fleet_manager.update_job_items_fleet_containers('TestCluster', 'TestQueue', max_cntrs,
min_cntrs, mock.ANY)
assert result == expected
def test_update_job_items_fleet_containers__decrease(mocker):
max_cntrs = 6
min_cntrs = 0
cur_cntrs = 2
opt_cntrs = 0
mocker.patch.object(fleet_manager, 'ecs')
mocker.patch.object(fleet_manager, 'calculate_optimal_job_item_container_capacity')
mocker.patch.object(fleet_manager, 'remove_job_item_containers')
fleet_manager.ecs.get_total_tasks_in_cluster.return_value = cur_cntrs
fleet_manager.calculate_optimal_job_item_container_capacity.return_value = opt_cntrs
result = fleet_manager.update_job_items_fleet_containers('TestCluster', 'TestQueue', max_cntrs,
min_cntrs, 99)
assert result == cur_cntrs-1
fleet_manager.remove_job_item_containers.assert_called_with(cur_cntrs-1)
def test_update_job_items_fleet_containers__unchanged(mocker):
max_cntrs = 6
min_cntrs = 0
cur_cntrs = 1
opt_cntrs = 1
mocker.patch.object(fleet_manager, 'ecs')
mocker.patch.object(fleet_manager, 'calculate_optimal_job_item_container_capacity')
mocker.patch.object(fleet_manager, 'add_job_item_containers')
mocker.patch.object(fleet_manager, 'remove_job_item_containers')
fleet_manager.ecs.get_total_tasks_in_cluster.return_value = cur_cntrs
fleet_manager.calculate_optimal_job_item_container_capacity.return_value = opt_cntrs
result = fleet_manager.update_job_items_fleet_containers('TestCluster', 'TestQueue', max_cntrs,
min_cntrs, mock.ANY)
assert result == opt_cntrs
fleet_manager.add_job_item_containers.assert_not_called()
fleet_manager.remove_job_item_containers.assert_not_called()
