Client Test

Client Test

rev. ec7b2441254af868911fccffe8d8dca83aff0045

Files changed:

tmp-codegen-diff/codegen-client-test/rest_json_extras/rust-client-codegen/src/operation/empty_struct_with_content_on_wire_op/_empty_struct_with_content_on_wire_op_output.rs

@@ -23,23 +106,117 @@
   43     43   
        ser: &mut dyn ::aws_smithy_schema::serde::ShapeSerializer,
   44     44   
    ) -> ::std::result::Result<(), ::aws_smithy_schema::serde::SerdeError> {
   45     45   
        if let Some(ref val) = self.empty {
   46     46   
            ser.write_struct(&EMPTYSTRUCTWITHCONTENTONWIREOPOUTPUT_MEMBER_EMPTY, val)?;
   47     47   
        }
   48     48   
        Ok(())
   49     49   
    }
   50     50   
}
   51     51   
impl EmptyStructWithContentOnWireOpOutput {
   52     52   
    /// Deserializes this structure from a [`ShapeDeserializer`].
   53         -
    pub fn deserialize<D: ::aws_smithy_schema::serde::ShapeDeserializer>(
   54         -
        deserializer: &mut D,
          53  +
    pub fn deserialize(
          54  +
        deserializer: &mut dyn ::aws_smithy_schema::serde::ShapeDeserializer,
   55     55   
    ) -> ::std::result::Result<Self, ::aws_smithy_schema::serde::SerdeError> {
   56     56   
        #[allow(unused_variables, unused_mut)]
   57     57   
        let mut builder = Self::builder();
   58     58   
        #[allow(
   59     59   
            unused_variables,
   60     60   
            unreachable_code,
   61     61   
            clippy::single_match,
   62     62   
            clippy::match_single_binding,
   63     63   
            clippy::diverging_sub_expression
   64     64   
        )]
   65         -
        deserializer.read_struct(&EMPTYSTRUCTWITHCONTENTONWIREOPOUTPUT_SCHEMA, (), |_, member, deser| {
          65  +
        deserializer.read_struct(&EMPTYSTRUCTWITHCONTENTONWIREOPOUTPUT_SCHEMA, &mut |member, deser| {
   66     66   
            match member.member_index() {
   67     67   
                Some(0) => {
   68     68   
                    builder.empty = Some(crate::types::EmptyStruct::deserialize(deser)?);
   69     69   
                }
   70     70   
                _ => {}
   71     71   
            }
   72     72   
            Ok(())
   73     73   
        })?;
   74     74   
        Ok(builder.build())
   75     75   
    }
   76     76   
}
          77  +
impl EmptyStructWithContentOnWireOpOutput {
          78  +
    /// Deserializes this structure from a body deserializer and HTTP response.
          79  +
    pub fn deserialize_with_response(
          80  +
        deserializer: &mut dyn ::aws_smithy_schema::serde::ShapeDeserializer,
          81  +
        _headers: &::aws_smithy_runtime_api::http::Headers,
          82  +
        _status: u16,
          83  +
        _body: &[u8],
          84  +
    ) -> ::std::result::Result<Self, ::aws_smithy_schema::serde::SerdeError> {
          85  +
        Self::deserialize(deserializer)
          86  +
    }
          87  +
}
   77     88   
impl EmptyStructWithContentOnWireOpOutput {
   78     89   
    /// Creates a new builder-style object to manufacture [`EmptyStructWithContentOnWireOpOutput`](crate::operation::empty_struct_with_content_on_wire_op::EmptyStructWithContentOnWireOpOutput).
   79     90   
    pub fn builder() -> crate::operation::empty_struct_with_content_on_wire_op::builders::EmptyStructWithContentOnWireOpOutputBuilder {
   80     91   
        crate::operation::empty_struct_with_content_on_wire_op::builders::EmptyStructWithContentOnWireOpOutputBuilder::default()
   81     92   
    }
   82     93   
}
   83     94   
   84     95   
/// A builder for [`EmptyStructWithContentOnWireOpOutput`](crate::operation::empty_struct_with_content_on_wire_op::EmptyStructWithContentOnWireOpOutput).
   85     96   
#[derive(::std::clone::Clone, ::std::cmp::PartialEq, ::std::default::Default, ::std::fmt::Debug)]
   86     97   
#[non_exhaustive]

tmp-codegen-diff/codegen-client-test/rest_json_extras/rust-client-codegen/src/operation/enum_query.rs

@@ -1,1 +40,44 @@
    1      1   
// Code generated by software.amazon.smithy.rust.codegen.smithy-rs. DO NOT EDIT.
    2      2   
/// Orchestration and serialization glue logic for `EnumQuery`.
    3      3   
#[derive(::std::clone::Clone, ::std::default::Default, ::std::fmt::Debug)]
    4      4   
#[non_exhaustive]
    5      5   
pub struct EnumQuery;
    6      6   
impl EnumQuery {
    7      7   
    /// Creates a new `EnumQuery`
    8      8   
    pub fn new() -> Self {
    9      9   
        Self
   10     10   
    }
          11  +
    /// The schema for this operation's input shape.
          12  +
    pub const INPUT_SCHEMA: &'static ::aws_smithy_schema::Schema = crate::operation::enum_query::EnumQueryInput::SCHEMA;
          13  +
    /// The schema for this operation's output shape.
          14  +
    pub const OUTPUT_SCHEMA: &'static ::aws_smithy_schema::Schema = crate::operation::enum_query::EnumQueryOutput::SCHEMA;
   11     15   
    pub(crate) async fn orchestrate(
   12     16   
        runtime_plugins: &::aws_smithy_runtime_api::client::runtime_plugin::RuntimePlugins,
   13     17   
        input: crate::operation::enum_query::EnumQueryInput,
   14     18   
    ) -> ::std::result::Result<
   15     19   
        crate::operation::enum_query::EnumQueryOutput,
   16     20   
        ::aws_smithy_runtime_api::client::result::SdkError<
   17     21   
            crate::operation::enum_query::EnumQueryError,
   18     22   
            ::aws_smithy_runtime_api::client::orchestrator::HttpResponse,
   19     23   
        >,
   20     24   
    > {
@@ -104,108 +231,317 @@
  124    128   
                crate::operation::enum_query::EnumQueryError,
  125    129   
            >::new());
  126    130   
  127    131   
        ::std::borrow::Cow::Owned(rcb)
  128    132   
    }
  129    133   
}
  130    134   
  131    135   
#[derive(Debug)]
  132    136   
struct EnumQueryResponseDeserializer;
  133    137   
impl ::aws_smithy_runtime_api::client::ser_de::DeserializeResponse for EnumQueryResponseDeserializer {
  134         -
    fn deserialize_nonstreaming(
         138  +
    fn deserialize_nonstreaming_with_config(
  135    139   
        &self,
  136    140   
        response: &::aws_smithy_runtime_api::client::orchestrator::HttpResponse,
         141  +
        _cfg: &::aws_smithy_types::config_bag::ConfigBag,
  137    142   
    ) -> ::aws_smithy_runtime_api::client::interceptors::context::OutputOrError {
  138    143   
        let (success, status) = (response.status().is_success(), response.status().as_u16());
  139         -
        let headers = response.headers();
  140         -
        let body = response.body().bytes().expect("body loaded");
  141    144   
        #[allow(unused_mut)]
  142    145   
        let mut force_error = false;
  143    146   
  144         -
        let parse_result = if !success && status != 200 || force_error {
  145         -
            crate::protocol_serde::shape_enum_query::de_enum_query_http_error(status, headers, body)
         147  +
        if !success && status != 200 || force_error {
         148  +
            let headers = response.headers();
         149  +
            let body = response.body().bytes().expect("body loaded");
         150  +
            #[allow(unused_mut)]
         151  +
            let mut generic_builder = crate::protocol_serde::parse_http_error_metadata(status, headers, body).map_err(|e| {
         152  +
                ::aws_smithy_runtime_api::client::orchestrator::OrchestratorError::other(::aws_smithy_runtime_api::box_error::BoxError::from(e))
         153  +
            })?;
         154  +
         155  +
            let generic = generic_builder.build();
         156  +
            let error_code = match generic.code() {
         157  +
                ::std::option::Option::Some(code) => code,
         158  +
                ::std::option::Option::None => {
         159  +
                    return ::std::result::Result::Err(::aws_smithy_runtime_api::client::orchestrator::OrchestratorError::other(
         160  +
                        ::aws_smithy_runtime_api::box_error::BoxError::from(crate::operation::enum_query::EnumQueryError::unhandled(generic)),
         161  +
                    ))
         162  +
                }
         163  +
            };
         164  +
            let _error_message = generic.message().map(|msg| msg.to_owned());
         165  +
            let protocol = _cfg
         166  +
                .load::<::aws_smithy_schema::protocol::SharedClientProtocol>()
         167  +
                .expect("a SharedClientProtocol is required");
         168  +
            let err = match error_code {
         169  +
                "ValidationException" => crate::operation::enum_query::EnumQueryError::ValidationError({
         170  +
                    let mut tmp = match protocol
         171  +
                        .deserialize_response(response, crate::types::error::ValidationError::SCHEMA, _cfg)
         172  +
                        .and_then(|mut deser| {
         173  +
                            crate::types::error::ValidationError::deserialize_with_response(
         174  +
                                &mut *deser,
         175  +
                                response.headers(),
         176  +
                                response.status().into(),
         177  +
                                body,
         178  +
                            )
         179  +
                        }) {
         180  +
                        ::std::result::Result::Ok(val) => val,
         181  +
                        ::std::result::Result::Err(e) => {
         182  +
                            return ::std::result::Result::Err(::aws_smithy_runtime_api::client::orchestrator::OrchestratorError::other(
         183  +
                                ::aws_smithy_runtime_api::box_error::BoxError::from(e),
         184  +
                            ))
         185  +
                        }
         186  +
                    };
         187  +
                    tmp.meta = generic;
         188  +
                    tmp
         189  +
                }),
         190  +
                "ExtraError" => crate::operation::enum_query::EnumQueryError::ExtraError({
         191  +
                    let mut tmp = match protocol
         192  +
                        .deserialize_response(response, crate::types::error::ExtraError::SCHEMA, _cfg)
         193  +
                        .and_then(|mut deser| {
         194  +
                            crate::types::error::ExtraError::deserialize_with_response(
         195  +
                                &mut *deser,
         196  +
                                response.headers(),
         197  +
                                response.status().into(),
         198  +
                                body,
         199  +
                            )
         200  +
                        }) {
         201  +
                        ::std::result::Result::Ok(val) => val,
         202  +
                        ::std::result::Result::Err(e) => {
         203  +
                            return ::std::result::Result::Err(::aws_smithy_runtime_api::client::orchestrator::OrchestratorError::other(
         204  +
                                ::aws_smithy_runtime_api::box_error::BoxError::from(e),
         205  +
                            ))
         206  +
                        }
         207  +
                    };
         208  +
                    tmp.meta = generic;
         209  +
                    if tmp.message.is_none() {
         210  +
                        tmp.message = _error_message;
         211  +
                    }
         212  +
                    tmp
         213  +
                }),
         214  +
                _ => crate::operation::enum_query::EnumQueryError::generic(generic),
         215  +
            };
         216  +
            ::std::result::Result::Err(::aws_smithy_runtime_api::client::orchestrator::OrchestratorError::operation(
         217  +
                ::aws_smithy_runtime_api::client::interceptors::context::Error::erase(err),
         218  +
            ))
  146    219   
        } else {
  147         -
            crate::protocol_serde::shape_enum_query::de_enum_query_http_response(status, headers, body)
  148         -
        };
  149         -
        crate::protocol_serde::type_erase_result(parse_result)
         220  +
            let protocol = _cfg
         221  +
                .load::<::aws_smithy_schema::protocol::SharedClientProtocol>()
         222  +
                .expect("a SharedClientProtocol is required");
         223  +
            let mut deser = protocol.deserialize_response(response, EnumQuery::OUTPUT_SCHEMA, _cfg).map_err(|e| {
         224  +
                ::aws_smithy_runtime_api::client::orchestrator::OrchestratorError::other(::aws_smithy_runtime_api::box_error::BoxError::from(e))
         225  +
            })?;
         226  +
            let body = response.body().bytes().expect("body loaded");
         227  +
            let output = crate::operation::enum_query::EnumQueryOutput::deserialize_with_response(
         228  +
                &mut *deser,
         229  +
                response.headers(),
         230  +
                response.status().into(),
         231  +
                body,
         232  +
            )
         233  +
            .map_err(|e| {
         234  +
                ::aws_smithy_runtime_api::client::orchestrator::OrchestratorError::other(::aws_smithy_runtime_api::box_error::BoxError::from(e))
         235  +
            })?;
         236  +
            ::std::result::Result::Ok(::aws_smithy_runtime_api::client::interceptors::context::Output::erase(output))
         237  +
        }
  150    238   
    }
  151    239   
}
  152    240   
#[derive(Debug)]
  153    241   
struct EnumQueryRequestSerializer;
  154    242   
impl ::aws_smithy_runtime_api::client::ser_de::SerializeRequest for EnumQueryRequestSerializer {
  155    243   
    #[allow(unused_mut, clippy::let_and_return, clippy::needless_borrow, clippy::useless_conversion)]
  156    244   
    fn serialize_input(
  157    245   
        &self,
  158    246   
        input: ::aws_smithy_runtime_api::client::interceptors::context::Input,
  159    247   
        _cfg: &mut ::aws_smithy_types::config_bag::ConfigBag,
  160    248   
    ) -> ::std::result::Result<::aws_smithy_runtime_api::client::orchestrator::HttpRequest, ::aws_smithy_runtime_api::box_error::BoxError> {
  161    249   
        let input = input.downcast::<crate::operation::enum_query::EnumQueryInput>().expect("correct type");
  162         -
        let _header_serialization_settings = _cfg
  163         -
            .load::<crate::serialization_settings::HeaderSerializationSettings>()
  164         -
            .cloned()
  165         -
            .unwrap_or_default();
  166         -
        let mut request_builder = {
  167         -
            #[allow(clippy::uninlined_format_args)]
  168         -
            fn uri_base(
  169         -
                _input: &crate::operation::enum_query::EnumQueryInput,
  170         -
                output: &mut ::std::string::String,
  171         -
            ) -> ::std::result::Result<(), ::aws_smithy_types::error::operation::BuildError> {
  172         -
                use ::std::fmt::Write as _;
  173         -
                let input_1 = &_input.r#enum;
  174         -
                let input_1 = input_1
  175         -
                    .as_ref()
  176         -
                    .ok_or_else(|| ::aws_smithy_types::error::operation::BuildError::missing_field("r#enum", "cannot be empty or unset"))?;
  177         -
                let r#enum = ::aws_smithy_http::label::fmt_string(input_1.as_str(), ::aws_smithy_http::label::EncodingStrategy::Default);
  178         -
                if r#enum.is_empty() {
  179         -
                    return ::std::result::Result::Err(::aws_smithy_types::error::operation::BuildError::missing_field(
  180         -
                        "r#enum",
  181         -
                        "cannot be empty or unset",
  182         -
                    ));
         250  +
        let protocol = _cfg
         251  +
            .load::<::aws_smithy_schema::protocol::SharedClientProtocol>()
         252  +
            .expect("a SharedClientProtocol is required");
         253  +
        if protocol.supports_http_bindings() {
         254  +
            let mut request = protocol
         255  +
                .serialize_body(&input, EnumQuery::INPUT_SCHEMA, "", _cfg)
         256  +
                .map_err(::aws_smithy_runtime_api::box_error::BoxError::from)?;
         257  +
            {
         258  +
                let mut uri = "/foo/{enum}".to_string();
         259  +
                let mut query_params: Vec<(String, String)> = Vec::new();
         260  +
                if let Some(ref val) = input.r#enum {
         261  +
                    uri = uri.replace("{enum}", &::aws_smithy_schema::http_protocol::percent_encode(&val.as_str()));
  183    262   
                }
  184         -
                ::std::write!(output, "/foo/{enum}", enum = r#enum).expect("formatting should succeed");
  185         -
                ::std::result::Result::Ok(())
  186         -
            }
  187         -
            #[allow(clippy::unnecessary_wraps)]
  188         -
            fn update_http_builder(
  189         -
                input: &crate::operation::enum_query::EnumQueryInput,
  190         -
                builder: ::http_1x::request::Builder,
  191         -
            ) -> ::std::result::Result<::http_1x::request::Builder, ::aws_smithy_types::error::operation::BuildError> {
  192         -
                let mut uri = ::std::string::String::new();
  193         -
                uri_base(input, &mut uri)?;
  194         -
                ::std::result::Result::Ok(builder.method("GET").uri(uri))
         263  +
                if !query_params.is_empty() {
         264  +
                    uri.push(if uri.contains('?') { '&' } else { '?' });
         265  +
                    let pairs: Vec<String> = query_params
         266  +
                        .iter()
         267  +
                        .map(|(k, v)| {
         268  +
                            format!(
         269  +
                                "{}={}",
         270  +
                                ::aws_smithy_schema::http_protocol::percent_encode(k),
         271  +
                                ::aws_smithy_schema::http_protocol::percent_encode(v)
         272  +
                            )
         273  +
                        })
         274  +
                        .collect();
         275  +
                    uri.push_str(&pairs.join("&"));
         276  +
                }
         277  +
                request.set_uri(uri.as_str()).expect("valid URI");
  195    278   
            }
  196         -
            let mut builder = update_http_builder(&input, ::http_1x::request::Builder::new())?;
  197         -
            builder
  198         -
        };
  199         -
        let body = ::aws_smithy_types::body::SdkBody::from("");
  200    279   
  201         -
        ::std::result::Result::Ok(request_builder.body(body).expect("valid request").try_into().unwrap())
         280  +
            return ::std::result::Result::Ok(request);
         281  +
        } else {
         282  +
            let mut request = protocol
         283  +
                .serialize_request(&input, EnumQuery::INPUT_SCHEMA, "", _cfg)
         284  +
                .map_err(::aws_smithy_runtime_api::box_error::BoxError::from)?;
         285  +
         286  +
            return ::std::result::Result::Ok(request);
         287  +
        }
  202    288   
    }
  203    289   
}
  204    290   
#[derive(Debug)]
  205    291   
struct EnumQueryEndpointParamsInterceptor;
  206    292   
  207    293   
impl ::aws_smithy_runtime_api::client::interceptors::Intercept for EnumQueryEndpointParamsInterceptor {
  208    294   
    fn name(&self) -> &'static str {
  209    295   
        "EnumQueryEndpointParamsInterceptor"
  210    296   
    }
  211    297   
@@ -271,357 +331,427 @@
  291    377   
            .expect("the config must have a deserializer");
  292    378   
  293    379   
        let parsed = de.deserialize_streaming(&mut http_response);
  294    380   
        let parsed = parsed.unwrap_or_else(|| {
  295    381   
            let http_response = http_response.map(|body| {
  296    382   
                ::aws_smithy_types::body::SdkBody::from(::bytes::Bytes::copy_from_slice(&::aws_smithy_protocol_test::decode_body_data(
  297    383   
                    body.bytes().unwrap(),
  298    384   
                    ::aws_smithy_protocol_test::MediaType::from("unknown"),
  299    385   
                )))
  300    386   
            });
  301         -
            de.deserialize_nonstreaming(&http_response)
         387  +
            // Build a config bag with the protocol for schema-based deserialization
         388  +
            #[allow(unused_mut)]
         389  +
            let mut test_cfg = ::aws_smithy_types::config_bag::ConfigBag::base();
         390  +
            {
         391  +
                let mut layer = ::aws_smithy_types::config_bag::Layer::new("test_protocol");
         392  +
                layer.store_put(::aws_smithy_schema::protocol::SharedClientProtocol::new(
         393  +
                    ::aws_smithy_json::protocol::aws_rest_json_1::AwsRestJsonProtocol::new(),
         394  +
                ));
         395  +
                test_cfg.push_shared_layer(layer.freeze());
         396  +
            }
         397  +
            de.deserialize_nonstreaming_with_config(&http_response, &test_cfg)
  302    398   
        });
  303    399   
        let parsed = parsed.expect_err("should be error response");
  304    400   
        let parsed: &crate::operation::enum_query::EnumQueryError = parsed.as_operation_error().expect("operation error").downcast_ref().unwrap();
  305    401   
        if let crate::operation::enum_query::EnumQueryError::ExtraError(parsed) = parsed {
  306    402   
            ::pretty_assertions::assert_eq!(parsed.message, expected_output.message, "Unexpected value for `message`");
  307    403   
        } else {
  308    404   
            panic!("wrong variant: Got: {:?}. Expected: {:?}", parsed, expected_output);
  309    405   
        }
  310    406   
    }
  311    407   
}

