def test_update_job_items_fleet_capacity__instances(mocker, monkeypatch):
mocker.patch.object(fleet_manager, 'update_job_items_fleet_containers')
mocker.patch.object(fleet_manager, 'update_job_items_fleet_instances')
mocker.patch.object(fleet_manager, 'autoscaling', autospec=True)
mocker.patch.object(fleet_manager, 'ecs', autospec=True)
monkeypatch.setattr(fleet_manager, 'AUTOSCALING_GROUP_JOB_ITEMS', 'TestGroup')
monkeypatch.setattr(fleet_manager, 'MAX_JOB_ITEM_INSTANCES', 3)
monkeypatch.setattr(fleet_manager, 'MIN_JOB_ITEM_INSTANCES', 0)
monkeypatch.setattr(fleet_manager, 'JOB_ITEM_CONTAINERS_PER_INSTANCE', 2)
fleet_manager.update_job_items_fleet_containers.return_value = 3
fleet_manager.update_job_items_fleet_capacity()
fleet_manager.update_job_items_fleet_instances.assert_called_with('TestGroup',
3, 0, 2, 3)
def test_update_job_items_fleet_capacity__containers(mocker, monkeypatch):
monkeypatch.setattr(fleet_manager, 'ECS_JOB_ITEMS_CLUSTER', 'TestCluster')
monkeypatch.setattr(fleet_manager, 'SQS_JOB_ITEMS_QUEUE', 'TestJobItemsQueue')
monkeypatch.setattr(fleet_manager, 'MAX_JOB_ITEM_CONTAINERS', 6)
monkeypatch.setattr(fleet_manager, 'MIN_JOB_ITEM_CONTAINERS', 0)
monkeypatch.setattr(fleet_manager, 'JOB_ITEMS_PER_CONTAINER', 2)
mocker.patch.object(fleet_manager, 'autoscaling')
mocker.patch.object(fleet_manager, 'ecs')
mocker.patch.object(fleet_manager, 'update_job_items_fleet_containers')
fleet_manager.update_job_items_fleet_containers.return_value = 5
fleet_manager.update_job_items_fleet_capacity()
fleet_manager.update_job_items_fleet_containers.assert_called_with('TestCluster',
'TestJobItemsQueue', 6, 0, 2)
