Client Test

Client Test

rev. 32b1b3c3761061baed26023be3219639e42d7d12 (ignoring whitespace)

Files changed:

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());
         150  +
        #[allow(unused_mut)]
         151  +
        let mut force_error = false;
         152  +
         153  +
        if !success && status != 200 || force_error {
  145    154   
            let headers = response.headers();
  146    155   
            let body = response.body().bytes().expect("body loaded");
  147    156   
            #[allow(unused_mut)]
  148         -
        let mut force_error = false;
         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  +
            })?;
  149    160   
  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)
  152         -
        } else {
  153         -
            crate::protocol_serde::shape_null_in_non_sparse::de_null_in_non_sparse_http_response(status, headers, body)
         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  +
                }
  154    171   
            };
  155         -
        crate::protocol_serde::type_erase_result(parse_result)
         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  +
            ))
         206  +
        } else {
         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());
         150  +
        #[allow(unused_mut)]
         151  +
        let mut force_error = false;
         152  +
         153  +
        if !success && status != 200 || force_error {
  145    154   
            let headers = response.headers();
  146    155   
            let body = response.body().bytes().expect("body loaded");
  147    156   
            #[allow(unused_mut)]
  148         -
        let mut force_error = false;
         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  +
            })?;
  149    160   
  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)
  152         -
        } else {
  153         -
            crate::protocol_serde::shape_primitive_int_header::de_primitive_int_header_http_response(status, headers, body)
         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  +
                }
  154    171   
            };
  155         -
        crate::protocol_serde::type_erase_result(parse_result)
         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  +
            ))
         227  +
        } else {
         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("");
         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)?;
  197    268   
  198         -
        ::std::result::Result::Ok(request_builder.body(body).expect("valid request").try_into().unwrap())
         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   
    }

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

@@ -1,1 +70,84 @@
    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 PrimitiveIntHeaderInput {}
    6      6   
static PRIMITIVEINTHEADERINPUT_SCHEMA_ID: ::aws_smithy_schema::ShapeId = ::aws_smithy_schema::ShapeId::from_static(
    7      7   
    "aws.protocoltests.restjson.synthetic#PrimitiveIntHeaderInput",
    8      8   
    "aws.protocoltests.restjson.synthetic",
    9      9   
    "PrimitiveIntHeaderInput",
   10     10   
);
   11     11   
static PRIMITIVEINTHEADERINPUT_SCHEMA: ::aws_smithy_schema::Schema =
   12         -
    ::aws_smithy_schema::Schema::new_struct(PRIMITIVEINTHEADERINPUT_SCHEMA_ID, ::aws_smithy_schema::ShapeType::Structure, &[]);
          12  +
    ::aws_smithy_schema::Schema::new_struct(PRIMITIVEINTHEADERINPUT_SCHEMA_ID, ::aws_smithy_schema::ShapeType::Structure, &[])
          13  +
        .with_http(aws_smithy_schema::traits::HttpTrait::new("POST", "/primitive", None));
   13     14   
impl PrimitiveIntHeaderInput {
   14     15   
    /// The schema for this shape.
   15     16   
    pub const SCHEMA: &'static ::aws_smithy_schema::Schema = &PRIMITIVEINTHEADERINPUT_SCHEMA;
   16     17   
}
   17     18   
