class Aws::Stubbing::Protocols::Rest

def apply_status_code(operation, resp, data)
  operation.output.shape.members.each do |member_name, member_ref|
    if member_ref.location == 'statusCode'
      resp.status_code = data[member_name] if data.key?(member_name)
    end
  end
end