tmp-codegen-diff/codegen-client-test/rest_json_extras/rust-client-codegen/src/operation/enum_query/_enum_query_input.rs

@@ -1,1 +109,121 @@
   17     17   
    "aws.protocoltests.restjson.synthetic",
   18     18   
    "EnumQueryInput",
   19     19   
);
   20     20   
static ENUMQUERYINPUT_MEMBER_ENUM: ::aws_smithy_schema::Schema = ::aws_smithy_schema::Schema::new_member(
   21     21   
    ::aws_smithy_schema::ShapeId::from_static(
   22     22   
        "aws.protocoltests.restjson.synthetic#EnumQueryInput$enum",
   23     23   
        "aws.protocoltests.restjson.synthetic",
   24     24   
        "EnumQueryInput",
   25     25   
    ),
   26     26   
    ::aws_smithy_schema::ShapeType::String,
   27         -
    "r##enum",
          27  +
    "enum",
   28     28   
    0,
   29     29   
)
   30     30   
.with_http_label();
   31     31   
static ENUMQUERYINPUT_SCHEMA: ::aws_smithy_schema::Schema = ::aws_smithy_schema::Schema::new_struct(
   32     32   
    ENUMQUERYINPUT_SCHEMA_ID,
   33     33   
    ::aws_smithy_schema::ShapeType::Structure,
   34     34   
    &[&ENUMQUERYINPUT_MEMBER_ENUM],
   35         -
);
          35  +
)
          36  +
.with_http(aws_smithy_schema::traits::HttpTrait::new("GET", "/foo/{enum}", None));
   36     37   
impl EnumQueryInput {
   37     38   
    /// The schema for this shape.
   38     39   
    pub const SCHEMA: &'static ::aws_smithy_schema::Schema = &ENUMQUERYINPUT_SCHEMA;
   39     40   
}
   40     41   
impl ::aws_smithy_schema::serde::SerializableStruct for EnumQueryInput {
   41     42   
    #[allow(unused_variables, clippy::diverging_sub_expression)]
   42     43   
    fn serialize_members(
   43     44   
        &self,
   44     45   
        ser: &mut dyn ::aws_smithy_schema::serde::ShapeSerializer,
   45     46   
    ) -> ::std::result::Result<(), ::aws_smithy_schema::serde::SerdeError> {
   46     47   
        if let Some(ref val) = self.r#enum {
   47     48   
            ser.write_string(&ENUMQUERYINPUT_MEMBER_ENUM, val.as_str())?;
   48     49   
        }
   49     50   
        Ok(())
   50     51   
    }
   51     52   
}
   52     53   
impl EnumQueryInput {
   53     54   
    /// Deserializes this structure from a [`ShapeDeserializer`].
   54         -
    pub fn deserialize<D: ::aws_smithy_schema::serde::ShapeDeserializer>(
   55         -
        deserializer: &mut D,
          55  +
    pub fn deserialize(
          56  +
        deserializer: &mut dyn ::aws_smithy_schema::serde::ShapeDeserializer,
   56     57   
    ) -> ::std::result::Result<Self, ::aws_smithy_schema::serde::SerdeError> {
   57     58   
        #[allow(unused_variables, unused_mut)]
   58     59   
        let mut builder = Self::builder();
   59     60   
        #[allow(
   60     61   
            unused_variables,
   61     62   
            unreachable_code,
   62     63   
            clippy::single_match,
   63     64   
            clippy::match_single_binding,
   64     65   
            clippy::diverging_sub_expression
   65     66   
        )]
   66         -
        deserializer.read_struct(&ENUMQUERYINPUT_SCHEMA, (), |_, member, deser| {
          67  +
        deserializer.read_struct(&ENUMQUERYINPUT_SCHEMA, &mut |member, deser| {
   67     68   
            match member.member_index() {
   68     69   
                Some(0) => {
   69     70   
                    builder.r#enum = Some(crate::types::StringEnum::from(deser.read_string(member)?.as_str()));
   70     71   
                }
   71     72   
                _ => {}
   72     73   
            }
   73     74   
            Ok(())
   74     75   
        })?;
   75     76   
        builder
   76     77   
            .build()
   77     78   
            .map_err(|e| aws_smithy_schema::serde::SerdeError::Custom { message: e.to_string() })
   78     79   
    }
   79     80   
}
          81  +
impl EnumQueryInput {
          82  +
    /// Deserializes this structure from a body deserializer and HTTP response.
          83  +
    pub fn deserialize_with_response(
          84  +
        deserializer: &mut dyn ::aws_smithy_schema::serde::ShapeDeserializer,
          85  +
        _headers: &::aws_smithy_runtime_api::http::Headers,
          86  +
        _status: u16,
          87  +
        _body: &[u8],
          88  +
    ) -> ::std::result::Result<Self, ::aws_smithy_schema::serde::SerdeError> {
          89  +
        Self::deserialize(deserializer)
          90  +
    }
          91  +
}
   80     92   
impl EnumQueryInput {
   81     93   
    /// Creates a new builder-style object to manufacture [`EnumQueryInput`](crate::operation::enum_query::EnumQueryInput).
   82     94   
    pub fn builder() -> crate::operation::enum_query::builders::EnumQueryInputBuilder {
   83     95   
        crate::operation::enum_query::builders::EnumQueryInputBuilder::default()
   84     96   
    }
   85     97   
}
   86     98   
   87     99   
/// A builder for [`EnumQueryInput`](crate::operation::enum_query::EnumQueryInput).
   88    100   
#[derive(::std::clone::Clone, ::std::cmp::PartialEq, ::std::default::Default, ::std::fmt::Debug)]
   89    101   
#[non_exhaustive]

tmp-codegen-diff/codegen-client-test/rest_json_extras/rust-client-codegen/src/operation/enum_query/_enum_query_output.rs

@@ -1,1 +65,76 @@
   18     18   
    #[allow(unused_variables, clippy::diverging_sub_expression)]
   19     19   
    fn serialize_members(
   20     20   
        &self,
   21     21   
        ser: &mut dyn ::aws_smithy_schema::serde::ShapeSerializer,
   22     22   
    ) -> ::std::result::Result<(), ::aws_smithy_schema::serde::SerdeError> {
   23     23   
        Ok(())
   24     24   
    }
   25     25   
}
   26     26   
impl EnumQueryOutput {
   27     27   
    /// Deserializes this structure from a [`ShapeDeserializer`].
   28         -
    pub fn deserialize<D: ::aws_smithy_schema::serde::ShapeDeserializer>(
   29         -
        deserializer: &mut D,
          28  +
    pub fn deserialize(
          29  +
        deserializer: &mut dyn ::aws_smithy_schema::serde::ShapeDeserializer,
   30     30   
    ) -> ::std::result::Result<Self, ::aws_smithy_schema::serde::SerdeError> {
   31     31   
        #[allow(unused_variables, unused_mut)]
   32     32   
        let mut builder = Self::builder();
   33     33   
        #[allow(
   34     34   
            unused_variables,
   35     35   
            unreachable_code,
   36     36   
            clippy::single_match,
   37     37   
            clippy::match_single_binding,
   38     38   
            clippy::diverging_sub_expression
   39     39   
        )]
   40         -
        deserializer.read_struct(&ENUMQUERYOUTPUT_SCHEMA, (), |_, member, deser| {
          40  +
        deserializer.read_struct(&ENUMQUERYOUTPUT_SCHEMA, &mut |member, deser| {
   41     41   
            match member.member_index() {
   42     42   
                _ => {}
   43     43   
            }
   44     44   
            Ok(())
   45     45   
        })?;
   46     46   
        Ok(builder.build())
   47     47   
    }
   48     48   
}
          49  +
impl EnumQueryOutput {
          50  +
    /// Deserializes this structure from a body deserializer and HTTP response.
          51  +
    pub fn deserialize_with_response(
          52  +
        _deserializer: &mut dyn ::aws_smithy_schema::serde::ShapeDeserializer,
          53  +
        _headers: &::aws_smithy_runtime_api::http::Headers,
          54  +
        _status: u16,
          55  +
        _body: &[u8],
          56  +
    ) -> ::std::result::Result<Self, ::aws_smithy_schema::serde::SerdeError> {
          57  +
        Ok(Self::builder().build())
          58  +
    }
          59  +
}
   49     60   
impl EnumQueryOutput {
   50     61   
    /// Creates a new builder-style object to manufacture [`EnumQueryOutput`](crate::operation::enum_query::EnumQueryOutput).
   51     62   
    pub fn builder() -> crate::operation::enum_query::builders::EnumQueryOutputBuilder {
   52     63   
        crate::operation::enum_query::builders::EnumQueryOutputBuilder::default()
   53     64   
    }
   54     65   
}
   55     66   
   56     67   
/// A builder for [`EnumQueryOutput`](crate::operation::enum_query::EnumQueryOutput).
   57     68   
#[derive(::std::clone::Clone, ::std::cmp::PartialEq, ::std::default::Default, ::std::fmt::Debug)]
   58     69   
#[non_exhaustive]

tmp-codegen-diff/codegen-client-test/rest_json_extras/rust-client-codegen/src/operation/escaped_string_values.rs

@@ -1,1 +40,44 @@
    1      1   
// Code generated by software.amazon.smithy.rust.codegen.smithy-rs. DO NOT EDIT.
    2      2   
/// Orchestration and serialization glue logic for `EscapedStringValues`.
    3      3   
#[derive(::std::clone::Clone, ::std::default::Default, ::std::fmt::Debug)]
    4      4   
#[non_exhaustive]
    5      5   
pub struct EscapedStringValues;
    6      6   
impl EscapedStringValues {
    7      7   
    /// Creates a new `EscapedStringValues`
    8      8   
    pub fn new() -> Self {
    9      9   
        Self
   10     10   
    }
          11  +
    /// The schema for this operation's input shape.
          12  +
    pub const INPUT_SCHEMA: &'static ::aws_smithy_schema::Schema = crate::operation::escaped_string_values::EscapedStringValuesInput::SCHEMA;
          13  +
    /// The schema for this operation's output shape.
          14  +
    pub const OUTPUT_SCHEMA: &'static ::aws_smithy_schema::Schema = crate::operation::escaped_string_values::EscapedStringValuesOutput::SCHEMA;
   11     15   
    pub(crate) async fn orchestrate(
   12     16   
        runtime_plugins: &::aws_smithy_runtime_api::client::runtime_plugin::RuntimePlugins,
   13     17   
        input: crate::operation::escaped_string_values::EscapedStringValuesInput,
   14     18   
    ) -> ::std::result::Result<
   15     19   
        crate::operation::escaped_string_values::EscapedStringValuesOutput,
   16     20   
        ::aws_smithy_runtime_api::client::result::SdkError<
   17     21   
            crate::operation::escaped_string_values::EscapedStringValuesError,
   18     22   
            ::aws_smithy_runtime_api::client::orchestrator::HttpResponse,
   19     23   
        >,
   20     24   
    > {
@@ -110,114 +234,299 @@
  130    134   
                crate::operation::escaped_string_values::EscapedStringValuesError,
  131    135   
            >::new());
  132    136   
  133    137   
        ::std::borrow::Cow::Owned(rcb)
  134    138   
    }
  135    139   
}
  136    140   
  137    141   
#[derive(Debug)]
  138    142   
struct EscapedStringValuesResponseDeserializer;
  139    143   