impl ::aws_smithy_schema::serde::SerializableStruct for PrimitiveIntHeaderInput {
   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 PrimitiveIntHeaderInput {
   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(&PRIMITIVEINTHEADERINPUT_SCHEMA, (), |_, member, deser| {
          41  +
        deserializer.read_struct(&PRIMITIVEINTHEADERINPUT_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 PrimitiveIntHeaderInput {
          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 PrimitiveIntHeaderInput {
   52     66   
    /// Creates a new builder-style object to manufacture [`PrimitiveIntHeaderInput`](crate::operation::primitive_int_header::PrimitiveIntHeaderInput).
   53     67   
    pub fn builder() -> crate::operation::primitive_int_header::builders::PrimitiveIntHeaderInputBuilder {
   54     68   
        crate::operation::primitive_int_header::builders::PrimitiveIntHeaderInputBuilder::default()
   55     69   
    }
   56     70   
}
   57     71   
   58     72   
/// A builder for [`PrimitiveIntHeaderInput`](crate::operation::primitive_int_header::PrimitiveIntHeaderInput).
   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/primitive_int_header/_primitive_int_header_output.rs

@@ -25,25 +108,126 @@
   45     45   
    ) -> ::std::result::Result<(), ::aws_smithy_schema::serde::SerdeError> {
   46     46   
        {
   47     47   
            let val = &self.field;
   48     48   
            ser.write_integer(&PRIMITIVEINTHEADEROUTPUT_MEMBER_FIELD, *val)?;
   49     49   
        }
   50     50   
        Ok(())
   51     51   
    }
   52     52   
}
   53     53   
impl PrimitiveIntHeaderOutput {
   54     54   
    /// Deserializes this structure from a [`ShapeDeserializer`].
   55         -
    pub fn deserialize<D: ::aws_smithy_schema::serde::ShapeDeserializer>(
   56         -
        deserializer: &mut D,
          55  +
    pub fn deserialize(
          56  +
        deserializer: &mut dyn ::aws_smithy_schema::serde::ShapeDeserializer,
   57     57   
    ) -> ::std::result::Result<Self, ::aws_smithy_schema::serde::SerdeError> {
   58     58   
        #[allow(unused_variables, unused_mut)]
   59     59   
        let mut builder = Self::builder();
   60     60   
        #[allow(
   61     61   
            unused_variables,
   62     62   
            unreachable_code,
   63     63   
            clippy::single_match,
   64     64   
            clippy::match_single_binding,
   65     65   
            clippy::diverging_sub_expression
   66     66   
        )]
   67         -
        deserializer.read_struct(&PRIMITIVEINTHEADEROUTPUT_SCHEMA, (), |_, member, deser| {
          67  +
        deserializer.read_struct(&PRIMITIVEINTHEADEROUTPUT_SCHEMA, &mut |member, deser| {
   68     68   
            match member.member_index() {
   69     69   
                Some(0) => {
   70     70   
                    builder.field = Some(deser.read_integer(member)?);
   71     71   
                }
   72     72   
                _ => {}
   73     73   
            }
   74     74   
            Ok(())
   75     75   
        })?;
   76     76   
        Ok(builder.build())
   77     77   
    }
   78     78   
}
          79  +
impl PrimitiveIntHeaderOutput {
          80  +
    /// Deserializes this structure from a body deserializer and HTTP response headers.
          81  +
    /// Header-bound members are read directly from headers, avoiding runtime
          82  +
    /// member iteration overhead. Body members are read via the deserializer.
          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  +
        #[allow(unused_variables, unused_mut)]
          90  +
        let mut builder = Self::builder();
          91  +
        if let Some(val) = headers.get("x-field") {
          92  +
            builder.field = val.parse::<i32>().ok();
          93  +
        }
          94  +
        Ok(builder.build())
          95  +
    }
          96  +
}
   79     97   
impl PrimitiveIntHeaderOutput {
   80     98   
    /// Creates a new builder-style object to manufacture [`PrimitiveIntHeaderOutput`](crate::operation::primitive_int_header::PrimitiveIntHeaderOutput).
   81     99   
    pub fn builder() -> crate::operation::primitive_int_header::builders::PrimitiveIntHeaderOutputBuilder {
   82    100   
        crate::operation::primitive_int_header::builders::PrimitiveIntHeaderOutputBuilder::default()
   83    101   
    }
   84    102   
}
   85    103   
   86    104   
/// A builder for [`PrimitiveIntHeaderOutput`](crate::operation::primitive_int_header::PrimitiveIntHeaderOutput).
   87    105   
#[derive(::std::clone::Clone, ::std::cmp::PartialEq, ::std::default::Default, ::std::fmt::Debug)]
   88    106   
#[non_exhaustive]

tmp-codegen-diff/codegen-client-test/rest_json_extras/rust-client-codegen/src/operation/primitive_int_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 `PrimitiveIntOp`.
    3      3   
#[derive(::std::clone::Clone, ::std::default::Default, ::std::fmt::Debug)]
    4      4   
#[non_exhaustive]
    5      5   
pub struct PrimitiveIntOp;
    6      6   
impl PrimitiveIntOp {
    7      7   
    /// Creates a new `PrimitiveIntOp`
    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_op::PrimitiveIntOpInput::SCHEMA;
          13  +
    /// The schema for this operation's output shape.
          14  +
    pub const OUTPUT_SCHEMA: &'static ::aws_smithy_schema::Schema = crate::operation::primitive_int_op::PrimitiveIntOpOutput::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_op::PrimitiveIntOpInput,
   14     18   
    ) -> ::std::result::Result<
   15     19   
        crate::operation::primitive_int_op::PrimitiveIntOpOutput,
   16     20   
        ::aws_smithy_runtime_api::client::result::SdkError<
   17     21   
            crate::operation::primitive_int_op::PrimitiveIntOpError,
   18     22   
            ::aws_smithy_runtime_api::client::orchestrator::HttpResponse,
   19     23   
        >,
   20     24   
    > {
@@ -104,108 +226,272 @@
  124    128   
                crate::operation::primitive_int_op::PrimitiveIntOpError,
  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 PrimitiveIntOpResponseDeserializer;
  133    137   
impl ::aws_smithy_runtime_api::client::ser_de::DeserializeResponse for PrimitiveIntOpResponseDeserializer {
  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());
         144  +
        #[allow(unused_mut)]
         145  +
        let mut force_error = false;
         146  +
         147  +
        if !success && status != 200 || force_error {
  139    148   
            let headers = response.headers();
  140    149   
            let body = response.body().bytes().expect("body loaded");
  141    150   
            #[allow(unused_mut)]
  142         -
        let mut force_error = false;
         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  +
            })?;
  143    154   
  144         -
        let parse_result = if !success && status != 200 || force_error {
  145         -
            crate::protocol_serde::shape_primitive_int_op::de_primitive_int_op_http_error(status, headers, body)
  146         -
        } else {
  147         -
            crate::protocol_serde::shape_primitive_int_op::de_primitive_int_op_http_response(status, headers, body)
         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::primitive_int_op::PrimitiveIntOpError::unhandled(
         161  +
                            generic,
         162  +
                        )),
         163  +
                    ))
         164  +
                }
  148    165   
            };
  149         -
        crate::protocol_serde::type_erase_result(parse_result)
         166  +
            let _error_message = generic.message().map(|msg| msg.to_owned());
         167  +
            let protocol = _cfg
         168  +
                .load::<::aws_smithy_schema::protocol::SharedClientProtocol>()
         169  +
                .expect("a SharedClientProtocol is required");
         170  +
            let err = match error_code {
         171  +
                "ExtraError" => crate::operation::primitive_int_op::PrimitiveIntOpError::ExtraError({
         172  +
                    let mut tmp = match protocol
         173  +
                        .deserialize_response(response, crate::types::error::ExtraError::SCHEMA, _cfg)
         174  +
                        .and_then(|mut deser| {
         175  +
                            crate::types::error::ExtraError::deserialize_with_response(
         176  +
                                &mut *deser,
         177  +
                                response.headers(),
         178  +
                                response.status().into(),
         179  +
                                body,
         180  +
                            )
         181  +
                        }) {
         182  +
                        ::std::result::Result::Ok(val) => val,
         183  +
                        ::std::result::Result::Err(e) => {
         184  +
                            return ::std::result::Result::Err(::aws_smithy_runtime_api::client::orchestrator::OrchestratorError::other(
         185  +
                                ::aws_smithy_runtime_api::box_error::BoxError::from(e),
         186  +
                            ))
         187  +
                        }
         188  +
                    };
         189  +
                    tmp.meta = generic;
         190  +
                    if tmp.message.is_none() {
         191  +
                        tmp.message = _error_message;
         192  +
                    }
         193  +
                    tmp
         194  +
                }),
         195  +
                _ => crate::operation::primitive_int_op::PrimitiveIntOpError::generic(generic),
         196  +
            };
         197  +
            ::std::result::Result::Err(::aws_smithy_runtime_api::client::orchestrator::OrchestratorError::operation(
         198  +
                ::aws_smithy_runtime_api::client::interceptors::context::Error::erase(err),
         199  +
            ))
         200  +
        } else {
         201  +
            let protocol = _cfg
         202  +
                .load::<::aws_smithy_schema::protocol::SharedClientProtocol>()
         203  +
                .expect("a SharedClientProtocol is required");
         204  +
            let mut deser = protocol
         205  +
                .deserialize_response(response, PrimitiveIntOp::OUTPUT_SCHEMA, _cfg)
         206  +
                .map_err(|e| {
         207  +
                    ::aws_smithy_runtime_api::client::orchestrator::OrchestratorError::other(::aws_smithy_runtime_api::box_error::BoxError::from(e))
         208  +
                })?;
         209  +
            let body = response.body().bytes().expect("body loaded");
         210  +
            let output = crate::operation::primitive_int_op::PrimitiveIntOpOutput::deserialize_with_response(
         211  +
                &mut *deser,
         212  +
                response.headers(),
         213  +
                response.status().into(),
         214  +
                body,
         215  +
            )
         216  +
            .map_err(|e| {
         217  +
                ::aws_smithy_runtime_api::client::orchestrator::OrchestratorError::other(::aws_smithy_runtime_api::box_error::BoxError::from(e))
         218  +
            })?;
         219  +
            ::std::result::Result::Ok(::aws_smithy_runtime_api::client::interceptors::context::Output::erase(output))
         220  +
        }
  150    221   
    }
  151    222   
}
  152    223   
#[derive(Debug)]
  153    224   
struct PrimitiveIntOpRequestSerializer;
  154    225   
impl ::aws_smithy_runtime_api::client::ser_de::SerializeRequest for PrimitiveIntOpRequestSerializer {
  155    226   
    #[allow(unused_mut, clippy::let_and_return, clippy::needless_borrow, clippy::useless_conversion)]
  156    227   
    fn serialize_input(
  157    228   
        &self,
  158    229   
        input: ::aws_smithy_runtime_api::client::interceptors::context::Input,
  159    230   
        _cfg: &mut ::aws_smithy_types::config_bag::ConfigBag,
  160    231   
    ) -> ::std::result::Result<::aws_smithy_runtime_api::client::orchestrator::HttpRequest, ::aws_smithy_runtime_api::box_error::BoxError> {
  161    232   
        let input = input
  162    233   
            .downcast::<crate::operation::primitive_int_op::PrimitiveIntOpInput>()
  163    234   
            .expect("correct type");
  164         -
        let _header_serialization_settings = _cfg
  165         -
            .load::<crate::serialization_settings::HeaderSerializationSettings>()
  166         -
            .cloned()
  167         -
            .unwrap_or_default();
  168         -
        let mut request_builder = {
  169         -
            #[allow(clippy::uninlined_format_args)]
  170         -
            fn uri_base(
  171         -
                _input: &crate::operation::primitive_int_op::PrimitiveIntOpInput,
  172         -
                output: &mut ::std::string::String,
  173         -
            ) -> ::std::result::Result<(), ::aws_smithy_types::error::operation::BuildError> {
  174         -
                use ::std::fmt::Write as _;
  175         -
                ::std::write!(output, "/primitive-document").expect("formatting should succeed");
  176         -
                ::std::result::Result::Ok(())
  177         -
            }
  178         -
            #[allow(clippy::unnecessary_wraps)]
  179         -
            fn update_http_builder(
  180         -
                input: &crate::operation::primitive_int_op::PrimitiveIntOpInput,
  181         -
                builder: ::http_1x::request::Builder,
  182         -
            ) -> ::std::result::Result<::http_1x::request::Builder, ::aws_smithy_types::error::operation::BuildError> {
  183         -
                let mut uri = ::std::string::String::new();
  184         -
                uri_base(input, &mut uri)?;
  185         -
                ::std::result::Result::Ok(builder.method("POST").uri(uri))
  186         -
            }
  187         -
            let mut builder = update_http_builder(&input, ::http_1x::request::Builder::new())?;
  188         -
            builder = _header_serialization_settings.set_default_header(builder, ::http_1x::header::CONTENT_TYPE, "application/json");
  189         -
            builder
  190         -
        };
  191         -
        let body = ::aws_smithy_types::body::SdkBody::from(crate::protocol_serde::shape_primitive_int_op::ser_primitive_int_op_input(&input)?);
  192         -
        if let Some(content_length) = body.content_length() {
  193         -
            let content_length = content_length.to_string();
  194         -
            request_builder = _header_serialization_settings.set_default_header(request_builder, ::http_1x::header::CONTENT_LENGTH, &content_length);
  195         -
        }
  196         -
        ::std::result::Result::Ok(request_builder.body(body).expect("valid request").try_into().unwrap())
         235  +
        let protocol = _cfg
         236  +
            .load::<::aws_smithy_schema::protocol::SharedClientProtocol>()
         237  +
            .expect("a SharedClientProtocol is required");
         238  +
        let mut request = protocol
         239  +
            .serialize_request(&input, PrimitiveIntOp::INPUT_SCHEMA, "", _cfg)
         240  +
            .map_err(::aws_smithy_runtime_api::box_error::BoxError::from)?;
         241  +
         242  +
        return ::std::result::Result::Ok(request);
  197    243   
    }
  198    244   
}
  199    245   
#[derive(Debug)]
  200    246   
struct PrimitiveIntOpEndpointParamsInterceptor;
  201    247   
  202    248   
impl ::aws_smithy_runtime_api::client::interceptors::Intercept for PrimitiveIntOpEndpointParamsInterceptor {
  203    249   
    fn name(&self) -> &'static str {
  204    250   
        "PrimitiveIntOpEndpointParamsInterceptor"
  205    251   
    }
  206    252
@@ -269,315 +329,385 @@
  289    335   
            .expect("the config must have a deserializer");
  290    336   
  291    337   
        let parsed = de.deserialize_streaming(&mut http_response);
  292    338   
        let parsed = parsed.unwrap_or_else(|| {
  293    339   
            let http_response = http_response.map(|body| {
  294    340   
                ::aws_smithy_types::body::SdkBody::from(::bytes::Bytes::copy_from_slice(&::aws_smithy_protocol_test::decode_body_data(
  295    341   
                    body.bytes().unwrap(),
  296    342   
                    ::aws_smithy_protocol_test::MediaType::from("unknown"),
  297    343   
                )))
  298    344   
            });
  299         -
            de.deserialize_nonstreaming(&http_response)
         345  +
            // Build a config bag with the protocol for schema-based deserialization
         346  +
            #[allow(unused_mut)]
         347  +
            let mut test_cfg = ::aws_smithy_types::config_bag::ConfigBag::base();
         348  +
            {
         349  +
                let mut layer = ::aws_smithy_types::config_bag::Layer::new("test_protocol");
         350  +
                layer.store_put(::aws_smithy_schema::protocol::SharedClientProtocol::new(
         351  +
                    ::aws_smithy_json::protocol::aws_rest_json_1::AwsRestJsonProtocol::new(),
         352  +
                ));
         353  +
                test_cfg.push_shared_layer(layer.freeze());
         354  +
            }
         355  +
            de.deserialize_nonstreaming_with_config(&http_response, &test_cfg)
  300    356   
        });
  301    357   
        let parsed = parsed.expect_err("should be error response");
  302    358   
        let parsed: &crate::operation::primitive_int_op::PrimitiveIntOpError =
  303    359   
            parsed.as_operation_error().expect("operation error").downcast_ref().unwrap();
  304    360   
        if let crate::operation::primitive_int_op::PrimitiveIntOpError::ExtraError(parsed) = parsed {
  305    361   
            ::pretty_assertions::assert_eq!(parsed.message, expected_output.message, "Unexpected value for `message`");
  306    362   
        } else {
  307    363   
            panic!("wrong variant: Got: {:?}. Expected: {:?}", parsed, expected_output);
  308    364   
        }
  309    365   
    }

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

