update tests for desired_capacity support.