impl ::aws_smithy_runtime_api::client::ser_de::DeserializeResponse for EscapedStringValuesResponseDeserializer {
  140         -
    fn deserialize_nonstreaming(
         144  +
    fn deserialize_nonstreaming_with_config(
  141    145   
        &self,
  142    146   
        response: &::aws_smithy_runtime_api::client::orchestrator::HttpResponse,
         147  +
        _cfg: &::aws_smithy_types::config_bag::ConfigBag,
  143    148   
    ) -> ::aws_smithy_runtime_api::client::interceptors::context::OutputOrError {
  144    149   
        let (success, status) = (response.status().is_success(), response.status().as_u16());
  145         -
        let headers = response.headers();
  146         -
        let body = response.body().bytes().expect("body loaded");
  147    150   
        #[allow(unused_mut)]
  148    151   
        let mut force_error = false;
  149    152   
  150         -
        let parse_result = if !success && status != 200 || force_error {
  151         -
            crate::protocol_serde::shape_escaped_string_values::de_escaped_string_values_http_error(status, headers, body)
         153  +
        if !success && status != 200 || force_error {
         154  +
            let headers = response.headers();
         155  +
            let body = response.body().bytes().expect("body loaded");
         156  +
            #[allow(unused_mut)]
         157  +
            let mut generic_builder = crate::protocol_serde::parse_http_error_metadata(status, headers, body).map_err(|e| {
         158  +
                ::aws_smithy_runtime_api::client::orchestrator::OrchestratorError::other(::aws_smithy_runtime_api::box_error::BoxError::from(e))
         159  +
            })?;
         160  +
         161  +
            let generic = generic_builder.build();
         162  +
            let error_code = match generic.code() {
         163  +
                ::std::option::Option::Some(code) => code,
         164  +
                ::std::option::Option::None => {
         165  +
                    return ::std::result::Result::Err(::aws_smithy_runtime_api::client::orchestrator::OrchestratorError::other(
         166  +
                        ::aws_smithy_runtime_api::box_error::BoxError::from(
         167  +
                            crate::operation::escaped_string_values::EscapedStringValuesError::unhandled(generic),
         168  +
                        ),
         169  +
                    ))
         170  +
                }
         171  +
            };
         172  +
            let _error_message = generic.message().map(|msg| msg.to_owned());
         173  +
            let protocol = _cfg
         174  +
                .load::<::aws_smithy_schema::protocol::SharedClientProtocol>()
         175  +
                .expect("a SharedClientProtocol is required");
         176  +
            let err = match error_code {
         177  +
                "ValidationException" => crate::operation::escaped_string_values::EscapedStringValuesError::ValidationError({
         178  +
                    let mut tmp = match protocol
         179  +
                        .deserialize_response(response, crate::types::error::ValidationError::SCHEMA, _cfg)
         180  +
                        .and_then(|mut deser| {
         181  +
                            crate::types::error::ValidationError::deserialize_with_response(
         182  +
                                &mut *deser,
         183  +
                                response.headers(),
         184  +
                                response.status().into(),
         185  +
                                body,
         186  +
                            )
         187  +
                        }) {
         188  +
                        ::std::result::Result::Ok(val) => val,
         189  +
                        ::std::result::Result::Err(e) => {
         190  +
                            return ::std::result::Result::Err(::aws_smithy_runtime_api::client::orchestrator::OrchestratorError::other(
         191  +
                                ::aws_smithy_runtime_api::box_error::BoxError::from(e),
         192  +
                            ))
         193  +
                        }
         194  +
                    };
         195  +
                    tmp.meta = generic;
         196  +
                    tmp
         197  +
                }),
         198  +
                "ExtraError" => crate::operation::escaped_string_values::EscapedStringValuesError::ExtraError({
         199  +
                    let mut tmp = match protocol
         200  +
                        .deserialize_response(response, crate::types::error::ExtraError::SCHEMA, _cfg)
         201  +
                        .and_then(|mut deser| {
         202  +
                            crate::types::error::ExtraError::deserialize_with_response(
         203  +
                                &mut *deser,
         204  +
                                response.headers(),
         205  +
                                response.status().into(),
         206  +
                                body,
         207  +
                            )
         208  +
                        }) {
         209  +
                        ::std::result::Result::Ok(val) => val,
         210  +
                        ::std::result::Result::Err(e) => {
         211  +
                            return ::std::result::Result::Err(::aws_smithy_runtime_api::client::orchestrator::OrchestratorError::other(
         212  +
                                ::aws_smithy_runtime_api::box_error::BoxError::from(e),
         213  +
                            ))
         214  +
                        }
         215  +
                    };
         216  +
                    tmp.meta = generic;
         217  +
                    if tmp.message.is_none() {
         218  +
                        tmp.message = _error_message;
         219  +
                    }
         220  +
                    tmp
         221  +
                }),
         222  +
                _ => crate::operation::escaped_string_values::EscapedStringValuesError::generic(generic),
         223  +
            };
         224  +
            ::std::result::Result::Err(::aws_smithy_runtime_api::client::orchestrator::OrchestratorError::operation(
         225  +
                ::aws_smithy_runtime_api::client::interceptors::context::Error::erase(err),
         226  +
            ))
  152    227   
        } else {
  153         -
            crate::protocol_serde::shape_escaped_string_values::de_escaped_string_values_http_response(status, headers, body)
  154         -
        };
  155         -
        crate::protocol_serde::type_erase_result(parse_result)
         228  +
            let protocol = _cfg
         229  +
                .load::<::aws_smithy_schema::protocol::SharedClientProtocol>()
         230  +
                .expect("a SharedClientProtocol is required");
         231  +
            let mut deser = protocol
         232  +
                .deserialize_response(response, EscapedStringValues::OUTPUT_SCHEMA, _cfg)
         233  +
                .map_err(|e| {
         234  +
                    ::aws_smithy_runtime_api::client::orchestrator::OrchestratorError::other(::aws_smithy_runtime_api::box_error::BoxError::from(e))
         235  +
                })?;
         236  +
            let body = response.body().bytes().expect("body loaded");
         237  +
            let output = crate::operation::escaped_string_values::EscapedStringValuesOutput::deserialize_with_response(
         238  +
                &mut *deser,
         239  +
                response.headers(),
         240  +
                response.status().into(),
         241  +
                body,
         242  +
            )
         243  +
            .map_err(|e| {
         244  +
                ::aws_smithy_runtime_api::client::orchestrator::OrchestratorError::other(::aws_smithy_runtime_api::box_error::BoxError::from(e))
         245  +
            })?;
         246  +
            ::std::result::Result::Ok(::aws_smithy_runtime_api::client::interceptors::context::Output::erase(output))
         247  +
        }
  156    248   
    }
  157    249   
}
  158    250   
#[derive(Debug)]
  159    251   
struct EscapedStringValuesRequestSerializer;
  160    252   
impl ::aws_smithy_runtime_api::client::ser_de::SerializeRequest for EscapedStringValuesRequestSerializer {
  161    253   
    #[allow(unused_mut, clippy::let_and_return, clippy::needless_borrow, clippy::useless_conversion)]
  162    254   
    fn serialize_input(
  163    255   
        &self,
  164    256   
        input: ::aws_smithy_runtime_api::client::interceptors::context::Input,
  165    257   
        _cfg: &mut ::aws_smithy_types::config_bag::ConfigBag,
  166    258   
    ) -> ::std::result::Result<::aws_smithy_runtime_api::client::orchestrator::HttpRequest, ::aws_smithy_runtime_api::box_error::BoxError> {
  167    259   
        let input = input
  168    260   
            .downcast::<crate::operation::escaped_string_values::EscapedStringValuesInput>()
  169    261   
            .expect("correct type");
  170         -
        let _header_serialization_settings = _cfg
  171         -
            .load::<crate::serialization_settings::HeaderSerializationSettings>()
  172         -
            .cloned()
  173         -
            .unwrap_or_default();
  174         -
        let mut request_builder = {
  175         -
            #[allow(clippy::uninlined_format_args)]
  176         -
            fn uri_base(
  177         -
                _input: &crate::operation::escaped_string_values::EscapedStringValuesInput,
  178         -
                output: &mut ::std::string::String,
  179         -
            ) -> ::std::result::Result<(), ::aws_smithy_types::error::operation::BuildError> {
  180         -
                use ::std::fmt::Write as _;
  181         -
                ::std::write!(output, "/escaped-string-values").expect("formatting should succeed");
  182         -
                ::std::result::Result::Ok(())
  183         -
            }
  184         -
            #[allow(clippy::unnecessary_wraps)]
  185         -
            fn update_http_builder(
  186         -
                input: &crate::operation::escaped_string_values::EscapedStringValuesInput,
  187         -
                builder: ::http_1x::request::Builder,
  188         -
            ) -> ::std::result::Result<::http_1x::request::Builder, ::aws_smithy_types::error::operation::BuildError> {
  189         -
                let mut uri = ::std::string::String::new();
  190         -
                uri_base(input, &mut uri)?;
  191         -
                ::std::result::Result::Ok(builder.method("POST").uri(uri))
  192         -
            }
  193         -
            let mut builder = update_http_builder(&input, ::http_1x::request::Builder::new())?;
  194         -
            builder = _header_serialization_settings.set_default_header(builder, ::http_1x::header::CONTENT_TYPE, "application/json");
  195         -
            builder
  196         -
        };
  197         -
        let body = ::aws_smithy_types::body::SdkBody::from(crate::protocol_serde::shape_escaped_string_values::ser_escaped_string_values_input(
  198         -
            &input,
  199         -
        )?);
  200         -
        if let Some(content_length) = body.content_length() {
  201         -
            let content_length = content_length.to_string();
  202         -
            request_builder = _header_serialization_settings.set_default_header(request_builder, ::http_1x::header::CONTENT_LENGTH, &content_length);
  203         -
        }
  204         -
        ::std::result::Result::Ok(request_builder.body(body).expect("valid request").try_into().unwrap())
         262  +
        let protocol = _cfg
         263  +
            .load::<::aws_smithy_schema::protocol::SharedClientProtocol>()
         264  +
            .expect("a SharedClientProtocol is required");
         265  +
        let mut request = protocol
         266  +
            .serialize_request(&input, EscapedStringValues::INPUT_SCHEMA, "", _cfg)
         267  +
            .map_err(::aws_smithy_runtime_api::box_error::BoxError::from)?;
         268  +
         269  +
        return ::std::result::Result::Ok(request);
  205    270   
    }
  206    271   
}
  207    272   
#[derive(Debug)]
  208    273   
struct EscapedStringValuesEndpointParamsInterceptor;
  209    274   
  210    275   
impl ::aws_smithy_runtime_api::client::interceptors::Intercept for EscapedStringValuesEndpointParamsInterceptor {
  211    276   
    fn name(&self) -> &'static str {
  212    277   
        "EscapedStringValuesEndpointParamsInterceptor"
  213    278   
    }
  214    279   
@@ -291,356 +392,477 @@
  311    376   
            .expect("the config must have a deserializer");
  312    377   
  313    378   
        let parsed = de.deserialize_streaming(&mut http_response);
  314    379   
        let parsed = parsed.unwrap_or_else(|| {
  315    380   
            let http_response = http_response.map(|body| {
  316    381   
                ::aws_smithy_types::body::SdkBody::from(::bytes::Bytes::copy_from_slice(&::aws_smithy_protocol_test::decode_body_data(
  317    382   
                    body.bytes().unwrap(),
  318    383   
                    ::aws_smithy_protocol_test::MediaType::from("unknown"),
  319    384   
                )))
  320    385   
            });
  321         -
            de.deserialize_nonstreaming(&http_response)
         386  +
            // Build a config bag with the protocol for schema-based deserialization
         387  +
            #[allow(unused_mut)]
         388  +
            let mut test_cfg = ::aws_smithy_types::config_bag::ConfigBag::base();
         389  +
            {
         390  +
                let mut layer = ::aws_smithy_types::config_bag::Layer::new("test_protocol");
         391  +
                layer.store_put(::aws_smithy_schema::protocol::SharedClientProtocol::new(
         392  +
                    ::aws_smithy_json::protocol::aws_rest_json_1::AwsRestJsonProtocol::new(),
         393  +
                ));
         394  +
                test_cfg.push_shared_layer(layer.freeze());
         395  +
            }
         396  +
            de.deserialize_nonstreaming_with_config(&http_response, &test_cfg)
  322    397   
        });
  323    398   
        let parsed = parsed
  324    399   
            .expect("should be successful response")
  325    400   
            .downcast::<crate::operation::escaped_string_values::EscapedStringValuesOutput>()
  326    401   
            .unwrap();
  327    402   
        ::pretty_assertions::assert_eq!(parsed.r#enum, expected_output.r#enum, "Unexpected value for `r#enum`");
  328    403   
        ::pretty_assertions::assert_eq!(parsed.some_string, expected_output.some_string, "Unexpected value for `some_string`");
  329    404   
    }
  330    405   
  331    406   
    /// Upper case error modeled lower case
  332    407   
    /// Test ID: ServiceLevelErrorClient
  333    408   
    #[::tokio::test]
  334    409   
    #[::tracing_test::traced_test]
  335    410   
    async fn service_level_error_client_response() {
  336    411   
        let expected_output = crate::types::error::ExtraError::builder().build();
  337    412   
        let mut http_response = ::aws_smithy_runtime_api::http::Response::try_from(
  338    413   
            ::http_1x::response::Builder::new()
  339    414   
                .header("X-Amzn-Errortype", "ExtraError")
  340    415   
                .status(500)
  341    416   
                .body(::aws_smithy_types::body::SdkBody::from(""))
  342    417   
                .unwrap(),
  343    418   
        )
  344    419   
        .unwrap();
  345    420   
        use ::aws_smithy_runtime_api::client::runtime_plugin::RuntimePlugin;
  346    421   
        use ::aws_smithy_runtime_api::client::ser_de::DeserializeResponse;
  347    422   
  348    423   
        let op = crate::operation::escaped_string_values::EscapedStringValues::new();
  349    424   
        let config = op.config().expect("the operation has config");
  350    425   
        let de = config
  351    426   
            .load::<::aws_smithy_runtime_api::client::ser_de::SharedResponseDeserializer>()
  352    427   
            .expect("the config must have a deserializer");
  353    428   
  354    429   
        let parsed = de.deserialize_streaming(&mut http_response);
  355    430   
        let parsed = parsed.unwrap_or_else(|| {
  356    431   
            let http_response = http_response.map(|body| {
  357    432   
                ::aws_smithy_types::body::SdkBody::from(::bytes::Bytes::copy_from_slice(&::aws_smithy_protocol_test::decode_body_data(
  358    433   
                    body.bytes().unwrap(),
  359    434   
                    ::aws_smithy_protocol_test::MediaType::from("unknown"),
  360    435   
                )))
  361    436   
            });
  362         -
            de.deserialize_nonstreaming(&http_response)
         437  +
            // Build a config bag with the protocol for schema-based deserialization
         438  +
            #[allow(unused_mut)]
         439  +
            let mut test_cfg = ::aws_smithy_types::config_bag::ConfigBag::base();
         440  +
            {
         441  +
                let mut layer = ::aws_smithy_types::config_bag::Layer::new("test_protocol");
         442  +
                layer.store_put(::aws_smithy_schema::protocol::SharedClientProtocol::new(
         443  +
                    ::aws_smithy_json::protocol::aws_rest_json_1::AwsRestJsonProtocol::new(),
         444  +
                ));
         445  +
                test_cfg.push_shared_layer(layer.freeze());
         446  +
            }
         447  +
            de.deserialize_nonstreaming_with_config(&http_response, &test_cfg)
  363    448   
        });
  364    449   
        let parsed = parsed.expect_err("should be error response");
  365    450   
        let parsed: &crate::operation::escaped_string_values::EscapedStringValuesError =
  366    451   
            parsed.as_operation_error().expect("operation error").downcast_ref().unwrap();
  367    452   
        if let crate::operation::escaped_string_values::EscapedStringValuesError::ExtraError(parsed) = parsed {
  368    453   
            ::pretty_assertions::assert_eq!(parsed.message, expected_output.message, "Unexpected value for `message`");
  369    454   
        } else {
  370    455   
            panic!("wrong variant: Got: {:?}. Expected: {:?}", parsed, expected_output);
  371    456   
        }
  372    457   
    }

tmp-codegen-diff/codegen-client-test/rest_json_extras/rust-client-codegen/src/operation/escaped_string_values/_escaped_string_values_input.rs

@@ -3,3 +131,143 @@
   23     23   
    "aws.protocoltests.restjson.synthetic",
   24     24   
    "EscapedStringValuesInput",
   25     25   
);
   26     26   
static ESCAPEDSTRINGVALUESINPUT_MEMBER_ENUM: ::aws_smithy_schema::Schema = ::aws_smithy_schema::Schema::new_member(
   27     27   
    ::aws_smithy_schema::ShapeId::from_static(
   28     28   
        "aws.protocoltests.restjson.synthetic#EscapedStringValuesInput$enum",
   29     29   
        "aws.protocoltests.restjson.synthetic",
   30     30   
        "EscapedStringValuesInput",
   31     31   
    ),
   32     32   
    ::aws_smithy_schema::ShapeType::String,
   33         -
    "r##enum",
          33  +
    "enum",
   34     34   
    0,
   35     35   
);
   36     36   
static ESCAPEDSTRINGVALUESINPUT_MEMBER_SOME_STRING: ::aws_smithy_schema::Schema = ::aws_smithy_schema::Schema::new_member(
   37     37   
    ::aws_smithy_schema::ShapeId::from_static(
   38     38   
        "aws.protocoltests.restjson.synthetic#EscapedStringValuesInput$someString",
   39     39   
        "aws.protocoltests.restjson.synthetic",
   40     40   
        "EscapedStringValuesInput",
   41     41   
    ),
   42     42   
    ::aws_smithy_schema::ShapeType::String,
   43         -
    "some_string",
          43  +
    "someString",
   44     44   
    1,
   45     45   
)
   46     46   
.with_json_name("also\"has\"quotes");
   47     47   
static ESCAPEDSTRINGVALUESINPUT_SCHEMA: ::aws_smithy_schema::Schema = ::aws_smithy_schema::Schema::new_struct(
   48     48   
    ESCAPEDSTRINGVALUESINPUT_SCHEMA_ID,
   49     49   
    ::aws_smithy_schema::ShapeType::Structure,
   50     50   
    &[&ESCAPEDSTRINGVALUESINPUT_MEMBER_ENUM, &ESCAPEDSTRINGVALUESINPUT_MEMBER_SOME_STRING],
   51         -
);
          51  +
)
          52  +
.with_http(aws_smithy_schema::traits::HttpTrait::new("POST", "/escaped-string-values", None));
   52     53   
impl EscapedStringValuesInput {
   53     54   
    /// The schema for this shape.
   54     55   
    pub const SCHEMA: &'static ::aws_smithy_schema::Schema = &ESCAPEDSTRINGVALUESINPUT_SCHEMA;
   55     56   
}
   56     57   
impl ::aws_smithy_schema::serde::SerializableStruct for EscapedStringValuesInput {
   57     58   
    #[allow(unused_variables, clippy::diverging_sub_expression)]
   58     59   
    fn serialize_members(
   59     60   
        &self,
   60     61   
        ser: &mut dyn ::aws_smithy_schema::serde::ShapeSerializer,
   61     62   
    ) -> ::std::result::Result<(), ::aws_smithy_schema::serde::SerdeError> {
   62     63   
        if let Some(ref val) = self.r#enum {
   63     64   
            ser.write_string(&ESCAPEDSTRINGVALUESINPUT_MEMBER_ENUM, val.as_str())?;
   64     65   
        }
   65     66   
        if let Some(ref val) = self.some_string {
   66     67   
            ser.write_string(&ESCAPEDSTRINGVALUESINPUT_MEMBER_SOME_STRING, val)?;
   67     68   
        }
   68     69   
        Ok(())
   69     70   
    }
   70     71   
}
   71     72   
impl EscapedStringValuesInput {
   72     73   
    /// Deserializes this structure from a [`ShapeDeserializer`].
   73         -
    pub fn deserialize<D: ::aws_smithy_schema::serde::ShapeDeserializer>(
   74         -
        deserializer: &mut D,
          74  +
    pub fn deserialize(
          75  +
        deserializer: &mut dyn ::aws_smithy_schema::serde::ShapeDeserializer,
   75     76   
    ) -> ::std::result::Result<Self, ::aws_smithy_schema::serde::SerdeError> {
   76     77   
        #[allow(unused_variables, unused_mut)]
   77     78   
        let mut builder = Self::builder();
   78     79   
        #[allow(
   79     80   
            unused_variables,
   80     81   
            unreachable_code,
   81     82   
            clippy::single_match,
   82     83   
            clippy::match_single_binding,
   83     84   
            clippy::diverging_sub_expression
   84     85   
        )]
   85         -
        deserializer.read_struct(&ESCAPEDSTRINGVALUESINPUT_SCHEMA, (), |_, member, deser| {
          86  +
        deserializer.read_struct(&ESCAPEDSTRINGVALUESINPUT_SCHEMA, &mut |member, deser| {
   86     87   
            match member.member_index() {
   87     88   
                Some(0) => {
   88     89   
                    builder.r#enum = Some(crate::types::EnumWithEscapedChars::from(deser.read_string(member)?.as_str()));
   89     90   
                }
   90     91   
                Some(1) => {
   91     92   
                    builder.some_string = Some(deser.read_string(member)?);
   92     93   
                }
   93     94   
                _ => {}
   94     95   
            }
   95     96   
            Ok(())
   96     97   
        })?;
   97     98   
        builder
   98     99   
            .build()
   99    100   
            .map_err(|e| aws_smithy_schema::serde::SerdeError::Custom { message: e.to_string() })
  100    101   
    }
  101    102   
}
         103  +
impl EscapedStringValuesInput {
         104  +
    /// Deserializes this structure from a body deserializer and HTTP response.
         105  +
    pub fn deserialize_with_response(
         106  +
        deserializer: &mut dyn ::aws_smithy_schema::serde::ShapeDeserializer,
         107  +
        _headers: &::aws_smithy_runtime_api::http::Headers,
         108  +
        _status: u16,
         109  +
        _body: &[u8],
         110  +
    ) -> ::std::result::Result<Self, ::aws_smithy_schema::serde::SerdeError> {
         111  +
        Self::deserialize(deserializer)
         112  +
    }
         113  +
}
  102    114   
impl EscapedStringValuesInput {
  103    115   
    /// Creates a new builder-style object to manufacture [`EscapedStringValuesInput`](crate::operation::escaped_string_values::EscapedStringValuesInput).
  104    116   
    pub fn builder() -> crate::operation::escaped_string_values::builders::EscapedStringValuesInputBuilder {
  105    117   
        crate::operation::escaped_string_values::builders::EscapedStringValuesInputBuilder::default()
  106    118   
    }
  107    119   
}
  108    120   
  109    121   
/// A builder for [`EscapedStringValuesInput`](crate::operation::escaped_string_values::EscapedStringValuesInput).
  110    122   
#[derive(::std::clone::Clone, ::std::cmp::PartialEq, ::std::default::Default, ::std::fmt::Debug)]
  111    123   
#[non_exhaustive]

tmp-codegen-diff/codegen-client-test/rest_json_extras/rust-client-codegen/src/operation/escaped_string_values/_escaped_string_values_output.rs

@@ -3,3 +129,140 @@
   23     23   
    "aws.protocoltests.restjson.synthetic",
   24     24   
    "EscapedStringValuesOutput",
   25     25   
);
   26     26   
static ESCAPEDSTRINGVALUESOUTPUT_MEMBER_ENUM: ::aws_smithy_schema::Schema = ::aws_smithy_schema::Schema::new_member(
   27     27   
    ::aws_smithy_schema::ShapeId::from_static(
   28     28   
        "aws.protocoltests.restjson.synthetic#EscapedStringValuesOutput$enum",
   29     29   
        "aws.protocoltests.restjson.synthetic",
   30     30   
        "EscapedStringValuesOutput",
   31     31   
    ),
   32     32   
    ::aws_smithy_schema::ShapeType::String,
   33         -
    "r##enum",
          33  +
    "enum",
   34     34   
    0,
   35     35   
);
   36     36   
static ESCAPEDSTRINGVALUESOUTPUT_MEMBER_SOME_STRING: ::aws_smithy_schema::Schema = ::aws_smithy_schema::Schema::new_member(
   37     37   
    ::aws_smithy_schema::ShapeId::from_static(
   38     38   
        "aws.protocoltests.restjson.synthetic#EscapedStringValuesOutput$someString",
   39     39   
        "aws.protocoltests.restjson.synthetic",
   40     40   
        "EscapedStringValuesOutput",
   41     41   
    ),
   42     42   
    ::aws_smithy_schema::ShapeType::String,
   43         -
    "some_string",
          43  +
    "someString",
   44     44   
    1,
   45     45   
)
   46     46   
.with_json_name("also\"has\"quotes");
   47     47   
static ESCAPEDSTRINGVALUESOUTPUT_SCHEMA: ::aws_smithy_schema::Schema = ::aws_smithy_schema::Schema::new_struct(
   48     48   
    ESCAPEDSTRINGVALUESOUTPUT_SCHEMA_ID,
   49     49   
    ::aws_smithy_schema::ShapeType::Structure,
   50     50   
    &[&ESCAPEDSTRINGVALUESOUTPUT_MEMBER_ENUM, &ESCAPEDSTRINGVALUESOUTPUT_MEMBER_SOME_STRING],
   51     51   
);
   52     52   
impl EscapedStringValuesOutput {
   53     53   
    /// The schema for this shape.
   54     54   
    pub const SCHEMA: &'static ::aws_smithy_schema::Schema = &ESCAPEDSTRINGVALUESOUTPUT_SCHEMA;
   55     55   
}
   56     56   
impl ::aws_smithy_schema::serde::SerializableStruct for EscapedStringValuesOutput {
   57     57   
    #[allow(unused_variables, clippy::diverging_sub_expression)]
   58     58   
    fn serialize_members(
   59     59   
        &self,
   60     60   
        ser: &mut dyn ::aws_smithy_schema::serde::ShapeSerializer,
   61     61   
    ) -> ::std::result::Result<(), ::aws_smithy_schema::serde::SerdeError> {
   62     62   
        if let Some(ref val) = self.r#enum {
   63     63   
            ser.write_string(&ESCAPEDSTRINGVALUESOUTPUT_MEMBER_ENUM, val.as_str())?;
   64     64   
        }
   65     65   
        if let Some(ref val) = self.some_string {
   66     66   
            ser.write_string(&ESCAPEDSTRINGVALUESOUTPUT_MEMBER_SOME_STRING, val)?;
   67     67   
        }
   68     68   
        Ok(())
   69     69   
    }
   70     70   
}
   71     71   
impl EscapedStringValuesOutput {
   72     72   
    /// Deserializes this structure from a [`ShapeDeserializer`].
   73         -
    pub fn deserialize<D: ::aws_smithy_schema::serde::ShapeDeserializer>(
   74         -
        deserializer: &mut D,
          73  +
    pub fn deserialize(
          74  +
        deserializer: &mut dyn ::aws_smithy_schema::serde::ShapeDeserializer,
   75     75   
    ) -> ::std::result::Result<Self, ::aws_smithy_schema::serde::SerdeError> {
   76     76   
        #[allow(unused_variables, unused_mut)]
   77     77   
        let mut builder = Self::builder();
   78     78   
        #[allow(
   79     79   
            unused_variables,
   80     80   
            unreachable_code,
   81     81   
            clippy::single_match,
   82     82   
            clippy::match_single_binding,
   83     83   
            clippy::diverging_sub_expression
   84     84   
        )]
   85         -
        deserializer.read_struct(&ESCAPEDSTRINGVALUESOUTPUT_SCHEMA, (), |_, member, deser| {
          85  +
        deserializer.read_struct(&ESCAPEDSTRINGVALUESOUTPUT_SCHEMA, &mut |member, deser| {
   86     86   
            match member.member_index() {
   87     87   
                Some(0) => {
   88     88   
                    builder.r#enum = Some(crate::types::EnumWithEscapedChars::from(deser.read_string(member)?.as_str()));
   89     89   
                }
   90     90   
                Some(1) => {
   91     91   
                    builder.some_string = Some(deser.read_string(member)?);
   92     92   
                }
   93     93   
                _ => {}
   94     94   
            }
   95     95   
            Ok(())
   96     96   
        })?;
   97     97   
        Ok(builder.build())
   98     98   
    }
   99     99   
}
         100  +
impl EscapedStringValuesOutput {
         101  +
    /// Deserializes this structure from a body deserializer and HTTP response.
         102  +
    pub fn deserialize_with_response(
         103  +
        deserializer: &mut dyn ::aws_smithy_schema::serde::ShapeDeserializer,
         104  +
        _headers: &::aws_smithy_runtime_api::http::Headers,
         105  +
        _status: u16,
         106  +
        _body: &[u8],
         107  +
    ) -> ::std::result::Result<Self, ::aws_smithy_schema::serde::SerdeError> {
         108  +
        Self::deserialize(deserializer)
         109  +
    }
         110  +
}
  100    111   
impl EscapedStringValuesOutput {
  101    112   
    /// Creates a new builder-style object to manufacture [`EscapedStringValuesOutput`](crate::operation::escaped_string_values::EscapedStringValuesOutput).
  102    113   
    pub fn builder() -> crate::operation::escaped_string_values::builders::EscapedStringValuesOutputBuilder {
  103    114   
        crate::operation::escaped_string_values::builders::EscapedStringValuesOutputBuilder::default()
  104    115   
    }
  105    116   
}
  106    117   
  107    118   
/// A builder for [`EscapedStringValuesOutput`](crate::operation::escaped_string_values::EscapedStringValuesOutput).
  108    119   
#[derive(::std::clone::Clone, ::std::cmp::PartialEq, ::std::default::Default, ::std::fmt::Debug)]
  109    120   
#[non_exhaustive]

tmp-codegen-diff/codegen-client-test/rest_json_extras/rust-client-codegen/src/operation/map_with_enum_key_op.rs

@@ -1,1 +40,44 @@
    1      1   
// Code generated by software.amazon.smithy.rust.codegen.smithy-rs. DO NOT EDIT.
    2      2   
/// Orchestration and serialization glue logic for `MapWithEnumKeyOp`.
    3      3   
#[derive(::std::clone::Clone, ::std::default::Default, ::std::fmt::Debug)]
    4      4   
#[non_exhaustive]
    5      5   
pub struct MapWithEnumKeyOp;
    6      6   
impl MapWithEnumKeyOp {
    7      7   
    /// Creates a new `MapWithEnumKeyOp`
    8      8   
    pub fn new() -> Self {
    9      9   
        Self
   10     10   
    }
          11  +
    /// The schema for this operation's input shape.
          12  +
    pub const INPUT_SCHEMA: &'static ::aws_smithy_schema::Schema = crate::operation::map_with_enum_key_op::MapWithEnumKeyOpInput::SCHEMA;
          13  +
    /// The schema for this operation's output shape.
          14  +
    pub const OUTPUT_SCHEMA: &'static ::aws_smithy_schema::Schema = crate::operation::map_with_enum_key_op::MapWithEnumKeyOpOutput::SCHEMA;
   11     15   
    pub(crate) async fn orchestrate(
   12     16   
        runtime_plugins: &::aws_smithy_runtime_api::client::runtime_plugin::RuntimePlugins,
   13     17   
        input: crate::operation::map_with_enum_key_op::MapWithEnumKeyOpInput,
   14     18   
    ) -> ::std::result::Result<
   15     19   
        crate::operation::map_with_enum_key_op::MapWithEnumKeyOpOutput,
   16     20   
        ::aws_smithy_runtime_api::client::result::SdkError<
   17     21   
            crate::operation::map_with_enum_key_op::MapWithEnumKeyOpError,
   18     22   
            ::aws_smithy_runtime_api::client::orchestrator::HttpResponse,
   19     23   
        >,
   20     24   
    > {
@@ -110,114 +233,299 @@
  130    134   
                crate::operation::map_with_enum_key_op::MapWithEnumKeyOpError,
  131    135   
            >::new());
  132    136   
  133    137   
        ::std::borrow::Cow::Owned(rcb)
  134    138   
    }
  135    139   
}
  136    140   
  137    141   
#[derive(Debug)]
  138    142   
struct MapWithEnumKeyOpResponseDeserializer;
  139    143   
impl ::aws_smithy_runtime_api::client::ser_de::DeserializeResponse for MapWithEnumKeyOpResponseDeserializer {
  140         -
    fn deserialize_nonstreaming(
         144  +
    fn deserialize_nonstreaming_with_config(
  141    145   
        &self,
  142    146   
        response: &::aws_smithy_runtime_api::client::orchestrator::HttpResponse,
         147  +
        _cfg: &::aws_smithy_types::config_bag::ConfigBag,
  143    148   
    ) -> ::aws_smithy_runtime_api::client::interceptors::context::OutputOrError {
  144    149   
        let (success, status) = (response.status().is_success(), response.status().as_u16());
  145         -
        let headers = response.headers();
  146         -
        let body = response.body().bytes().expect("body loaded");
  147    150   
        #[allow(unused_mut)]
  148    151   
        let mut force_error = false;
  149    152   
  150         -
        let parse_result = if !success && status != 200 || force_error {
  151         -
            crate::protocol_serde::shape_map_with_enum_key_op::de_map_with_enum_key_op_http_error(status, headers, body)
         153  +
        if !success && status != 200 || force_error {
         154  +
            let headers = response.headers();
         155  +
            let body = response.body().bytes().expect("body loaded");
         156  +
            #[allow(unused_mut)]
         157  +
            let mut generic_builder = crate::protocol_serde::parse_http_error_metadata(status, headers, body).map_err(|e| {
         158  +
                ::aws_smithy_runtime_api::client::orchestrator::OrchestratorError::other(::aws_smithy_runtime_api::box_error::BoxError::from(e))
         159  +
            })?;
         160  +
         161  +
            let generic = generic_builder.build();
         162  +
            let error_code = match generic.code() {
         163  +
                ::std::option::Option::Some(code) => code,
         164  +
                ::std::option::Option::None => {
         165  +
                    return ::std::result::Result::Err(::aws_smithy_runtime_api::client::orchestrator::OrchestratorError::other(
         166  +
                        ::aws_smithy_runtime_api::box_error::BoxError::from(
         167  +
                            crate::operation::map_with_enum_key_op::MapWithEnumKeyOpError::unhandled(generic),
         168  +
                        ),
         169  +
                    ))
         170  +
                }
         171  +
            };
         172  +
            let _error_message = generic.message().map(|msg| msg.to_owned());
         173  +
            let protocol = _cfg
         174  +
                .load::<::aws_smithy_schema::protocol::SharedClientProtocol>()
         175  +
                .expect("a SharedClientProtocol is required");
         176  +
            let err = match error_code {
         177  +
                "ValidationException" => crate::operation::map_with_enum_key_op::MapWithEnumKeyOpError::ValidationError({
         178  +
                    let mut tmp = match protocol
         179  +
                        .deserialize_response(response, crate::types::error::ValidationError::SCHEMA, _cfg)
         180  +
                        .and_then(|mut deser| {
         181  +
                            crate::types::error::ValidationError::deserialize_with_response(
         182  +
                                &mut *deser,
         183  +
                                response.headers(),
         184  +
                                response.status().into(),
         185  +
                                body,
         186  +
                            )
         187  +
                        }) {
         188  +
                        ::std::result::Result::Ok(val) => val,
         189  +
                        ::std::result::Result::Err(e) => {
         190  +
                            return ::std::result::Result::Err(::aws_smithy_runtime_api::client::orchestrator::OrchestratorError::other(
         191  +
                                ::aws_smithy_runtime_api::box_error::BoxError::from(e),
         192  +
                            ))
         193  +
                        }
         194  +
                    };
         195  +
                    tmp.meta = generic;
         196  +
                    tmp
         197  +
                }),
         198  +
                "ExtraError" => crate::operation::map_with_enum_key_op::MapWithEnumKeyOpError::ExtraError({
         199  +
                    let mut tmp = match protocol
         200  +
                        .deserialize_response(response, crate::types::error::ExtraError::SCHEMA, _cfg)
         201  +
                        .and_then(|mut deser| {
         202  +
                            crate::types::error::ExtraError::deserialize_with_response(
         203  +
                                &mut *deser,
         204  +
                                response.headers(),
         205  +
                                response.status().into(),
         206  +
                                body,
         207  +
                            )
         208  +
                        }) {
         209  +
                        ::std::result::Result::Ok(val) => val,
         210  +
                        ::std::result::Result::Err(e) => {
         211  +
                            return ::std::result::Result::Err(::aws_smithy_runtime_api::client::orchestrator::OrchestratorError::other(
         212  +
                                ::aws_smithy_runtime_api::box_error::BoxError::from(e),
         213  +
                            ))
         214  +
                        }
         215  +
                    };
         216  +
                    tmp.meta = generic;
         217  +
                    if tmp.message.is_none() {
         218  +
                        tmp.message = _error_message;
         219  +
                    }
         220  +
                    tmp
         221  +
                }),
         222  +
                _ => crate::operation::map_with_enum_key_op::MapWithEnumKeyOpError::generic(generic),
         223  +
            };
         224  +
            ::std::result::Result::Err(::aws_smithy_runtime_api::client::orchestrator::OrchestratorError::operation(
         225  +
                ::aws_smithy_runtime_api::client::interceptors::context::Error::erase(err),
         226  +
            ))
  152    227   
        } else {
  153         -
            crate::protocol_serde::shape_map_with_enum_key_op::de_map_with_enum_key_op_http_response(status, headers, body)
  154         -
        };
  155         -
        crate::protocol_serde::type_erase_result(parse_result)
         228  +
            let protocol = _cfg
         229  +
                .load::<::aws_smithy_schema::protocol::SharedClientProtocol>()
         230  +
                .expect("a SharedClientProtocol is required");
         231  +
            let mut deser = protocol
         232  +
                .deserialize_response(response, MapWithEnumKeyOp::OUTPUT_SCHEMA, _cfg)
         233  +
                .map_err(|e| {
         234  +
                    ::aws_smithy_runtime_api::client::orchestrator::OrchestratorError::other(::aws_smithy_runtime_api::box_error::BoxError::from(e))
         235  +
                })?;
         236  +
            let body = response.body().bytes().expect("body loaded");
         237  +
            let output = crate::operation::map_with_enum_key_op::MapWithEnumKeyOpOutput::deserialize_with_response(
         238  +
                &mut *deser,
         239  +
                response.headers(),
         240  +
                response.status().into(),
         241  +
                body,
         242  +
            )
         243  +
            .map_err(|e| {
         244  +
                ::aws_smithy_runtime_api::client::orchestrator::OrchestratorError::other(::aws_smithy_runtime_api::box_error::BoxError::from(e))
         245  +
            })?;
         246  +
            ::std::result::Result::Ok(::aws_smithy_runtime_api::client::interceptors::context::Output::erase(output))
         247  +
        }
  156    248   
    }
  157    249   
}
  158    250   
#[derive(Debug)]
  159    251   
struct MapWithEnumKeyOpRequestSerializer;
  160    252   
impl ::aws_smithy_runtime_api::client::ser_de::SerializeRequest for MapWithEnumKeyOpRequestSerializer {
  161    253   
    #[allow(unused_mut, clippy::let_and_return, clippy::needless_borrow, clippy::useless_conversion)]
  162    254   
    fn serialize_input(
  163    255   
        &self,
  164    256   
        input: ::aws_smithy_runtime_api::client::interceptors::context::Input,
  165    257   
        _cfg: &mut ::aws_smithy_types::config_bag::ConfigBag,
  166    258   
    ) -> ::std::result::Result<::aws_smithy_runtime_api::client::orchestrator::HttpRequest, ::aws_smithy_runtime_api::box_error::BoxError> {
  167    259   
        let input = input
  168    260   
            .downcast::<crate::operation::map_with_enum_key_op::MapWithEnumKeyOpInput>()
  169    261   
            .expect("correct type");
  170         -
        let _header_serialization_settings = _cfg
  171         -
            .load::<crate::serialization_settings::HeaderSerializationSettings>()
  172         -
            .cloned()
  173         -
            .unwrap_or_default();
  174         -
        let mut request_builder = {
  175         -
            #[allow(clippy::uninlined_format_args)]
  176         -
            fn uri_base(
  177         -
                _input: &crate::operation::map_with_enum_key_op::MapWithEnumKeyOpInput,
  178         -
                output: &mut ::std::string::String,
  179         -
            ) -> ::std::result::Result<(), ::aws_smithy_types::error::operation::BuildError> {
  180         -
                use ::std::fmt::Write as _;
  181         -
                ::std::write!(output, "/map-with-enum-key").expect("formatting should succeed");
  182         -
                ::std::result::Result::Ok(())
  183         -
            }
  184         -
            #[allow(clippy::unnecessary_wraps)]
  185         -
            fn update_http_builder(
  186         -
                input: &crate::operation::map_with_enum_key_op::MapWithEnumKeyOpInput,
  187         -
                builder: ::http_1x::request::Builder,
  188         -
            ) -> ::std::result::Result<::http_1x::request::Builder, ::aws_smithy_types::error::operation::BuildError> {
  189         -
                let mut uri = ::std::string::String::new();
  190         -
                uri_base(input, &mut uri)?;
  191         -
                ::std::result::Result::Ok(builder.method("POST").uri(uri))
  192         -
            }
  193         -
            let mut builder = update_http_builder(&input, ::http_1x::request::Builder::new())?;
  194         -
            builder = _header_serialization_settings.set_default_header(builder, ::http_1x::header::CONTENT_TYPE, "application/json");
  195         -
            builder
  196         -
        };
  197         -
        let body =
  198         -
            ::aws_smithy_types::body::SdkBody::from(crate::protocol_serde::shape_map_with_enum_key_op::ser_map_with_enum_key_op_input(&input)?);
  199         -
        if let Some(content_length) = body.content_length() {
  200         -
            let content_length = content_length.to_string();
  201         -
            request_builder = _header_serialization_settings.set_default_header(request_builder, ::http_1x::header::CONTENT_LENGTH, &content_length);
  202         -
        }
  203         -
        ::std::result::Result::Ok(request_builder.body(body).expect("valid request").try_into().unwrap())
         262  +
        let protocol = _cfg
         263  +
            .load::<::aws_smithy_schema::protocol::SharedClientProtocol>()
         264  +
            .expect("a SharedClientProtocol is required");
         265  +
        let mut request = protocol
         266  +
            .serialize_request(&input, MapWithEnumKeyOp::INPUT_SCHEMA, "", _cfg)
         267  +
            .map_err(::aws_smithy_runtime_api::box_error::BoxError::from)?;
         268  +
         269  +
        return ::std::result::Result::Ok(request);
  204    270   
    }
  205    271   
}
  206    272   
#[derive(Debug)]
  207    273   
struct MapWithEnumKeyOpEndpointParamsInterceptor;
  208    274   
  209    275   
impl ::aws_smithy_runtime_api::client::interceptors::Intercept for MapWithEnumKeyOpEndpointParamsInterceptor {
  210    276   
    fn name(&self) -> &'static str {
  211    277   
        "MapWithEnumKeyOpEndpointParamsInterceptor"
  212    278   
    }
  213    279   
@@ -292,358 +392,478 @@
  312    378   
            .expect("the config must have a deserializer");
  313    379   
  314    380   
        let parsed = de.deserialize_streaming(&mut http_response);
  315    381   
        let parsed = parsed.unwrap_or_else(|| {
  316    382   
            let http_response = http_response.map(|body| {
  317    383   
                ::aws_smithy_types::body::SdkBody::from(::bytes::Bytes::copy_from_slice(&::aws_smithy_protocol_test::decode_body_data(
  318    384   
                    body.bytes().unwrap(),
  319    385   
                    ::aws_smithy_protocol_test::MediaType::from("unknown"),
  320    386   
                )))
  321    387   
            });
  322         -
            de.deserialize_nonstreaming(&http_response)
         388  +
            // Build a config bag with the protocol for schema-based deserialization
         389  +
            #[allow(unused_mut)]
         390  +
            let mut test_cfg = ::aws_smithy_types::config_bag::ConfigBag::base();
         391  +
            {
         392  +
                let mut layer = ::aws_smithy_types::config_bag::Layer::new("test_protocol");
         393  +
                layer.store_put(::aws_smithy_schema::protocol::SharedClientProtocol::new(
         394  +
                    ::aws_smithy_json::protocol::aws_rest_json_1::AwsRestJsonProtocol::new(),
         395  +
                ));
         396  +
                test_cfg.push_shared_layer(layer.freeze());
         397  +
            }
         398  +
            de.deserialize_nonstreaming_with_config(&http_response, &test_cfg)
  323    399   
        });
  324    400   
        let parsed = parsed
  325    401   
            .expect("should be successful response")
  326    402   
            .downcast::<crate::operation::map_with_enum_key_op::MapWithEnumKeyOpOutput>()
  327    403   
            .unwrap();
  328    404   
        ::pretty_assertions::assert_eq!(parsed.map, expected_output.map, "Unexpected value for `map`");
  329    405   
    }
  330    406   
  331    407   
    /// Upper case error modeled lower case
  332    408   
    /// Test ID: ServiceLevelErrorClient
  333    409   
    #[::tokio::test]
  334    410   
    #[::tracing_test::traced_test]
  335    411   
    async fn service_level_error_client_response() {
  336    412   
        let expected_output = crate::types::error::ExtraError::builder().build();
  337    413   
        let mut http_response = ::aws_smithy_runtime_api::http::Response::try_from(
  338    414   
            ::http_1x::response::Builder::new()
  339    415   
                .header("X-Amzn-Errortype", "ExtraError")
  340    416   
                .status(500)
  341    417   
                .body(::aws_smithy_types::body::SdkBody::from(""))
  342    418   
                .unwrap(),
  343    419   
        )
  344    420   
        .unwrap();
  345    421   
        use ::aws_smithy_runtime_api::client::runtime_plugin::RuntimePlugin;
  346    422   
        use ::aws_smithy_runtime_api::client::ser_de::DeserializeResponse;
  347    423   
  348    424   
        let op = crate::operation::map_with_enum_key_op::MapWithEnumKeyOp::new();
  349    425   
        let config = op.config().expect("the operation has config");
  350    426   
        let de = config
  351    427   
            .load::<::aws_smithy_runtime_api::client::ser_de::SharedResponseDeserializer>()
  352    428   
            .expect("the config must have a deserializer");
  353    429   
  354    430   
        let parsed = de.deserialize_streaming(&mut http_response);
  355    431   
        let parsed = parsed.unwrap_or_else(|| {
  356    432   
            let http_response = http_response.map(|body| {
  357    433   
                ::aws_smithy_types::body::SdkBody::from(::bytes::Bytes::copy_from_slice(&::aws_smithy_protocol_test::decode_body_data(
  358    434   
                    body.bytes().unwrap(),
  359    435   
                    ::aws_smithy_protocol_test::MediaType::from("unknown"),
  360    436   
                )))
  361    437   
            });
  362         -
            de.deserialize_nonstreaming(&http_response)
         438  +
            // Build a config bag with the protocol for schema-based deserialization
         439  +
            #[allow(unused_mut)]
         440  +
            let mut test_cfg = ::aws_smithy_types::config_bag::ConfigBag::base();
         441  +
            {
         442  +
                let mut layer = ::aws_smithy_types::config_bag::Layer::new("test_protocol");
         443  +
                layer.store_put(::aws_smithy_schema::protocol::SharedClientProtocol::new(
         444  +
                    ::aws_smithy_json::protocol::aws_rest_json_1::AwsRestJsonProtocol::new(),
         445  +
                ));
         446  +
                test_cfg.push_shared_layer(layer.freeze());
         447  +
            }
         448  +
            de.deserialize_nonstreaming_with_config(&http_response, &test_cfg)
  363    449   
        });
  364    450   
        let parsed = parsed.expect_err("should be error response");
  365    451   
        let parsed: &crate::operation::map_with_enum_key_op::MapWithEnumKeyOpError =
  366    452   
            parsed.as_operation_error().expect("operation error").downcast_ref().unwrap();
  367    453   
        if let crate::operation::map_with_enum_key_op::MapWithEnumKeyOpError::ExtraError(parsed) = parsed {
  368    454   
            ::pretty_assertions::assert_eq!(parsed.message, expected_output.message, "Unexpected value for `message`");
  369    455   
        } else {
  370    456   
            panic!("wrong variant: Got: {:?}. Expected: {:?}", parsed, expected_output);
  371    457   
        }
  372    458   
    }

tmp-codegen-diff/codegen-client-test/rest_json_extras/rust-client-codegen/src/operation/map_with_enum_key_op/_map_with_enum_key_op_input.rs

@@ -4,4 +127,136 @@
   24     24   
        "MapWithEnumKeyOpInput",
   25     25   
    ),
   26     26   
    ::aws_smithy_schema::ShapeType::Map,
   27     27   
    "map",
   28     28   
    0,
   29     29   
);
   30     30   
static MAPWITHENUMKEYOPINPUT_SCHEMA: ::aws_smithy_schema::Schema = ::aws_smithy_schema::Schema::new_struct(
   31     31   
    MAPWITHENUMKEYOPINPUT_SCHEMA_ID,
   32     32   
    ::aws_smithy_schema::ShapeType::Structure,
   33     33   
    &[&MAPWITHENUMKEYOPINPUT_MEMBER_MAP],
   34         -
);
          34  +
)
          35  +
.with_http(aws_smithy_schema::traits::HttpTrait::new("POST", "/map-with-enum-key", None));
   35     36   
impl MapWithEnumKeyOpInput {
   36     37   
    /// The schema for this shape.
   37     38   
    pub const SCHEMA: &'static ::aws_smithy_schema::Schema = &MAPWITHENUMKEYOPINPUT_SCHEMA;
   38     39   
}
   39     40   
impl ::aws_smithy_schema::serde::SerializableStruct for MapWithEnumKeyOpInput {
   40     41   
    #[allow(unused_variables, clippy::diverging_sub_expression)]
   41     42   
    fn serialize_members(
   42     43   
        &self,
   43     44   
        ser: &mut dyn ::aws_smithy_schema::serde::ShapeSerializer,
   44     45   
    ) -> ::std::result::Result<(), ::aws_smithy_schema::serde::SerdeError> {
   45     46   
        if let Some(ref val) = self.map {
   46     47   
            ser.write_map(
   47     48   
                &MAPWITHENUMKEYOPINPUT_MEMBER_MAP,
   48     49   
                &|ser: &mut dyn ::aws_smithy_schema::serde::ShapeSerializer| {
   49     50   
                    for (key, value) in val {
   50     51   
                        ser.write_string(&::aws_smithy_schema::prelude::STRING, key.as_str())?;
   51     52   
                        ser.write_string(&::aws_smithy_schema::prelude::STRING, value)?;
   52     53   
                    }
   53     54   
                    Ok(())
   54     55   
                },
   55     56   
            )?;
   56     57   
        }
   57     58   
        Ok(())
   58     59   
    }
   59     60   
}
   60     61   
impl MapWithEnumKeyOpInput {
   61     62   
    /// Deserializes this structure from a [`ShapeDeserializer`].
   62         -
    pub fn deserialize<D: ::aws_smithy_schema::serde::ShapeDeserializer>(
   63         -
        deserializer: &mut D,
          63  +
    pub fn deserialize(
          64  +
        deserializer: &mut dyn ::aws_smithy_schema::serde::ShapeDeserializer,
   64     65   
    ) -> ::std::result::Result<Self, ::aws_smithy_schema::serde::SerdeError> {
   65     66   
        #[allow(unused_variables, unused_mut)]
   66     67   
        let mut builder = Self::builder();
   67     68   
        #[allow(
   68     69   
            unused_variables,
   69     70   
            unreachable_code,
   70     71   
            clippy::single_match,
   71     72   
            clippy::match_single_binding,
   72     73   
            clippy::diverging_sub_expression
   73     74   
        )]
   74         -
        deserializer.read_struct(&MAPWITHENUMKEYOPINPUT_SCHEMA, (), |_, member, deser| {
          75  +
        deserializer.read_struct(&MAPWITHENUMKEYOPINPUT_SCHEMA, &mut |member, deser| {
   75     76   
            match member.member_index() {
   76     77   
                Some(0) => {
   77     78   
                    builder.map = Some({
   78         -
                        let container = if let Some(cap) = deser.container_size() {
   79         -
                            std::collections::HashMap::with_capacity(cap)
   80         -
                        } else {
   81         -
                            std::collections::HashMap::new()
   82         -
                        };
   83         -
                        deser.read_map(member, container, |mut map, key, deser| {
   84         -
                            map.insert(crate::types::StringEnum::from(key.as_str()), deser.read_string(member)?);
   85         -
                            Ok(map)
   86         -
                        })?
          79  +
                        let mut container = std::collections::HashMap::new();
          80  +
                        deser.read_map(member, &mut |key, deser| {
          81  +
                            container.insert(crate::types::StringEnum::from(key.as_str()), deser.read_string(member)?);
          82  +
                            Ok(())
          83  +
                        })?;
          84  +
                        container
   87     85   
                    });
   88     86   
                }
   89     87   
                _ => {}
   90     88   
            }
   91     89   
            Ok(())
   92     90   
        })?;
   93     91   
        builder
   94     92   
            .build()
   95     93   
            .map_err(|e| aws_smithy_schema::serde::SerdeError::Custom { message: e.to_string() })
   96     94   
    }
   97     95   
}
          96  +
impl MapWithEnumKeyOpInput {
          97  +
    /// Deserializes this structure from a body deserializer and HTTP response.
          98  +
    pub fn deserialize_with_response(
          99  +
        deserializer: &mut dyn ::aws_smithy_schema::serde::ShapeDeserializer,
         100  +
        _headers: &::aws_smithy_runtime_api::http::Headers,
         101  +
        _status: u16,
         102  +
        _body: &[u8],
         103  +
    ) -> ::std::result::Result<Self, ::aws_smithy_schema::serde::SerdeError> {
         104  +
        Self::deserialize(deserializer)
         105  +
    }
         106  +
}
   98    107   
impl MapWithEnumKeyOpInput {
   99    108   
    /// Creates a new builder-style object to manufacture [`MapWithEnumKeyOpInput`](crate::operation::map_with_enum_key_op::MapWithEnumKeyOpInput).
  100    109   
    pub fn builder() -> crate::operation::map_with_enum_key_op::builders::MapWithEnumKeyOpInputBuilder {
  101    110   
        crate::operation::map_with_enum_key_op::builders::MapWithEnumKeyOpInputBuilder::default()
  102    111   
    }
  103    112   
}
  104    113   
  105    114   
/// A builder for [`MapWithEnumKeyOpInput`](crate::operation::map_with_enum_key_op::MapWithEnumKeyOpInput).
  106    115   
#[derive(::std::clone::Clone, ::std::cmp::PartialEq, ::std::default::Default, ::std::fmt::Debug)]
  107    116   
#[non_exhaustive]

tmp-codegen-diff/codegen-client-test/rest_json_extras/rust-client-codegen/src/operation/map_with_enum_key_op/_map_with_enum_key_op_output.rs

@@ -32,32 +125,133 @@
   52     52   
                    }
   53     53   
                    Ok(())
   54     54   
                },
   55     55   
            )?;
   56     56   
        }
   57     57   
        Ok(())
   58     58   
    }
   59     59   
}
   60     60   
impl MapWithEnumKeyOpOutput {
   61     61   
    /// Deserializes this structure from a [`ShapeDeserializer`].
   62         -
    pub fn deserialize<D: ::aws_smithy_schema::serde::ShapeDeserializer>(
   63         -
        deserializer: &mut D,
          62  +
    pub fn deserialize(
          63  +
        deserializer: &mut dyn ::aws_smithy_schema::serde::ShapeDeserializer,
   64     64   
    ) -> ::std::result::Result<Self, ::aws_smithy_schema::serde::SerdeError> {
   65     65   
        #[allow(unused_variables, unused_mut)]
   66     66   
        let mut builder = Self::builder();
   67     67   
        #[allow(
   68     68   
            unused_variables,
   69     69   
            unreachable_code,
   70     70   
            clippy::single_match,
   71     71   
            clippy::match_single_binding,
   72     72   
            clippy::diverging_sub_expression
   73     73   
        )]
   74         -
        deserializer.read_struct(&MAPWITHENUMKEYOPOUTPUT_SCHEMA, (), |_, member, deser| {
          74  +
        deserializer.read_struct(&MAPWITHENUMKEYOPOUTPUT_SCHEMA, &mut |member, deser| {
   75     75   
            match member.member_index() {
   76     76   
                Some(0) => {
   77     77   
                    builder.map = Some({
   78         -
                        let container = if let Some(cap) = deser.container_size() {
   79         -
                            std::collections::HashMap::with_capacity(cap)
   80         -
                        } else {
   81         -
                            std::collections::HashMap::new()
   82         -
                        };
   83         -
                        deser.read_map(member, container, |mut map, key, deser| {
   84         -
                            map.insert(crate::types::StringEnum::from(key.as_str()), deser.read_string(member)?);
   85         -
                            Ok(map)
   86         -
                        })?
          78  +
                        let mut container = std::collections::HashMap::new();
          79  +
                        deser.read_map(member, &mut |key, deser| {
          80  +
                            container.insert(crate::types::StringEnum::from(key.as_str()), deser.read_string(member)?);
          81  +
                            Ok(())
          82  +
                        })?;
          83  +
                        container
   87     84   
                    });
   88     85   
                }
   89     86   
                _ => {}
   90     87   
            }
   91     88   
            Ok(())
   92     89   
        })?;
   93     90   
        Ok(builder.build())
   94     91   
    }
   95     92   
}
          93  +
impl MapWithEnumKeyOpOutput {
          94  +
    /// Deserializes this structure from a body deserializer and HTTP response.
          95  +
    pub fn deserialize_with_response(
          96  +
        deserializer: &mut dyn ::aws_smithy_schema::serde::ShapeDeserializer,
          97  +
        _headers: &::aws_smithy_runtime_api::http::Headers,
          98  +
        _status: u16,
          99  +
        _body: &[u8],
         100  +
    ) -> ::std::result::Result<Self, ::aws_smithy_schema::serde::SerdeError> {
         101  +
        Self::deserialize(deserializer)
         102  +
    }
         103  +
}
   96    104   
impl MapWithEnumKeyOpOutput {
   97    105   
    /// Creates a new builder-style object to manufacture [`MapWithEnumKeyOpOutput`](crate::operation::map_with_enum_key_op::MapWithEnumKeyOpOutput).
   98    106   
    pub fn builder() -> crate::operation::map_with_enum_key_op::builders::MapWithEnumKeyOpOutputBuilder {
   99    107   
        crate::operation::map_with_enum_key_op::builders::MapWithEnumKeyOpOutputBuilder::default()
  100    108   
    }
  101    109   
}
  102    110   
  103    111   
/// A builder for [`MapWithEnumKeyOpOutput`](crate::operation::map_with_enum_key_op::MapWithEnumKeyOpOutput).
  104    112   
#[derive(::std::clone::Clone, ::std::cmp::PartialEq, ::std::default::Default, ::std::fmt::Debug)]
  105    113   
#[non_exhaustive]

tmp-codegen-diff/codegen-client-test/rest_json_extras/rust-client-codegen/src/operation/null_in_non_sparse.rs

@@ -1,1 +40,44 @@
    1      1   
// Code generated by software.amazon.smithy.rust.codegen.smithy-rs. DO NOT EDIT.
    2      2   
/// Orchestration and serialization glue logic for `NullInNonSparse`.
    3      3   
#[derive(::std::clone::Clone, ::std::default::Default, ::std::fmt::Debug)]
    4      4   
#[non_exhaustive]
    5      5   
pub struct NullInNonSparse;
    6      6   
impl NullInNonSparse {
    7      7   
    /// Creates a new `NullInNonSparse`
    8      8   
    pub fn new() -> Self {
    9      9   
        Self
   10     10   
    }
          11  +
    /// The schema for this operation's input shape.
          12  +
    pub const INPUT_SCHEMA: &'static ::aws_smithy_schema::Schema = crate::operation::null_in_non_sparse::NullInNonSparseInput::SCHEMA;
          13  +
    /// The schema for this operation's output shape.
          14  +
    pub const OUTPUT_SCHEMA: &'static ::aws_smithy_schema::Schema = crate::operation::null_in_non_sparse::NullInNonSparseOutput::SCHEMA;
   11     15   
    pub(crate) async fn orchestrate(
   12     16   
        runtime_plugins: &::aws_smithy_runtime_api::client::runtime_plugin::RuntimePlugins,
   13     17   
        input: crate::operation::null_in_non_sparse::NullInNonSparseInput,
   14     18   
    ) -> ::std::result::Result<
   15     19   
        crate::operation::null_in_non_sparse::NullInNonSparseOutput,
   16     20   
        ::aws_smithy_runtime_api::client::result::SdkError<
   17     21   
            crate::operation::null_in_non_sparse::NullInNonSparseError,
   18     22   
            ::aws_smithy_runtime_api::client::orchestrator::HttpResponse,
   19     23   
        >,
   20     24   
    > {
@@ -110,114 +228,278 @@
  130    134   
                crate::operation::null_in_non_sparse::NullInNonSparseError,
  131    135   
            >::new());
  132    136   
  133    137   
        ::std::borrow::Cow::Owned(rcb)
  134    138   
    }
  135    139   
}
  136    140   
  137    141   
#[derive(Debug)]
  138    142   
struct NullInNonSparseResponseDeserializer;
  139    143   
impl ::aws_smithy_runtime_api::client::ser_de::DeserializeResponse for NullInNonSparseResponseDeserializer {
  140         -
    fn deserialize_nonstreaming(
         144  +
    fn deserialize_nonstreaming_with_config(
  141    145   
        &self,
  142    146   
        response: &::aws_smithy_runtime_api::client::orchestrator::HttpResponse,
         147  +
        _cfg: &::aws_smithy_types::config_bag::ConfigBag,
  143    148   
    ) -> ::aws_smithy_runtime_api::client::interceptors::context::OutputOrError {
  144    149   
        let (success, status) = (response.status().is_success(), response.status().as_u16());
  145         -
        let headers = response.headers();
  146         -
        let body = response.body().bytes().expect("body loaded");
  147    150   
        #[allow(unused_mut)]
  148    151   
        let mut force_error = false;
  149    152   
  150         -
        let parse_result = if !success && status != 200 || force_error {
  151         -
            crate::protocol_serde::shape_null_in_non_sparse::de_null_in_non_sparse_http_error(status, headers, body)
         153  +
        if !success && status != 200 || force_error {
         154  +
            let headers = response.headers();
         155  +
            let body = response.body().bytes().expect("body loaded");
         156  +
            #[allow(unused_mut)]
         157  +
            let mut generic_builder = crate::protocol_serde::parse_http_error_metadata(status, headers, body).map_err(|e| {
         158  +
                ::aws_smithy_runtime_api::client::orchestrator::OrchestratorError::other(::aws_smithy_runtime_api::box_error::BoxError::from(e))
         159  +
            })?;
         160  +
         161  +
            let generic = generic_builder.build();
         162  +
            let error_code = match generic.code() {
         163  +
                ::std::option::Option::Some(code) => code,
         164  +
                ::std::option::Option::None => {
         165  +
                    return ::std::result::Result::Err(::aws_smithy_runtime_api::client::orchestrator::OrchestratorError::other(
         166  +
                        ::aws_smithy_runtime_api::box_error::BoxError::from(crate::operation::null_in_non_sparse::NullInNonSparseError::unhandled(
         167  +
                            generic,
         168  +
                        )),
         169  +
                    ))
         170  +
                }
         171  +
            };
         172  +
            let _error_message = generic.message().map(|msg| msg.to_owned());
         173  +
            let protocol = _cfg
         174  +
                .load::<::aws_smithy_schema::protocol::SharedClientProtocol>()
         175  +
                .expect("a SharedClientProtocol is required");
         176  +
            let err = match error_code {
         177  +
                "ExtraError" => crate::operation::null_in_non_sparse::NullInNonSparseError::ExtraError({
         178  +
                    let mut tmp = match protocol
         179  +
                        .deserialize_response(response, crate::types::error::ExtraError::SCHEMA, _cfg)
         180  +
                        .and_then(|mut deser| {
         181  +
                            crate::types::error::ExtraError::deserialize_with_response(
         182  +
                                &mut *deser,
         183  +
                                response.headers(),
         184  +
                                response.status().into(),
         185  +
                                body,
         186  +
                            )
         187  +
                        }) {
         188  +
                        ::std::result::Result::Ok(val) => val,
         189  +
                        ::std::result::Result::Err(e) => {
         190  +
                            return ::std::result::Result::Err(::aws_smithy_runtime_api::client::orchestrator::OrchestratorError::other(
         191  +
                                ::aws_smithy_runtime_api::box_error::BoxError::from(e),
         192  +
                            ))
         193  +
                        }
         194  +
                    };
         195  +
                    tmp.meta = generic;
         196  +
                    if tmp.message.is_none() {
         197  +
                        tmp.message = _error_message;
         198  +
                    }
         199  +
                    tmp
         200  +
                }),
         201  +
                _ => crate::operation::null_in_non_sparse::NullInNonSparseError::generic(generic),
         202  +
            };
         203  +
            ::std::result::Result::Err(::aws_smithy_runtime_api::client::orchestrator::OrchestratorError::operation(
         204  +
                ::aws_smithy_runtime_api::client::interceptors::context::Error::erase(err),
         205  +
            ))
  152    206   
        } else {
  153         -
            crate::protocol_serde::shape_null_in_non_sparse::de_null_in_non_sparse_http_response(status, headers, body)
  154         -
        };
  155         -
        crate::protocol_serde::type_erase_result(parse_result)
         207  +
            let protocol = _cfg
         208  +
                .load::<::aws_smithy_schema::protocol::SharedClientProtocol>()
         209  +
                .expect("a SharedClientProtocol is required");
         210  +
            let mut deser = protocol
         211  +
                .deserialize_response(response, NullInNonSparse::OUTPUT_SCHEMA, _cfg)
         212  +
                .map_err(|e| {
         213  +
                    ::aws_smithy_runtime_api::client::orchestrator::OrchestratorError::other(::aws_smithy_runtime_api::box_error::BoxError::from(e))
         214  +
                })?;
         215  +
            let body = response.body().bytes().expect("body loaded");
         216  +
            let output = crate::operation::null_in_non_sparse::NullInNonSparseOutput::deserialize_with_response(
         217  +
                &mut *deser,
         218  +
                response.headers(),
         219  +
                response.status().into(),
         220  +
                body,
         221  +
            )
         222  +
            .map_err(|e| {
         223  +
                ::aws_smithy_runtime_api::client::orchestrator::OrchestratorError::other(::aws_smithy_runtime_api::box_error::BoxError::from(e))
         224  +
            })?;
         225  +
            ::std::result::Result::Ok(::aws_smithy_runtime_api::client::interceptors::context::Output::erase(output))
         226  +
        }
  156    227   
    }
  157    228   
}
  158    229   
#[derive(Debug)]
  159    230   
struct NullInNonSparseRequestSerializer;
  160    231   
impl ::aws_smithy_runtime_api::client::ser_de::SerializeRequest for NullInNonSparseRequestSerializer {
  161    232   
    #[allow(unused_mut, clippy::let_and_return, clippy::needless_borrow, clippy::useless_conversion)]
  162    233   
    fn serialize_input(
  163    234   
        &self,
  164    235   
        input: ::aws_smithy_runtime_api::client::interceptors::context::Input,
  165    236   
        _cfg: &mut ::aws_smithy_types::config_bag::ConfigBag,
  166    237   
    ) -> ::std::result::Result<::aws_smithy_runtime_api::client::orchestrator::HttpRequest, ::aws_smithy_runtime_api::box_error::BoxError> {
  167    238   
        let input = input
  168    239   
            .downcast::<crate::operation::null_in_non_sparse::NullInNonSparseInput>()
  169    240   
            .expect("correct type");
  170         -
        let _header_serialization_settings = _cfg
  171         -
            .load::<crate::serialization_settings::HeaderSerializationSettings>()
  172         -
            .cloned()
  173         -
            .unwrap_or_default();
  174         -
        let mut request_builder = {
  175         -
            #[allow(clippy::uninlined_format_args)]
  176         -
            fn uri_base(
  177         -
                _input: &crate::operation::null_in_non_sparse::NullInNonSparseInput,
  178         -
                output: &mut ::std::string::String,
  179         -
            ) -> ::std::result::Result<(), ::aws_smithy_types::error::operation::BuildError> {
  180         -
                use ::std::fmt::Write as _;
  181         -
                ::std::write!(output, "/null-in-non-sparse").expect("formatting should succeed");
  182         -
                ::std::result::Result::Ok(())
  183         -
            }
  184         -
            #[allow(clippy::unnecessary_wraps)]
  185         -
            fn update_http_builder(
  186         -
                input: &crate::operation::null_in_non_sparse::NullInNonSparseInput,
  187         -
                builder: ::http_1x::request::Builder,
  188         -
            ) -> ::std::result::Result<::http_1x::request::Builder, ::aws_smithy_types::error::operation::BuildError> {
  189         -
                let mut uri = ::std::string::String::new();
  190         -
                uri_base(input, &mut uri)?;
  191         -
                ::std::result::Result::Ok(builder.method("POST").uri(uri))
  192         -
            }
  193         -
            let mut builder = update_http_builder(&input, ::http_1x::request::Builder::new())?;
  194         -
            builder
  195         -
        };
  196         -
        let body = ::aws_smithy_types::body::SdkBody::from("");
         241  +
        let protocol = _cfg
         242  +
            .load::<::aws_smithy_schema::protocol::SharedClientProtocol>()
         243  +
            .expect("a SharedClientProtocol is required");
         244  +
        let mut request = protocol
         245  +
            .serialize_request(&input, NullInNonSparse::INPUT_SCHEMA, "", _cfg)
         246  +
            .map_err(::aws_smithy_runtime_api::box_error::BoxError::from)?;
  197    247   
  198         -
        ::std::result::Result::Ok(request_builder.body(body).expect("valid request").try_into().unwrap())
         248  +
        return ::std::result::Result::Ok(request);
  199    249   
    }
  200    250   
}
  201    251   
#[derive(Debug)]
  202    252   
struct NullInNonSparseEndpointParamsInterceptor;
  203    253   
  204    254   
impl ::aws_smithy_runtime_api::client::interceptors::Intercept for NullInNonSparseEndpointParamsInterceptor {
  205    255   
    fn name(&self) -> &'static str {
  206    256   
        "NullInNonSparseEndpointParamsInterceptor"
  207    257   
    }
  208    258   
@@ -249,299 +351,421 @@
  269    319   
            .expect("the config must have a deserializer");
  270    320   
  271    321   
        let parsed = de.deserialize_streaming(&mut http_response);
  272    322   
        let parsed = parsed.unwrap_or_else(|| {
  273    323   
            let http_response = http_response.map(|body| {
  274    324   
                ::aws_smithy_types::body::SdkBody::from(::bytes::Bytes::copy_from_slice(&::aws_smithy_protocol_test::decode_body_data(
  275    325   
                    body.bytes().unwrap(),
  276    326   
                    ::aws_smithy_protocol_test::MediaType::from("unknown"),
  277    327   
                )))
  278    328   
            });
  279         -
            de.deserialize_nonstreaming(&http_response)
         329  +
            // Build a config bag with the protocol for schema-based deserialization
         330  +
            #[allow(unused_mut)]
         331  +
            let mut test_cfg = ::aws_smithy_types::config_bag::ConfigBag::base();
         332  +
            {
         333  +
                let mut layer = ::aws_smithy_types::config_bag::Layer::new("test_protocol");
         334  +
                layer.store_put(::aws_smithy_schema::protocol::SharedClientProtocol::new(
         335  +
                    ::aws_smithy_json::protocol::aws_rest_json_1::AwsRestJsonProtocol::new(),
         336  +
                ));
         337  +
                test_cfg.push_shared_layer(layer.freeze());
         338  +
            }
         339  +
            de.deserialize_nonstreaming_with_config(&http_response, &test_cfg)
  280    340   
        });
  281    341   
        let parsed = parsed
  282    342   
            .expect("should be successful response")
  283    343   
            .downcast::<crate::operation::null_in_non_sparse::NullInNonSparseOutput>()
  284    344   
            .unwrap();
  285    345   
        ::pretty_assertions::assert_eq!(parsed.list, expected_output.list, "Unexpected value for `list`");
  286    346   
        ::pretty_assertions::assert_eq!(parsed.map, expected_output.map, "Unexpected value for `map`");
  287    347   
        ::pretty_assertions::assert_eq!(parsed.union, expected_output.union, "Unexpected value for `union`");
  288    348   
    }
  289    349   
  290    350   
    /// Upper case error modeled lower case
  291    351   
    /// Test ID: ServiceLevelErrorClient
  292    352   
    #[::tokio::test]
  293    353   
    #[::tracing_test::traced_test]
  294    354   
    async fn service_level_error_client_response() {
  295    355   
        let expected_output = crate::types::error::ExtraError::builder().build();
  296    356   
        let mut http_response = ::aws_smithy_runtime_api::http::Response::try_from(
  297    357   
            ::http_1x::response::Builder::new()
  298    358   
                .header("X-Amzn-Errortype", "ExtraError")
  299    359   
                .status(500)
  300    360   
                .body(::aws_smithy_types::body::SdkBody::from(""))
  301    361   
                .unwrap(),
  302    362   
        )
  303    363   
        .unwrap();
  304    364   
        use ::aws_smithy_runtime_api::client::runtime_plugin::RuntimePlugin;
  305    365   
        use ::aws_smithy_runtime_api::client::ser_de::DeserializeResponse;
  306    366   
  307    367   
        let op = crate::operation::null_in_non_sparse::NullInNonSparse::new();
  308    368   
        let config = op.config().expect("the operation has config");
  309    369   
        let de = config
  310    370   
            .load::<::aws_smithy_runtime_api::client::ser_de::SharedResponseDeserializer>()
  311    371   
            .expect("the config must have a deserializer");
  312    372   
  313    373   
        let parsed = de.deserialize_streaming(&mut http_response);
  314    374   
        let parsed = parsed.unwrap_or_else(|| {
  315    375   
            let http_response = http_response.map(|body| {
  316    376   
                ::aws_smithy_types::body::SdkBody::from(::bytes::Bytes::copy_from_slice(&::aws_smithy_protocol_test::decode_body_data(
  317    377   
                    body.bytes().unwrap(),
  318    378   
                    ::aws_smithy_protocol_test::MediaType::from("unknown"),
  319    379   
                )))
  320    380   
            });
  321         -
            de.deserialize_nonstreaming(&http_response)
         381  +
            // Build a config bag with the protocol for schema-based deserialization
         382  +
            #[allow(unused_mut)]
         383  +
            let mut test_cfg = ::aws_smithy_types::config_bag::ConfigBag::base();
         384  +
            {
         385  +
                let mut layer = ::aws_smithy_types::config_bag::Layer::new("test_protocol");
         386  +
                layer.store_put(::aws_smithy_schema::protocol::SharedClientProtocol::new(
         387  +
                    ::aws_smithy_json::protocol::aws_rest_json_1::AwsRestJsonProtocol::new(),
         388  +
                ));
         389  +
                test_cfg.push_shared_layer(layer.freeze());
         390  +
            }
         391  +
            de.deserialize_nonstreaming_with_config(&http_response, &test_cfg)
  322    392   
        });
  323    393   
        let parsed = parsed.expect_err("should be error response");
  324    394   
        let parsed: &crate::operation::null_in_non_sparse::NullInNonSparseError =
  325    395   
            parsed.as_operation_error().expect("operation error").downcast_ref().unwrap();
  326    396   
        if let crate::operation::null_in_non_sparse::NullInNonSparseError::ExtraError(parsed) = parsed {
  327    397   
            ::pretty_assertions::assert_eq!(parsed.message, expected_output.message, "Unexpected value for `message`");
  328    398   
        } else {
  329    399   
            panic!("wrong variant: Got: {:?}. Expected: {:?}", parsed, expected_output);
  330    400   
        }
  331    401   
    }

