def test_update_fleet_capacity(mocker): mocker.patch.object(fleet_manager, 'update_jobs_fleet_capacity') mocker.patch.object(fleet_manager, 'update_job_items_fleet_capacity') fleet_manager.update_fleet_capacity() fleet_manager.update_jobs_fleet_capacity.assert_called_with() fleet_manager.update_job_items_fleet_capacity.assert_called_with()
def update_fleet(): count = 0 while count < FLEET_UPDATE_MAX_RUNS: try: tasks_log.info("Update Jobs and JobItems Fleets. Attempt " + str(count)) fleet_manager.update_fleet_capacity() except Exception, err: tasks_log.error(traceback.format_exc()) finally: