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)