tmp-codegen-diff/codegen-client-test/rest_json_extras/rust-client-codegen/src/operation/null_in_non_sparse/_null_in_non_sparse_input.rs

@@ -1,1 +69,83 @@
    2      2   
#[allow(missing_docs)] // documentation missing in model
    3      3   
#[non_exhaustive]
    4      4   
#[derive(::std::clone::Clone, ::std::cmp::PartialEq, ::std::fmt::Debug)]
    5      5   
pub struct NullInNonSparseInput {}
    6      6   
static NULLINNONSPARSEINPUT_SCHEMA_ID: ::aws_smithy_schema::ShapeId = ::aws_smithy_schema::ShapeId::from_static(
    7      7   
    "aws.protocoltests.restjson.synthetic#NullInNonSparseInput",
    8      8   
    "aws.protocoltests.restjson.synthetic",
    9      9   
    "NullInNonSparseInput",
   10     10   
);
   11     11   
static NULLINNONSPARSEINPUT_SCHEMA: ::aws_smithy_schema::Schema =
   12         -
    ::aws_smithy_schema::Schema::new_struct(NULLINNONSPARSEINPUT_SCHEMA_ID, ::aws_smithy_schema::ShapeType::Structure, &[]);
          12  +
    ::aws_smithy_schema::Schema::new_struct(NULLINNONSPARSEINPUT_SCHEMA_ID, ::aws_smithy_schema::ShapeType::Structure, &[])
          13  +
        .with_http(aws_smithy_schema::traits::HttpTrait::new("POST", "/null-in-non-sparse", None));
   13     14   
impl NullInNonSparseInput {
   14     15   
    /// The schema for this shape.
   15     16   
    pub const SCHEMA: &'static ::aws_smithy_schema::Schema = &NULLINNONSPARSEINPUT_SCHEMA;
   16     17   
}
   17     18   
impl ::aws_smithy_schema::serde::SerializableStruct for NullInNonSparseInput {
   18     19   
    #[allow(unused_variables, clippy::diverging_sub_expression)]
   19     20   
    fn serialize_members(
   20     21   
        &self,
   21     22   
        ser: &mut dyn ::aws_smithy_schema::serde::ShapeSerializer,
   22     23   
    ) -> ::std::result::Result<(), ::aws_smithy_schema::serde::SerdeError> {
   23     24   
        Ok(())
   24     25   
    }
   25     26   
}
   26     27   
impl NullInNonSparseInput {
   27     28   
    /// Deserializes this structure from a [`ShapeDeserializer`].
   28         -
    pub fn deserialize<D: ::aws_smithy_schema::serde::ShapeDeserializer>(
   29         -
        deserializer: &mut D,
          29  +
    pub fn deserialize(
          30  +
        deserializer: &mut dyn ::aws_smithy_schema::serde::ShapeDeserializer,
   30     31   
    ) -> ::std::result::Result<Self, ::aws_smithy_schema::serde::SerdeError> {
   31     32   
        #[allow(unused_variables, unused_mut)]
   32     33   
        let mut builder = Self::builder();
   33     34   
        #[allow(
   34     35   
            unused_variables,
   35     36   
            unreachable_code,
   36     37   
            clippy::single_match,
   37     38   
            clippy::match_single_binding,
   38     39   
            clippy::diverging_sub_expression
   39     40   
        )]
   40         -
        deserializer.read_struct(&NULLINNONSPARSEINPUT_SCHEMA, (), |_, member, deser| {
          41  +
        deserializer.read_struct(&NULLINNONSPARSEINPUT_SCHEMA, &mut |member, deser| {
   41     42   
            match member.member_index() {
   42     43   
                _ => {}
   43     44   
            }
   44     45   
            Ok(())
   45     46   
        })?;
   46     47   
        builder
   47     48   
            .build()
   48     49   
            .map_err(|e| aws_smithy_schema::serde::SerdeError::Custom { message: e.to_string() })
   49     50   
    }
   50     51   
}
          52  +
impl NullInNonSparseInput {
          53  +
    /// Deserializes this structure from a body deserializer and HTTP response.
          54  +
    pub fn deserialize_with_response(
          55  +
        _deserializer: &mut dyn ::aws_smithy_schema::serde::ShapeDeserializer,
          56  +
        _headers: &::aws_smithy_runtime_api::http::Headers,
          57  +
        _status: u16,
          58  +
        _body: &[u8],
          59  +
    ) -> ::std::result::Result<Self, ::aws_smithy_schema::serde::SerdeError> {
          60  +
        Self::builder()
          61  +
            .build()
          62  +
            .map_err(|e| aws_smithy_schema::serde::SerdeError::Custom { message: e.to_string() })
          63  +
    }
          64  +
}
   51     65   
impl NullInNonSparseInput {
   52     66   
    /// Creates a new builder-style object to manufacture [`NullInNonSparseInput`](crate::operation::null_in_non_sparse::NullInNonSparseInput).
   53     67   
    pub fn builder() -> crate::operation::null_in_non_sparse::builders::NullInNonSparseInputBuilder {
   54     68   
        crate::operation::null_in_non_sparse::builders::NullInNonSparseInputBuilder::default()
   55     69   
    }
   56     70   
}
   57     71   
   58     72   
/// A builder for [`NullInNonSparseInput`](crate::operation::null_in_non_sparse::NullInNonSparseInput).
   59     73   
#[derive(::std::clone::Clone, ::std::cmp::PartialEq, ::std::default::Default, ::std::fmt::Debug)]
   60     74   
#[non_exhaustive]

tmp-codegen-diff/codegen-client-test/rest_json_extras/rust-client-codegen/src/operation/null_in_non_sparse/_null_in_non_sparse_output.rs

@@ -77,77 +196,184 @@
   97     97   
                &|ser: &mut dyn ::aws_smithy_schema::serde::ShapeSerializer| {
   98     98   
                    for (key, value) in val {
   99     99   
                        ser.write_string(&::aws_smithy_schema::prelude::STRING, key)?;
  100    100   
                        ser.write_string(&::aws_smithy_schema::prelude::STRING, value)?;
  101    101   
                    }
  102    102   
                    Ok(())
  103    103   
                },
  104    104   
            )?;
  105    105   
        }
  106    106   
        if let Some(ref val) = self.union {
  107         -
            ser.write_null(&NULLINNONSPARSEOUTPUT_MEMBER_UNION)?;
         107  +
            ser.write_struct(&NULLINNONSPARSEOUTPUT_MEMBER_UNION, val)?;
  108    108   
        }
  109    109   
        Ok(())
  110    110   
    }
  111    111   
}
  112    112   
impl NullInNonSparseOutput {
  113    113   
    /// Deserializes this structure from a [`ShapeDeserializer`].
  114         -
    pub fn deserialize<D: ::aws_smithy_schema::serde::ShapeDeserializer>(
  115         -
        deserializer: &mut D,
         114  +
    pub fn deserialize(
         115  +
        deserializer: &mut dyn ::aws_smithy_schema::serde::ShapeDeserializer,
  116    116   
    ) -> ::std::result::Result<Self, ::aws_smithy_schema::serde::SerdeError> {
  117    117   
        #[allow(unused_variables, unused_mut)]
  118    118   
        let mut builder = Self::builder();
  119    119   
        #[allow(
  120    120   
            unused_variables,
  121    121   
            unreachable_code,
  122    122   
            clippy::single_match,
  123    123   
            clippy::match_single_binding,
  124    124   
            clippy::diverging_sub_expression
  125    125   
        )]
  126         -
        deserializer.read_struct(&NULLINNONSPARSEOUTPUT_SCHEMA, (), |_, member, deser| {
         126  +
        deserializer.read_struct(&NULLINNONSPARSEOUTPUT_SCHEMA, &mut |member, deser| {
  127    127   
            match member.member_index() {
  128    128   
                Some(0) => {
  129         -
                    builder.list = Some({
  130         -
                        let container = if let Some(cap) = deser.container_size() {
  131         -
                            Vec::with_capacity(cap)
  132         -
                        } else {
  133         -
                            Vec::new()
  134         -
                        };
  135         -
                        deser.read_list(member, container, |mut list, deser| {
  136         -
                            list.push(deser.read_string(member)?);
  137         -
                            Ok(list)
  138         -
                        })?
  139         -
                    });
         129  +
                    builder.list = Some(deser.read_string_list(member)?);
  140    130   
                }
  141    131   
                Some(1) => {
  142         -
                    builder.map = Some({
  143         -
                        let container = if let Some(cap) = deser.container_size() {
  144         -
                            std::collections::HashMap::with_capacity(cap)
  145         -
                        } else {
  146         -
                            std::collections::HashMap::new()
  147         -
                        };
  148         -
                        deser.read_map(member, container, |mut map, key, deser| {
  149         -
                            map.insert(key, deser.read_string(member)?);
  150         -
                            Ok(map)
  151         -
                        })?
  152         -
                    });
         132  +
                    builder.map = Some(deser.read_string_string_map(member)?);
  153    133   
                }
  154    134   
                Some(2) => {
  155         -
                    builder.union = Some({
  156         -
                        let _ = member;
  157         -
                        todo!("deserialize aggregate")
  158         -
                    });
         135  +
                    builder.union = Some(crate::types::SingleElementUnion::deserialize(deser)?);
  159    136   
                }
  160    137   
                _ => {}
  161    138   
            }
  162    139   
            Ok(())
  163    140   
        })?;
  164    141   
        Ok(builder.build())
  165    142   
    }
  166    143   
}
         144  +
impl NullInNonSparseOutput {
         145  +
    /// Deserializes this structure from a body deserializer and HTTP response.
         146  +
    pub fn deserialize_with_response(
         147  +
        deserializer: &mut dyn ::aws_smithy_schema::serde::ShapeDeserializer,
         148  +
        _headers: &::aws_smithy_runtime_api::http::Headers,
         149  +
        _status: u16,
         150  +
        _body: &[u8],
         151  +
    ) -> ::std::result::Result<Self, ::aws_smithy_schema::serde::SerdeError> {
         152  +
        Self::deserialize(deserializer)
         153  +
    }
         154  +
}
  167    155   
impl NullInNonSparseOutput {
  168    156   
    /// Creates a new builder-style object to manufacture [`NullInNonSparseOutput`](crate::operation::null_in_non_sparse::NullInNonSparseOutput).
  169    157   
    pub fn builder() -> crate::operation::null_in_non_sparse::builders::NullInNonSparseOutputBuilder {
  170    158   
        crate::operation::null_in_non_sparse::builders::NullInNonSparseOutputBuilder::default()
  171    159   
    }
  172    160   
}
  173    161   
  174    162   
/// A builder for [`NullInNonSparseOutput`](crate::operation::null_in_non_sparse::NullInNonSparseOutput).
  175    163   
#[derive(::std::clone::Clone, ::std::cmp::PartialEq, ::std::default::Default, ::std::fmt::Debug)]
  176    164   
#[non_exhaustive]

tmp-codegen-diff/codegen-client-test/rest_json_extras/rust-client-codegen/src/operation/primitive_int_header.rs

@@ -1,1 +40,44 @@
    1      1   
// Code generated by software.amazon.smithy.rust.codegen.smithy-rs. DO NOT EDIT.
    2      2   
/// Orchestration and serialization glue logic for `PrimitiveIntHeader`.
    3      3   
#[derive(::std::clone::Clone, ::std::default::Default, ::std::fmt::Debug)]
    4      4   
#[non_exhaustive]
    5      5   
pub struct PrimitiveIntHeader;
    6      6   
impl PrimitiveIntHeader {
    7      7   
    /// Creates a new `PrimitiveIntHeader`
    8      8   
    pub fn new() -> Self {
    9      9   
        Self
   10     10   
    }
          11  +
    /// The schema for this operation's input shape.
          12  +
    pub const INPUT_SCHEMA: &'static ::aws_smithy_schema::Schema = crate::operation::primitive_int_header::PrimitiveIntHeaderInput::SCHEMA;
          13  +
    /// The schema for this operation's output shape.
          14  +
    pub const OUTPUT_SCHEMA: &'static ::aws_smithy_schema::Schema = crate::operation::primitive_int_header::PrimitiveIntHeaderOutput::SCHEMA;
   11     15   
    pub(crate) async fn orchestrate(
   12     16   
        runtime_plugins: &::aws_smithy_runtime_api::client::runtime_plugin::RuntimePlugins,
   13     17   
        input: crate::operation::primitive_int_header::PrimitiveIntHeaderInput,
   14     18   
    ) -> ::std::result::Result<
   15     19   
        crate::operation::primitive_int_header::PrimitiveIntHeaderOutput,
   16     20   
        ::aws_smithy_runtime_api::client::result::SdkError<
   17     21   
            crate::operation::primitive_int_header::PrimitiveIntHeaderError,
   18     22   
            ::aws_smithy_runtime_api::client::orchestrator::HttpResponse,
   19     23   
        >,
   20     24   
    > {
@@ -110,114 +228,299 @@
  130    134   
                crate::operation::primitive_int_header::PrimitiveIntHeaderError,
  131    135   
            >::new());
  132    136   
  133    137   
        ::std::borrow::Cow::Owned(rcb)
  134    138   
    }
  135    139   
}
  136    140   
  137    141   
#[derive(Debug)]
  138    142   
struct PrimitiveIntHeaderResponseDeserializer;
  139    143   