@@ -4,4 +108,120 @@
   24     24   
        "PrimitiveIntOpInput",
   25     25   
    ),
   26     26   
    ::aws_smithy_schema::ShapeType::Integer,
   27     27   
    "value",
   28     28   
    0,
   29     29   
);
   30     30   
static PRIMITIVEINTOPINPUT_SCHEMA: ::aws_smithy_schema::Schema = ::aws_smithy_schema::Schema::new_struct(
   31     31   
    PRIMITIVEINTOPINPUT_SCHEMA_ID,
   32     32   
    ::aws_smithy_schema::ShapeType::Structure,
   33     33   
    &[&PRIMITIVEINTOPINPUT_MEMBER_VALUE],
   34         -
);
          34  +
)
          35  +
.with_http(aws_smithy_schema::traits::HttpTrait::new("POST", "/primitive-document", None));
   35     36   
impl PrimitiveIntOpInput {
   36     37   
    /// The schema for this shape.
   37     38   
    pub const SCHEMA: &'static ::aws_smithy_schema::Schema = &PRIMITIVEINTOPINPUT_SCHEMA;
   38     39   
}
   39     40   
impl ::aws_smithy_schema::serde::SerializableStruct for PrimitiveIntOpInput {
   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.value {
   46     47   
            ser.write_integer(&PRIMITIVEINTOPINPUT_MEMBER_VALUE, *val)?;
   47     48   
        }
   48     49   
        Ok(())
   49     50   
    }
   50     51   
}
   51     52   
