XD-1182 Updating stub test class to match new method signatures