impl ::aws_smithy_runtime_api::client::ser_de::DeserializeResponse for PrimitiveIntHeaderResponseDeserializer {
  140         -
    fn deserialize_nonstreaming(
         144  +
    fn deserialize_nonstreaming_with_config(
  141    145   
        &self,
  142    146   
        response: &::aws_smithy_runtime_api::client::orchestrator::HttpResponse,
         147  +
        _cfg: &::aws_smithy_types::config_bag::ConfigBag,
  143    148   
    ) -> ::aws_smithy_runtime_api::client::interceptors::context::OutputOrError {
  144    149   
        let (success, status) = (response.status().is_success(), response.status().as_u16());
  145         -
        let headers = response.headers();
  146         -
        let body = response.body().bytes().expect("body loaded");
  147    150   
        #[allow(unused_mut)]
  148    151   
        let mut force_error = false;
  149    152   
  150         -
        let parse_result = if !success && status != 200 || force_error {
  151         -
            crate::protocol_serde::shape_primitive_int_header::de_primitive_int_header_http_error(status, headers, body)
         153  +
        if !success && status != 200 || force_error {
         154  +
            let headers = response.headers();
         155  +
            let body = response.body().bytes().expect("body loaded");
         156  +
            #[allow(unused_mut)]
         157  +
            let mut generic_builder = crate::protocol_serde::parse_http_error_metadata(status, headers, body).map_err(|e| {
         158  +
                ::aws_smithy_runtime_api::client::orchestrator::OrchestratorError::other(::aws_smithy_runtime_api::box_error::BoxError::from(e))
         159  +
            })?;
         160  +
         161  +
            let generic = generic_builder.build();
         162  +
            let error_code = match generic.code() {
         163  +
                ::std::option::Option::Some(code) => code,
         164  +
                ::std::option::Option::None => {
         165  +
                    return ::std::result::Result::Err(::aws_smithy_runtime_api::client::orchestrator::OrchestratorError::other(
         166  +
                        ::aws_smithy_runtime_api::box_error::BoxError::from(
         167  +
                            crate::operation::primitive_int_header::PrimitiveIntHeaderError::unhandled(generic),
         168  +
                        ),
         169  +
                    ))
         170  +
                }
         171  +
            };
         172  +
            let _error_message = generic.message().map(|msg| msg.to_owned());
         173  +
            let protocol = _cfg
         174  +
                .load::<::aws_smithy_schema::protocol::SharedClientProtocol>()
         175  +
                .expect("a SharedClientProtocol is required");
         176  +
            let err = match error_code {
         177  +
                "ValidationException" => crate::operation::primitive_int_header::PrimitiveIntHeaderError::ValidationError({
         178  +
                    let mut tmp = match protocol
         179  +
                        .deserialize_response(response, crate::types::error::ValidationError::SCHEMA, _cfg)
         180  +
                        .and_then(|mut deser| {
         181  +
                            crate::types::error::ValidationError::deserialize_with_response(
         182  +
                                &mut *deser,
         183  +
                                response.headers(),
         184  +
                                response.status().into(),
         185  +
                                body,
         186  +
                            )
         187  +
                        }) {
         188  +
                        ::std::result::Result::Ok(val) => val,
         189  +
                        ::std::result::Result::Err(e) => {
         190  +
                            return ::std::result::Result::Err(::aws_smithy_runtime_api::client::orchestrator::OrchestratorError::other(
         191  +
                                ::aws_smithy_runtime_api::box_error::BoxError::from(e),
         192  +
                            ))
         193  +
                        }
         194  +
                    };
         195  +
                    tmp.meta = generic;
         196  +
                    tmp
         197  +
                }),
         198  +
                "ExtraError" => crate::operation::primitive_int_header::PrimitiveIntHeaderError::ExtraError({
         199  +
                    let mut tmp = match protocol
         200  +
                        .deserialize_response(response, crate::types::error::ExtraError::SCHEMA, _cfg)
         201  +
                        .and_then(|mut deser| {
         202  +
                            crate::types::error::ExtraError::deserialize_with_response(
         203  +
                                &mut *deser,
         204  +
                                response.headers(),
         205  +
                                response.status().into(),
         206  +
                                body,
         207  +
                            )
         208  +
                        }) {
         209  +
                        ::std::result::Result::Ok(val) => val,
         210  +
                        ::std::result::Result::Err(e) => {
         211  +
                            return ::std::result::Result::Err(::aws_smithy_runtime_api::client::orchestrator::OrchestratorError::other(
         212  +
                                ::aws_smithy_runtime_api::box_error::BoxError::from(e),
         213  +
                            ))
         214  +
                        }
         215  +
                    };
         216  +
                    tmp.meta = generic;
         217  +
                    if tmp.message.is_none() {
         218  +
                        tmp.message = _error_message;
         219  +
                    }
         220  +
                    tmp
         221  +
                }),
         222  +
                _ => crate::operation::primitive_int_header::PrimitiveIntHeaderError::generic(generic),
         223  +
            };
         224  +
            ::std::result::Result::Err(::aws_smithy_runtime_api::client::orchestrator::OrchestratorError::operation(
         225  +
                ::aws_smithy_runtime_api::client::interceptors::context::Error::erase(err),
         226  +
            ))
  152    227   
        } else {
  153         -
            crate::protocol_serde::shape_primitive_int_header::de_primitive_int_header_http_response(status, headers, body)
  154         -
        };
  155         -
        crate::protocol_serde::type_erase_result(parse_result)
         228  +
            let protocol = _cfg
         229  +
                .load::<::aws_smithy_schema::protocol::SharedClientProtocol>()
         230  +
                .expect("a SharedClientProtocol is required");
         231  +
            let mut deser = protocol
         232  +
                .deserialize_response(response, PrimitiveIntHeader::OUTPUT_SCHEMA, _cfg)
         233  +
                .map_err(|e| {
         234  +
                    ::aws_smithy_runtime_api::client::orchestrator::OrchestratorError::other(::aws_smithy_runtime_api::box_error::BoxError::from(e))
         235  +
                })?;
         236  +
            let body = response.body().bytes().expect("body loaded");
         237  +
            let output = crate::operation::primitive_int_header::PrimitiveIntHeaderOutput::deserialize_with_response(
         238  +
                &mut *deser,
         239  +
                response.headers(),
         240  +
                response.status().into(),
         241  +
                body,
         242  +
            )
         243  +
            .map_err(|e| {
         244  +
                ::aws_smithy_runtime_api::client::orchestrator::OrchestratorError::other(::aws_smithy_runtime_api::box_error::BoxError::from(e))
         245  +
            })?;
         246  +
            ::std::result::Result::Ok(::aws_smithy_runtime_api::client::interceptors::context::Output::erase(output))
         247  +
        }
  156    248   
    }
  157    249   
}
  158    250   
#[derive(Debug)]
  159    251   
struct PrimitiveIntHeaderRequestSerializer;
  160    252   
impl ::aws_smithy_runtime_api::client::ser_de::SerializeRequest for PrimitiveIntHeaderRequestSerializer {
  161    253   
    #[allow(unused_mut, clippy::let_and_return, clippy::needless_borrow, clippy::useless_conversion)]
  162    254   
    fn serialize_input(
  163    255   
        &self,
  164    256   
        input: ::aws_smithy_runtime_api::client::interceptors::context::Input,
  165    257   
        _cfg: &mut ::aws_smithy_types::config_bag::ConfigBag,
  166    258   
    ) -> ::std::result::Result<::aws_smithy_runtime_api::client::orchestrator::HttpRequest, ::aws_smithy_runtime_api::box_error::BoxError> {
  167    259   
        let input = input
  168    260   
            .downcast::<crate::operation::primitive_int_header::PrimitiveIntHeaderInput>()
  169    261   
            .expect("correct type");
  170         -
        let _header_serialization_settings = _cfg
  171         -
            .load::<crate::serialization_settings::HeaderSerializationSettings>()
  172         -
            .cloned()
  173         -
            .unwrap_or_default();
  174         -
        let mut request_builder = {
  175         -
            #[allow(clippy::uninlined_format_args)]
  176         -
            fn uri_base(
  177         -
                _input: &crate::operation::primitive_int_header::PrimitiveIntHeaderInput,
  178         -
                output: &mut ::std::string::String,
  179         -
            ) -> ::std::result::Result<(), ::aws_smithy_types::error::operation::BuildError> {
  180         -
                use ::std::fmt::Write as _;
  181         -
                ::std::write!(output, "/primitive").expect("formatting should succeed");
  182         -
                ::std::result::Result::Ok(())
  183         -
            }
  184         -
            #[allow(clippy::unnecessary_wraps)]
  185         -
            fn update_http_builder(
  186         -
                input: &crate::operation::primitive_int_header::PrimitiveIntHeaderInput,
  187         -
                builder: ::http_1x::request::Builder,
  188         -
            ) -> ::std::result::Result<::http_1x::request::Builder, ::aws_smithy_types::error::operation::BuildError> {
  189         -
                let mut uri = ::std::string::String::new();
  190         -
                uri_base(input, &mut uri)?;
  191         -
                ::std::result::Result::Ok(builder.method("POST").uri(uri))
  192         -
            }
  193         -
            let mut builder = update_http_builder(&input, ::http_1x::request::Builder::new())?;
  194         -
            builder
  195         -
        };
  196         -
        let body = ::aws_smithy_types::body::SdkBody::from("");
  197         -
  198         -
        ::std::result::Result::Ok(request_builder.body(body).expect("valid request").try_into().unwrap())
         262  +
        let protocol = _cfg
         263  +
            .load::<::aws_smithy_schema::protocol::SharedClientProtocol>()
         264  +
            .expect("a SharedClientProtocol is required");
         265  +
        let mut request = protocol
         266  +
            .serialize_request(&input, PrimitiveIntHeader::INPUT_SCHEMA, "", _cfg)
         267  +
            .map_err(::aws_smithy_runtime_api::box_error::BoxError::from)?;
         268  +
         269  +
        return ::std::result::Result::Ok(request);
  199    270   
    }
  200    271   
}
  201    272   
#[derive(Debug)]
  202    273   
struct PrimitiveIntHeaderEndpointParamsInterceptor;
  203    274   
  204    275   
impl ::aws_smithy_runtime_api::client::interceptors::Intercept for PrimitiveIntHeaderEndpointParamsInterceptor {
  205    276   
    fn name(&self) -> &'static str {
  206    277   
        "PrimitiveIntHeaderEndpointParamsInterceptor"
  207    278   
    }
  208    279   
@@ -242,313 +381,482 @@
  262    333   
            .expect("the config must have a deserializer");
  263    334   
  264    335   
        let parsed = de.deserialize_streaming(&mut http_response);
  265    336   
        let parsed = parsed.unwrap_or_else(|| {
  266    337   
            let http_response = http_response.map(|body| {
  267    338   
                ::aws_smithy_types::body::SdkBody::from(::bytes::Bytes::copy_from_slice(&::aws_smithy_protocol_test::decode_body_data(
  268    339   
                    body.bytes().unwrap(),
  269    340   
                    ::aws_smithy_protocol_test::MediaType::from("unknown"),
  270    341   
                )))
  271    342   
            });
  272         -
            de.deserialize_nonstreaming(&http_response)
         343  +
            // Build a config bag with the protocol for schema-based deserialization
         344  +
            #[allow(unused_mut)]
         345  +
            let mut test_cfg = ::aws_smithy_types::config_bag::ConfigBag::base();
         346  +
            {
         347  +
                let mut layer = ::aws_smithy_types::config_bag::Layer::new("test_protocol");
         348  +
                layer.store_put(::aws_smithy_schema::protocol::SharedClientProtocol::new(
         349  +
                    ::aws_smithy_json::protocol::aws_rest_json_1::AwsRestJsonProtocol::new(),
         350  +
                ));
         351  +
                test_cfg.push_shared_layer(layer.freeze());
         352  +
            }
         353  +
            de.deserialize_nonstreaming_with_config(&http_response, &test_cfg)
  273    354   
        });
  274    355   
        let parsed = parsed
  275    356   
            .expect("should be successful response")
  276    357   
            .downcast::<crate::operation::primitive_int_header::PrimitiveIntHeaderOutput>()
  277    358   
            .unwrap();
  278    359   
        ::pretty_assertions::assert_eq!(parsed.field, expected_output.field, "Unexpected value for `field`");
  279    360   
    }
  280    361   
    /// Test ID: DeserPrimitiveHeaderMissing
  281    362   
    #[::tokio::test]
  282    363   
    #[::tracing_test::traced_test]
  283    364   
    async fn deser_primitive_header_missing_response() {
  284    365   
        let expected_output = crate::operation::primitive_int_header::PrimitiveIntHeaderOutput::builder()
  285    366   
            .set_field(::std::option::Option::Some(0))
  286    367   
            .build();
  287    368   
        let mut http_response = ::aws_smithy_runtime_api::http::Response::try_from(
  288    369   
            ::http_1x::response::Builder::new()
  289    370   
                .status(200)
  290    371   
                .body(::aws_smithy_types::body::SdkBody::from(vec![]))
  291    372   
                .unwrap(),
  292    373   
        )
  293    374   
        .unwrap();
  294    375   
        use ::aws_smithy_runtime_api::client::runtime_plugin::RuntimePlugin;
  295    376   
        use ::aws_smithy_runtime_api::client::ser_de::DeserializeResponse;
  296    377   
  297    378   
        let op = crate::operation::primitive_int_header::PrimitiveIntHeader::new();
  298    379   
        let config = op.config().expect("the operation has config");
  299    380   
        let de = config
  300    381   
            .load::<::aws_smithy_runtime_api::client::ser_de::SharedResponseDeserializer>()
  301    382   
            .expect("the config must have a deserializer");
  302    383   
  303    384   
        let parsed = de.deserialize_streaming(&mut http_response);
  304    385   
        let parsed = parsed.unwrap_or_else(|| {
  305    386   
            let http_response = http_response.map(|body| {
  306    387   
                ::aws_smithy_types::body::SdkBody::from(::bytes::Bytes::copy_from_slice(&::aws_smithy_protocol_test::decode_body_data(
  307    388   
                    body.bytes().unwrap(),
  308    389   
                    ::aws_smithy_protocol_test::MediaType::from("unknown"),
  309    390   
                )))
  310    391   
            });
  311         -
            de.deserialize_nonstreaming(&http_response)
         392  +
            // Build a config bag with the protocol for schema-based deserialization
         393  +
            #[allow(unused_mut)]
         394  +
            let mut test_cfg = ::aws_smithy_types::config_bag::ConfigBag::base();
         395  +
            {
         396  +
                let mut layer = ::aws_smithy_types::config_bag::Layer::new("test_protocol");
         397  +
                layer.store_put(::aws_smithy_schema::protocol::SharedClientProtocol::new(
         398  +
                    ::aws_smithy_json::protocol::aws_rest_json_1::AwsRestJsonProtocol::new(),
         399  +
                ));
         400  +
                test_cfg.push_shared_layer(layer.freeze());
         401  +
            }
         402  +
            de.deserialize_nonstreaming_with_config(&http_response, &test_cfg)
  312    403   
        });
  313    404   
        let parsed = parsed
  314    405   
            .expect("should be successful response")
  315    406   
            .downcast::<crate::operation::primitive_int_header::PrimitiveIntHeaderOutput>()
  316    407   
            .unwrap();
  317    408   
        ::pretty_assertions::assert_eq!(parsed.field, expected_output.field, "Unexpected value for `field`");
  318    409   
    }
  319    410   
  320    411   
    /// Upper case error modeled lower case
  321    412   
    /// Test ID: ServiceLevelErrorClient
  322    413   
    #[::tokio::test]
  323    414   
    #[::tracing_test::traced_test]
  324    415   
    async fn service_level_error_client_response() {
  325    416   
        let expected_output = crate::types::error::ExtraError::builder().build();
  326    417   
        let mut http_response = ::aws_smithy_runtime_api::http::Response::try_from(
  327    418   
            ::http_1x::response::Builder::new()
  328    419   
                .header("X-Amzn-Errortype", "ExtraError")
  329    420   
                .status(500)
  330    421   
                .body(::aws_smithy_types::body::SdkBody::from(""))
  331    422   
                .unwrap(),
  332    423   
        )
  333    424   
        .unwrap();
  334    425   
        use ::aws_smithy_runtime_api::client::runtime_plugin::RuntimePlugin;
  335    426   
        use ::aws_smithy_runtime_api::client::ser_de::DeserializeResponse;
  336    427   
  337    428   
        let op = crate::operation::primitive_int_header::PrimitiveIntHeader::new();
  338    429   
        let config = op.config().expect("the operation has config");
  339    430   
        let de = config
  340    431   
            .load::<::aws_smithy_runtime_api::client::ser_de::SharedResponseDeserializer>()
  341    432   
            .expect("the config must have a deserializer");
  342    433   
  343    434   
        let parsed = de.deserialize_streaming(&mut http_response);
  344    435   
        let parsed = parsed.unwrap_or_else(|| {
  345    436   
            let http_response = http_response.map(|body| {
  346    437   
                ::aws_smithy_types::body::SdkBody::from(::bytes::Bytes::copy_from_slice(&::aws_smithy_protocol_test::decode_body_data(
  347    438   
                    body.bytes().unwrap(),
  348    439   
                    ::aws_smithy_protocol_test::MediaType::from("unknown"),
  349    440   
                )))
  350    441   
            });
  351         -
            de.deserialize_nonstreaming(&http_response)
         442  +
            // Build a config bag with the protocol for schema-based deserialization
         443  +
            #[allow(unused_mut)]
         444  +
            let mut test_cfg = ::aws_smithy_types::config_bag::ConfigBag::base();
         445  +
            {
         446  +
                let mut layer = ::aws_smithy_types::config_bag::Layer::new("test_protocol");
         447  +
                layer.store_put(::aws_smithy_schema::protocol::SharedClientProtocol::new(
         448  +
                    ::aws_smithy_json::protocol::aws_rest_json_1::AwsRestJsonProtocol::new(),
         449  +
                ));
         450  +
                test_cfg.push_shared_layer(layer.freeze());
         451  +
            }
         452  +
            de.deserialize_nonstreaming_with_config(&http_response, &test_cfg)
  352    453   
        });
  353    454   
        let parsed = parsed.expect_err("should be error response");
  354    455   
        let parsed: &crate::operation::primitive_int_header::PrimitiveIntHeaderError =
  355    456   
            parsed.as_operation_error().expect("operation error").downcast_ref().unwrap();
  356    457   
        if let crate::operation::primitive_int_header::PrimitiveIntHeaderError::ExtraError(parsed) = parsed {
  357    458   
            ::pretty_assertions::assert_eq!(parsed.message, expected_output.message, "Unexpected value for `message`");
  358    459   
        } else {
  359    460   
            panic!("wrong variant: Got: {:?}. Expected: {:?}", parsed, expected_output);
  360    461   
        }
  361    462   
    }