impl PrimitiveIntOpInput {
   52     53   
    /// Deserializes this structure from a [`ShapeDeserializer`].
   53         -
    pub fn deserialize<D: ::aws_smithy_schema::serde::ShapeDeserializer>(
   54         -
        deserializer: &mut D,
          54  +
    pub fn deserialize(
          55  +
        deserializer: &mut dyn ::aws_smithy_schema::serde::ShapeDeserializer,
   55     56   
    ) -> ::std::result::Result<Self, ::aws_smithy_schema::serde::SerdeError> {
   56     57   
        #[allow(unused_variables, unused_mut)]
   57     58   
        let mut builder = Self::builder();
   58     59   
        #[allow(
   59     60   
            unused_variables,
   60     61   
            unreachable_code,
   61     62   
            clippy::single_match,
   62     63   
            clippy::match_single_binding,
   63     64   
            clippy::diverging_sub_expression
   64     65   
        )]
   65         -
        deserializer.read_struct(&PRIMITIVEINTOPINPUT_SCHEMA, (), |_, member, deser| {
          66  +
        deserializer.read_struct(&PRIMITIVEINTOPINPUT_SCHEMA, &mut |member, deser| {
   66     67   
            match member.member_index() {
   67     68   
                Some(0) => {
   68     69   
                    builder.value = Some(deser.read_integer(member)?);
   69     70   
                }
   70     71   
                _ => {}
   71     72   
            }
   72     73   
            Ok(())
   73     74   
        })?;
   74     75   
        builder
   75     76   
            .build()
   76     77   
            .map_err(|e| aws_smithy_schema::serde::SerdeError::Custom { message: e.to_string() })
   77     78   
    }
   78     79   
}
          80  +
impl PrimitiveIntOpInput {
          81  +
    /// Deserializes this structure from a body deserializer and HTTP response.
          82  +
    pub fn deserialize_with_response(
          83  +
        deserializer: &mut dyn ::aws_smithy_schema::serde::ShapeDeserializer,
          84  +
        _headers: &::aws_smithy_runtime_api::http::Headers,
          85  +
        _status: u16,
          86  +
        _body: &[u8],
          87  +
    ) -> ::std::result::Result<Self, ::aws_smithy_schema::serde::SerdeError> {
          88  +
        Self::deserialize(deserializer)
          89  +
    }
          90  +
}
   79     91   
impl PrimitiveIntOpInput {
   80     92   
    /// Creates a new builder-style object to manufacture [`PrimitiveIntOpInput`](crate::operation::primitive_int_op::PrimitiveIntOpInput).
   81     93   
    pub fn builder() -> crate::operation::primitive_int_op::builders::PrimitiveIntOpInputBuilder {
   82     94   
        crate::operation::primitive_int_op::builders::PrimitiveIntOpInputBuilder::default()
   83     95   
    }
   84     96   
}
   85     97   
   86     98   
/// A builder for [`PrimitiveIntOpInput`](crate::operation::primitive_int_op::PrimitiveIntOpInput).
   87     99   
#[derive(::std::clone::Clone, ::std::cmp::PartialEq, ::std::default::Default, ::std::fmt::Debug)]
   88    100   
#[non_exhaustive]

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

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

tmp-codegen-diff/codegen-client-test/rest_json_extras/rust-client-codegen/src/operation/query_precedence.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 `QueryPrecedence`.
    3      3   
#[derive(::std::clone::Clone, ::std::default::Default, ::std::fmt::Debug)]
    4      4   
#[non_exhaustive]
    5      5   
pub struct QueryPrecedence;
    6      6   
impl QueryPrecedence {
    7      7   
    /// Creates a new `QueryPrecedence`
    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::query_precedence::QueryPrecedenceInput::SCHEMA;
          13  +
    /// The schema for this operation's output shape.
          14  +
    pub const OUTPUT_SCHEMA: &'static ::aws_smithy_schema::Schema = crate::operation::query_precedence::QueryPrecedenceOutput::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::query_precedence::QueryPrecedenceInput,
   14     18   
    ) -> ::std::result::Result<
   15     19   
        crate::operation::query_precedence::QueryPrecedenceOutput,
   16     20   
        ::aws_smithy_runtime_api::client::result::SdkError<
   17     21   
            crate::operation::query_precedence::QueryPrecedenceError,
   18     22   
            ::aws_smithy_runtime_api::client::orchestrator::HttpResponse,
   19     23   
        >,
   20     24   
    > {
@@ -110,114 +251,316 @@
  130    134   
                crate::operation::query_precedence::QueryPrecedenceError,
  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 QueryPrecedenceResponseDeserializer;
  139    143   
impl ::aws_smithy_runtime_api::client::ser_de::DeserializeResponse for QueryPrecedenceResponseDeserializer {
  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());
         150  +
        #[allow(unused_mut)]
         151  +
        let mut force_error = false;
         152  +
         153  +
        if !success && status != 200 || force_error {
  145    154   
            let headers = response.headers();
  146    155   
            let body = response.body().bytes().expect("body loaded");
  147    156   
            #[allow(unused_mut)]
  148         -
        let mut force_error = false;
         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  +
            })?;
  149    160   
  150         -
        let parse_result = if !success && status != 200 || force_error {
  151         -
            crate::protocol_serde::shape_query_precedence::de_query_precedence_http_error(status, headers, body)
  152         -
        } else {
  153         -
            crate::protocol_serde::shape_query_precedence::de_query_precedence_http_response(status, headers, body)
         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::query_precedence::QueryPrecedenceError::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::query_precedence::QueryPrecedenceError::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  +
                        }
  154    194   
                    };
  155         -
        crate::protocol_serde::type_erase_result(parse_result)
         195  +
                    tmp.meta = generic;
         196  +
                    if tmp.message.is_none() {
         197  +
                        tmp.message = _error_message;
         198  +
                    }
         199  +
                    tmp
         200  +
                }),
         201  +
                _ => crate::operation::query_precedence::QueryPrecedenceError::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  +
            ))
         206  +
        } else {
         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, QueryPrecedence::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::query_precedence::QueryPrecedenceOutput::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 QueryPrecedenceRequestSerializer;
  160    231   
impl ::aws_smithy_runtime_api::client::ser_de::SerializeRequest for QueryPrecedenceRequestSerializer {
  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::query_precedence::QueryPrecedenceInput>()
  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::query_precedence::QueryPrecedenceInput,
  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, "/Precedence").expect("formatting should succeed");
  182         -
                ::std::result::Result::Ok(())
  183         -
            }
  184         -
            fn uri_query(
  185         -
                _input: &crate::operation::query_precedence::QueryPrecedenceInput,
  186         -
                mut output: &mut ::std::string::String,
  187         -
            ) -> ::std::result::Result<(), ::aws_smithy_types::error::operation::BuildError> {
  188         -
                let mut query = ::aws_smithy_http::query::Writer::new(output);
  189         -
                let protected_params = ["bar"];
  190         -
                if let ::std::option::Option::Some(inner_1) = &_input.baz {
  191         -
                    {
  192         -
                        for (k, v) in inner_1 {
  193         -
                            if !protected_params.contains(&k.as_str()) {
  194         -
                                query.push_kv(&::aws_smithy_http::query::fmt_string(k), &::aws_smithy_http::query::fmt_string(v));
  195         -
                            }
  196         -
                        }
  197         -
                    }
  198         -
                }
  199         -
                if let ::std::option::Option::Some(inner_2) = &_input.foo {
         241  +
        let protocol = _cfg
         242  +
            .load::<::aws_smithy_schema::protocol::SharedClientProtocol>()
         243  +
            .expect("a SharedClientProtocol is required");
         244  +
        if protocol.supports_http_bindings() {
         245  +
            let mut request = protocol
         246  +
                .serialize_body(&input, QueryPrecedence::INPUT_SCHEMA, "", _cfg)
         247  +
                .map_err(::aws_smithy_runtime_api::box_error::BoxError::from)?;
  200    248   
            {
  201         -
                        query.push_kv("bar", &::aws_smithy_http::query::fmt_string(inner_2));
  202         -
                    }
         249  +
                let mut uri = "/Precedence".to_string();
         250  +
                let mut query_params: Vec<(String, String)> = Vec::new();
         251  +
                if let Some(ref val) = input.foo {
         252  +
                    query_params.push(("bar".to_string(), val.to_string()));
         253  +
                }
         254  +
                if let Some(ref map) = input.baz {
         255  +
                    for (k, v) in map {
         256  +
                        if matches!(k.as_str(), "bar") {
         257  +
                            continue;
         258  +
                        }
         259  +
                        query_params.push((k.clone(), v.clone()));
         260  +
                    }
         261  +
                }
         262  +
                if !query_params.is_empty() {
         263  +
                    uri.push(if uri.contains('?') { '&' } else { '?' });
         264  +
                    let pairs: Vec<String> = query_params
         265  +
                        .iter()
         266  +
                        .map(|(k, v)| {
         267  +
                            format!(
         268  +
                                "{}={}",
         269  +
                                ::aws_smithy_schema::http_protocol::percent_encode(k),
         270  +
                                ::aws_smithy_schema::http_protocol::percent_encode(v)
         271  +
                            )
         272  +
                        })
         273  +
                        .collect();
         274  +
                    uri.push_str(&pairs.join("&"));
  203    275   
                }
  204         -
                ::std::result::Result::Ok(())
         276  +
                request.set_uri(uri.as_str()).expect("valid URI");
  205    277   
            }
  206         -
            #[allow(clippy::unnecessary_wraps)]
  207         -
            fn update_http_builder(
  208         -
                input: &crate::operation::query_precedence::QueryPrecedenceInput,
  209         -
                builder: ::http_1x::request::Builder,
  210         -
            ) -> ::std::result::Result<::http_1x::request::Builder, ::aws_smithy_types::error::operation::BuildError> {
  211         -
                let mut uri = ::std::string::String::new();
  212         -
                uri_base(input, &mut uri)?;
  213         -
                uri_query(input, &mut uri)?;
  214         -
                ::std::result::Result::Ok(builder.method("POST").uri(uri))
  215         -
            }
  216         -
            let mut builder = update_http_builder(&input, ::http_1x::request::Builder::new())?;
  217         -
            builder
  218         -
        };
  219         -
        let body = ::aws_smithy_types::body::SdkBody::from("");
  220    278   
  221         -
        ::std::result::Result::Ok(request_builder.body(body).expect("valid request").try_into().unwrap())
         279  +
            return ::std::result::Result::Ok(request);
         280  +
        } else {
         281  +
            let mut request = protocol
         282  +
                .serialize_request(&input, QueryPrecedence::INPUT_SCHEMA, "", _cfg)
         283  +
                .map_err(::aws_smithy_runtime_api::box_error::BoxError::from)?;
         284  +
         285  +
            return ::std::result::Result::Ok(request);
         286  +
        }
  222    287   
    }
  223    288   
}
  224    289   
#[derive(Debug)]
  225    290   
struct QueryPrecedenceEndpointParamsInterceptor;
  226    291   
  227    292   
impl ::aws_smithy_runtime_api::client::interceptors::Intercept for QueryPrecedenceEndpointParamsInterceptor {
  228    293   
    fn name(&self) -> &'static str {
  229    294   
        "QueryPrecedenceEndpointParamsInterceptor"
  230    295   
    }
  231    296
@@ -381,446 +441,516 @@
  401    466   
            .expect("the config must have a deserializer");
  402    467   
  403    468   
        let parsed = de.deserialize_streaming(&mut http_response);
  404    469   
        let parsed = parsed.unwrap_or_else(|| {
  405    470   
            let http_response = http_response.map(|body| {
  406    471   
                ::aws_smithy_types::body::SdkBody::from(::bytes::Bytes::copy_from_slice(&::aws_smithy_protocol_test::decode_body_data(
  407    472   
                    body.bytes().unwrap(),
  408    473   
                    ::aws_smithy_protocol_test::MediaType::from("unknown"),
  409    474   
                )))
  410    475   
            });
  411         -
            de.deserialize_nonstreaming(&http_response)
         476  +
            // Build a config bag with the protocol for schema-based deserialization
         477  +
            #[allow(unused_mut)]
         478  +
            let mut test_cfg = ::aws_smithy_types::config_bag::ConfigBag::base();
         479  +
            {
         480  +
                let mut layer = ::aws_smithy_types::config_bag::Layer::new("test_protocol");
         481  +
                layer.store_put(::aws_smithy_schema::protocol::SharedClientProtocol::new(
         482  +
                    ::aws_smithy_json::protocol::aws_rest_json_1::AwsRestJsonProtocol::new(),
         483  +
                ));
         484  +
                test_cfg.push_shared_layer(layer.freeze());
         485  +
            }
         486  +
            de.deserialize_nonstreaming_with_config(&http_response, &test_cfg)
  412    487   
        });
  413    488   
        let parsed = parsed.expect_err("should be error response");
  414    489   
        let parsed: &crate::operation::query_precedence::QueryPrecedenceError =
  415    490   
            parsed.as_operation_error().expect("operation error").downcast_ref().unwrap();
  416    491   
        if let crate::operation::query_precedence::QueryPrecedenceError::ExtraError(parsed) = parsed {
  417    492   
            ::pretty_assertions::assert_eq!(parsed.message, expected_output.message, "Unexpected value for `message`");
  418    493   
        } else {
  419    494   
            panic!("wrong variant: Got: {:?}. Expected: {:?}", parsed, expected_output);
  420    495   
        }
  421    496   
    }

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

@@ -22,22 +151,153 @@
   42     42   
    ),
   43     43   
    ::aws_smithy_schema::ShapeType::Map,
   44     44   
    "baz",
   45     45   
    1,
   46     46   
)
   47     47   
.with_http_query_params();
   48     48   
static QUERYPRECEDENCEINPUT_SCHEMA: ::aws_smithy_schema::Schema = ::aws_smithy_schema::Schema::new_struct(
   49     49   
    QUERYPRECEDENCEINPUT_SCHEMA_ID,
   50     50   
    ::aws_smithy_schema::ShapeType::Structure,
   51     51   
    &[&QUERYPRECEDENCEINPUT_MEMBER_FOO, &QUERYPRECEDENCEINPUT_MEMBER_BAZ],
   52         -
);
          52  +
)
          53  +
.with_http(aws_smithy_schema::traits::HttpTrait::new("POST", "/Precedence", None));
   53     54   
impl QueryPrecedenceInput {
   54     55   
    /// The schema for this shape.
   55     56   
    pub const SCHEMA: &'static ::aws_smithy_schema::Schema = &QUERYPRECEDENCEINPUT_SCHEMA;
   56     57   
}
   57     58   
impl ::aws_smithy_schema::serde::SerializableStruct for QueryPrecedenceInput {
   58     59   
    #[allow(unused_variables, clippy::diverging_sub_expression)]
   59     60   
    fn serialize_members(
   60     61   
        &self,
   61     62   
        ser: &mut dyn ::aws_smithy_schema::serde::ShapeSerializer,
   62     63   
    ) -> ::std::result::Result<(), ::aws_smithy_schema::serde::SerdeError> {
   63     64   
        if let Some(ref val) = self.foo {
   64     65   
            ser.write_string(&QUERYPRECEDENCEINPUT_MEMBER_FOO, val)?;
   65     66   
        }
   66     67   
        if let Some(ref val) = self.baz {
   67     68   
            ser.write_map(
   68     69   
                &QUERYPRECEDENCEINPUT_MEMBER_BAZ,
   69     70   
                &|ser: &mut dyn ::aws_smithy_schema::serde::ShapeSerializer| {
   70     71   
                    for (key, value) in val {
   71     72   
                        ser.write_string(&::aws_smithy_schema::prelude::STRING, key)?;
   72     73   
                        ser.write_string(&::aws_smithy_schema::prelude::STRING, value)?;
   73     74   
                    }
   74     75   
                    Ok(())
   75     76   
                },
   76     77   
            )?;
   77     78   
        }
   78     79   
        Ok(())
   79     80   
    }
   80     81   
}
   81     82   
impl QueryPrecedenceInput {
   82     83   
    /// Deserializes this structure from a [`ShapeDeserializer`].
   83         -
    pub fn deserialize<D: ::aws_smithy_schema::serde::ShapeDeserializer>(
   84         -
        deserializer: &mut D,
          84  +
    pub fn deserialize(
          85  +
        deserializer: &mut dyn ::aws_smithy_schema::serde::ShapeDeserializer,
   85     86   
    ) -> ::std::result::Result<Self, ::aws_smithy_schema::serde::SerdeError> {
   86     87   
        #[allow(unused_variables, unused_mut)]
   87     88   
        let mut builder = Self::builder();
   88     89   
        #[allow(
   89     90   
            unused_variables,
   90     91   
            unreachable_code,
   91     92   
            clippy::single_match,
   92     93   
            clippy::match_single_binding,
   93     94   
            clippy::diverging_sub_expression
   94     95   
        )]
   95         -
        deserializer.read_struct(&QUERYPRECEDENCEINPUT_SCHEMA, (), |_, member, deser| {
          96  +
        deserializer.read_struct(&QUERYPRECEDENCEINPUT_SCHEMA, &mut |member, deser| {
   96     97   
            match member.member_index() {
   97     98   
                Some(0) => {
   98     99   
                    builder.foo = Some(deser.read_string(member)?);
   99    100   
                }
  100    101   
                Some(1) => {
  101         -
                    builder.baz = Some({
  102         -
                        let container = if let Some(cap) = deser.container_size() {
  103         -
                            std::collections::HashMap::with_capacity(cap)
  104         -
                        } else {
  105         -
                            std::collections::HashMap::new()
  106         -
                        };
  107         -
                        deser.read_map(member, container, |mut map, key, deser| {
  108         -
                            map.insert(key, deser.read_string(member)?);
  109         -
                            Ok(map)
  110         -
                        })?
  111         -
                    });
         102  +
                    builder.baz = Some(deser.read_string_string_map(member)?);
  112    103   
                }
  113    104   
                _ => {}
  114    105   
            }
  115    106   
            Ok(())
  116    107   
        })?;
  117    108   
        builder
  118    109   
            .build()
  119    110   
            .map_err(|e| aws_smithy_schema::serde::SerdeError::Custom { message: e.to_string() })
  120    111   
    }
  121    112   
}
         113  +
impl QueryPrecedenceInput {
         114  +
    /// Deserializes this structure from a body deserializer and HTTP response.
         115  +
    pub fn deserialize_with_response(
         116  +
        deserializer: &mut dyn ::aws_smithy_schema::serde::ShapeDeserializer,
         117  +
        _headers: &::aws_smithy_runtime_api::http::Headers,
         118  +
        _status: u16,
         119  +
        _body: &[u8],
         120  +
    ) -> ::std::result::Result<Self, ::aws_smithy_schema::serde::SerdeError> {
         121  +
        Self::deserialize(deserializer)
         122  +
    }
         123  +
}
  122    124   
impl QueryPrecedenceInput {
  123    125   
    /// Creates a new builder-style object to manufacture [`QueryPrecedenceInput`](crate::operation::query_precedence::QueryPrecedenceInput).
  124    126   
    pub fn builder() -> crate::operation::query_precedence::builders::QueryPrecedenceInputBuilder {
  125    127   
        crate::operation::query_precedence::builders::QueryPrecedenceInputBuilder::default()
  126    128   
    }
  127    129   
}
  128    130   
  129    131   
/// A builder for [`QueryPrecedenceInput`](crate::operation::query_precedence::QueryPrecedenceInput).
  130    132   
#[derive(::std::clone::Clone, ::std::cmp::PartialEq, ::std::default::Default, ::std::fmt::Debug)]
  131    133   
#[non_exhaustive]

tmp-codegen-diff/codegen-client-test/rest_json_extras/rust-client-codegen/src/operation/query_precedence/_query_precedence_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 QueryPrecedenceOutput {
   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(&QUERYPRECEDENCEOUTPUT_SCHEMA, (), |_, member, deser| {
          40  +
        deserializer.read_struct(&QUERYPRECEDENCEOUTPUT_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 QueryPrecedenceOutput {
          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 QueryPrecedenceOutput {
   50     61   
    /// Creates a new builder-style object to manufacture [`QueryPrecedenceOutput`](crate::operation::query_precedence::QueryPrecedenceOutput).
   51     62   
    pub fn builder() -> crate::operation::query_precedence::builders::QueryPrecedenceOutputBuilder {
   52     63   
        crate::operation::query_precedence::builders::QueryPrecedenceOutputBuilder::default()
   53     64   
    }
   54     65   
}
   55     66   
   56     67   
/// A builder for [`QueryPrecedenceOutput`](crate::operation::query_precedence::QueryPrecedenceOutput).
   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/status_response.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 `StatusResponse`.
    3      3   
#[derive(::std::clone::Clone, ::std::default::Default, ::std::fmt::Debug)]
    4      4   
#[non_exhaustive]
    5      5   
pub struct StatusResponse;
    6      6   
impl StatusResponse {
    7      7   
    /// Creates a new `StatusResponse`
    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::status_response::StatusResponseInput::SCHEMA;
          13  +
    /// The schema for this operation's output shape.
          14  +
    pub const OUTPUT_SCHEMA: &'static ::aws_smithy_schema::Schema = crate::operation::status_response::StatusResponseOutput::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::status_response::StatusResponseInput,
   14     18   
    ) -> ::std::result::Result<
   15     19   
        crate::operation::status_response::StatusResponseOutput,
   16     20   
        ::aws_smithy_runtime_api::client::result::SdkError<
   17     21   
            crate::operation::status_response::StatusResponseError,
   18     22   
            ::aws_smithy_runtime_api::client::orchestrator::HttpResponse,
   19     23   
        >,
   20     24   
    > {
@@ -104,108 +222,272 @@
  124    128   
                crate::operation::status_response::StatusResponseError,
  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 StatusResponseResponseDeserializer;
  133    137   
impl ::aws_smithy_runtime_api::client::ser_de::DeserializeResponse for StatusResponseResponseDeserializer {
  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());
         144  +
        #[allow(unused_mut)]
         145  +
        let mut force_error = false;
         146  +
         147  +
        if !success && status != 200 || force_error {
  139    148   
            let headers = response.headers();
  140    149   
            let body = response.body().bytes().expect("body loaded");
  141    150   
            #[allow(unused_mut)]
  142         -
        let mut force_error = false;
         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  +
            })?;
  143    154   
  144         -
        let parse_result = if !success && status != 200 || force_error {
  145         -
            crate::protocol_serde::shape_status_response::de_status_response_http_error(status, headers, body)
  146         -
        } else {
  147         -
            crate::protocol_serde::shape_status_response::de_status_response_http_response(status, headers, body)
         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::status_response::StatusResponseError::unhandled(
         161  +
                            generic,
         162  +
                        )),
         163  +
                    ))
         164  +
                }
  148    165   
            };
  149         -
        crate::protocol_serde::type_erase_result(parse_result)
         166  +
            let _error_message = generic.message().map(|msg| msg.to_owned());
         167  +
            let protocol = _cfg
         168  +
                .load::<::aws_smithy_schema::protocol::SharedClientProtocol>()
         169  +
                .expect("a SharedClientProtocol is required");
         170  +
            let err = match error_code {
         171  +
                "ExtraError" => crate::operation::status_response::StatusResponseError::ExtraError({
         172  +
                    let mut tmp = match protocol
         173  +
                        .deserialize_response(response, crate::types::error::ExtraError::SCHEMA, _cfg)
         174  +
                        .and_then(|mut deser| {
         175  +
                            crate::types::error::ExtraError::deserialize_with_response(
         176  +
                                &mut *deser,
         177  +
                                response.headers(),
         178  +
                                response.status().into(),
         179  +
                                body,
         180  +
                            )
         181  +
                        }) {
         182  +
                        ::std::result::Result::Ok(val) => val,
         183  +
                        ::std::result::Result::Err(e) => {
         184  +
                            return ::std::result::Result::Err(::aws_smithy_runtime_api::client::orchestrator::OrchestratorError::other(
         185  +
                                ::aws_smithy_runtime_api::box_error::BoxError::from(e),
         186  +
                            ))
         187  +
                        }
         188  +
                    };
         189  +
                    tmp.meta = generic;
         190  +
                    if tmp.message.is_none() {
         191  +
                        tmp.message = _error_message;
         192  +
                    }
         193  +
                    tmp
         194  +
                }),
         195  +
                _ => crate::operation::status_response::StatusResponseError::generic(generic),
         196  +
            };
         197  +
            ::std::result::Result::Err(::aws_smithy_runtime_api::client::orchestrator::OrchestratorError::operation(
         198  +
                ::aws_smithy_runtime_api::client::interceptors::context::Error::erase(err),
         199  +
            ))
         200  +
        } else {
         201  +
            let protocol = _cfg
         202  +
                .load::<::aws_smithy_schema::protocol::SharedClientProtocol>()
         203  +
                .expect("a SharedClientProtocol is required");
         204  +
            let mut deser = protocol
         205  +
                .deserialize_response(response, StatusResponse::OUTPUT_SCHEMA, _cfg)
         206  +
                .map_err(|e| {
         207  +
                    ::aws_smithy_runtime_api::client::orchestrator::OrchestratorError::other(::aws_smithy_runtime_api::box_error::BoxError::from(e))
         208  +
                })?;
         209  +
            let body = response.body().bytes().expect("body loaded");
         210  +
            let output = crate::operation::status_response::StatusResponseOutput::deserialize_with_response(
         211  +
                &mut *deser,
         212  +
                response.headers(),
         213  +
                response.status().into(),
         214  +
                body,
         215  +
            )
         216  +
            .map_err(|e| {
         217  +
                ::aws_smithy_runtime_api::client::orchestrator::OrchestratorError::other(::aws_smithy_runtime_api::box_error::BoxError::from(e))
         218  +
            })?;
         219  +
            ::std::result::Result::Ok(::aws_smithy_runtime_api::client::interceptors::context::Output::erase(output))
         220  +
        }
  150    221   
    }
  151    222   
}
  152    223   
#[derive(Debug)]
  153    224   
struct StatusResponseRequestSerializer;
  154    225   
impl ::aws_smithy_runtime_api::client::ser_de::SerializeRequest for StatusResponseRequestSerializer {
  155    226   
    #[allow(unused_mut, clippy::let_and_return, clippy::needless_borrow, clippy::useless_conversion)]
  156    227   
    fn serialize_input(
  157    228   
        &self,
  158    229   
        input: ::aws_smithy_runtime_api::client::interceptors::context::Input,
  159    230   
        _cfg: &mut ::aws_smithy_types::config_bag::ConfigBag,
  160    231   
    ) -> ::std::result::Result<::aws_smithy_runtime_api::client::orchestrator::HttpRequest, ::aws_smithy_runtime_api::box_error::BoxError> {
  161    232   
        let input = input
  162    233   
            .downcast::<crate::operation::status_response::StatusResponseInput>()
  163    234   
            .expect("correct type");
  164         -
        let _header_serialization_settings = _cfg
  165         -
            .load::<crate::serialization_settings::HeaderSerializationSettings>()
  166         -
            .cloned()
  167         -
            .unwrap_or_default();
  168         -
        let mut request_builder = {
  169         -
            #[allow(clippy::uninlined_format_args)]
  170         -
            fn uri_base(
  171         -
                _input: &crate::operation::status_response::StatusResponseInput,
  172         -
                output: &mut ::std::string::String,
  173         -
            ) -> ::std::result::Result<(), ::aws_smithy_types::error::operation::BuildError> {
  174         -
                use ::std::fmt::Write as _;
  175         -
                ::std::write!(output, "/").expect("formatting should succeed");
  176         -
                ::std::result::Result::Ok(())
  177         -
            }
  178         -
            #[allow(clippy::unnecessary_wraps)]
  179         -
            fn update_http_builder(
  180         -
                input: &crate::operation::status_response::StatusResponseInput,
  181         -
                builder: ::http_1x::request::Builder,
  182         -
            ) -> ::std::result::Result<::http_1x::request::Builder, ::aws_smithy_types::error::operation::BuildError> {
  183         -
                let mut uri = ::std::string::String::new();
  184         -
                uri_base(input, &mut uri)?;
  185         -
                ::std::result::Result::Ok(builder.method("POST").uri(uri))
  186         -
            }
  187         -
            let mut builder = update_http_builder(&input, ::http_1x::request::Builder::new())?;
  188         -
            builder
  189         -
        };
  190         -
        let body = ::aws_smithy_types::body::SdkBody::from("");
         235  +
        let protocol = _cfg
         236  +
            .load::<::aws_smithy_schema::protocol::SharedClientProtocol>()
         237  +
            .expect("a SharedClientProtocol is required");
         238  +
        let mut request = protocol
         239  +
            .serialize_request(&input, StatusResponse::INPUT_SCHEMA, "", _cfg)
         240  +
            .map_err(::aws_smithy_runtime_api::box_error::BoxError::from)?;
  191    241   
  192         -
        ::std::result::Result::Ok(request_builder.body(body).expect("valid request").try_into().unwrap())
         242  +
        return ::std::result::Result::Ok(request);
  193    243   
    }
  194    244   
}
  195    245   
#[derive(Debug)]
  196    246   
struct StatusResponseEndpointParamsInterceptor;
  197    247   
  198    248   
impl ::aws_smithy_runtime_api::client::interceptors::Intercept for StatusResponseEndpointParamsInterceptor {
  199    249   
    fn name(&self) -> &'static str {
  200    250   
        "StatusResponseEndpointParamsInterceptor"
  201    251   
    }
  202    252
@@ -235,285 +295,355 @@
  255    305   
            .expect("the config must have a deserializer");
  256    306   
  257    307   
        let parsed = de.deserialize_streaming(&mut http_response);
  258    308   
        let parsed = parsed.unwrap_or_else(|| {
  259    309   
            let http_response = http_response.map(|body| {
  260    310   
                ::aws_smithy_types::body::SdkBody::from(::bytes::Bytes::copy_from_slice(&::aws_smithy_protocol_test::decode_body_data(
  261    311   
                    body.bytes().unwrap(),
  262    312   
                    ::aws_smithy_protocol_test::MediaType::from("unknown"),
  263    313   
                )))
  264    314   
            });
  265         -
            de.deserialize_nonstreaming(&http_response)
         315  +
            // Build a config bag with the protocol for schema-based deserialization
         316  +
            #[allow(unused_mut)]
         317  +
            let mut test_cfg = ::aws_smithy_types::config_bag::ConfigBag::base();
         318  +
            {
         319  +
                let mut layer = ::aws_smithy_types::config_bag::Layer::new("test_protocol");
         320  +
                layer.store_put(::aws_smithy_schema::protocol::SharedClientProtocol::new(
         321  +
                    ::aws_smithy_json::protocol::aws_rest_json_1::AwsRestJsonProtocol::new(),
         322  +
                ));
         323  +
                test_cfg.push_shared_layer(layer.freeze());
         324  +
            }
         325  +
            de.deserialize_nonstreaming_with_config(&http_response, &test_cfg)
  266    326   
        });
  267    327   
        let parsed = parsed.expect_err("should be error response");
  268    328   
        let parsed: &crate::operation::status_response::StatusResponseError =
  269    329   
            parsed.as_operation_error().expect("operation error").downcast_ref().unwrap();
  270    330   
        if let crate::operation::status_response::StatusResponseError::ExtraError(parsed) = parsed {
  271    331   
            ::pretty_assertions::assert_eq!(parsed.message, expected_output.message, "Unexpected value for `message`");
  272    332   
        } else {
  273    333   
            panic!("wrong variant: Got: {:?}. Expected: {:?}", parsed, expected_output);
  274    334   
        }
  275    335   
    }