Client Test

Client Test

rev. ce76c569583a3fdee1e9d39e082a8d5efd7b13b7

Files changed:

tmp-codegen-diff/codegen-client-test/aws_query/rust-client-codegen/Cargo.toml

@@ -25,25 +62,65 @@
   45     45   
features = ["client", "test-util"]
   46     46   
[dev-dependencies.aws-smithy-types]
   47     47   
path = "/home/build/workspace/smithy-rs/rust-runtime/aws-smithy-types"
   48     48   
[dev-dependencies.bytes]
   49     49   
version = "1.4.0"
   50     50   
[dev-dependencies.pretty_assertions]
   51     51   
version = "1.3.0"
   52     52   
[dev-dependencies.tokio]
   53     53   
version = "1.23.1"
   54     54   
features = ["macros", "test-util", "rt-multi-thread"]
          55  +
[dev-dependencies.tracing-test]
          56  +
version = "0.2.5"
          57  +
features = ["no-env-filter"]
   55     58   
[features]
   56     59   
test-util = ["aws-smithy-runtime/test-util"]
   57     60   
rt-tokio = ["aws-smithy-async/rt-tokio", "aws-smithy-types/rt-tokio"]
   58     61   
behavior-version-latest = []
   59     62   
rustls = ["aws-smithy-runtime/tls-rustls"]
   60     63   
default = ["rt-tokio", "rustls"]
   61     64   
   62     65   

tmp-codegen-diff/codegen-client-test/aws_query/rust-client-codegen/src/operation/datetime_offsets.rs

@@ -190,190 +291,292 @@
  210    210   
        let params = crate::config::endpoint::Params::builder().build().map_err(|err| {
  211    211   
            ::aws_smithy_runtime_api::client::interceptors::error::ContextAttachedError::new("endpoint params could not be built", err)
  212    212   
        })?;
  213    213   
        cfg.interceptor_state()
  214    214   
            .store_put(::aws_smithy_runtime_api::client::endpoint::EndpointResolverParams::new(params));
  215    215   
        ::std::result::Result::Ok(())
  216    216   
    }
  217    217   
}
  218    218   
#[allow(unreachable_code, unused_variables)]
  219    219   
#[cfg(test)]
  220         -
mod datetime_offsets_request_test {
         220  +
mod datetime_offsets_test {
         221  +
  221    222   
    /// Ensures that clients can correctly parse datetime (timestamps) with offsets
  222    223   
    /// Test ID: AwsQueryDateTimeWithNegativeOffset
  223    224   
    #[::tokio::test]
  224         -
    #[allow(unused_mut)]
         225  +
    #[::tracing_test::traced_test]
  225    226   
    async fn aws_query_date_time_with_negative_offset_response() {
  226    227   
        let expected_output = crate::operation::datetime_offsets::DatetimeOffsetsOutput::builder()
  227    228   
            .set_datetime(::std::option::Option::Some(::aws_smithy_types::DateTime::from_fractional_secs(
  228    229   
                1576540098, 0_f64,
  229    230   
            )))
  230    231   
            .build();
  231    232   
        let mut http_response = ::aws_smithy_runtime_api::http::Response::try_from(::http::response::Builder::new()
  232    233   
        .header("Content-Type", "text/xml")
  233    234   
        .status(200)
  234    235   
                    .body(::aws_smithy_types::body::SdkBody::from("<DatetimeOffsetsResponse xmlns=\"https://example.com/\">\n    <DatetimeOffsetsResult>\n        <datetime>2019-12-16T22:48:18-01:00</datetime>\n    </DatetimeOffsetsResult>\n</DatetimeOffsetsResponse>\n"))
  235    236   
                    .unwrap()
  236    237   
                    ).unwrap();
  237    238   
        use ::aws_smithy_runtime_api::client::runtime_plugin::RuntimePlugin;
  238    239   
        use ::aws_smithy_runtime_api::client::ser_de::DeserializeResponse;
  239    240   
  240    241   
        let op = crate::operation::datetime_offsets::DatetimeOffsets::new();
  241    242   
        let config = op.config().expect("the operation has config");
  242    243   
        let de = config
  243    244   
            .load::<::aws_smithy_runtime_api::client::ser_de::SharedResponseDeserializer>()
  244    245   
            .expect("the config must have a deserializer");
  245    246   
  246    247   
        let parsed = de.deserialize_streaming(&mut http_response);
  247    248   
        let parsed = parsed.unwrap_or_else(|| {
  248    249   
            let http_response =
  249    250   
                http_response.map(|body| ::aws_smithy_types::body::SdkBody::from(::bytes::Bytes::copy_from_slice(body.bytes().unwrap())));
  250    251   
            de.deserialize_nonstreaming(&http_response)
  251    252   
        });
  252    253   
        let parsed = parsed
  253    254   
            .expect("should be successful response")
  254    255   
            .downcast::<crate::operation::datetime_offsets::DatetimeOffsetsOutput>()
  255    256   
            .unwrap();
  256    257   
        ::pretty_assertions::assert_eq!(parsed.datetime, expected_output.datetime, "Unexpected value for `datetime`");
  257    258   
    }
  258    259   
    /// Ensures that clients can correctly parse datetime (timestamps) with offsets
  259    260   
    /// Test ID: AwsQueryDateTimeWithPositiveOffset
  260    261   
    #[::tokio::test]
  261         -
    #[allow(unused_mut)]
         262  +
    #[::tracing_test::traced_test]
  262    263   
    async fn aws_query_date_time_with_positive_offset_response() {
  263    264   
        let expected_output = crate::operation::datetime_offsets::DatetimeOffsetsOutput::builder()
  264    265   
            .set_datetime(::std::option::Option::Some(::aws_smithy_types::DateTime::from_fractional_secs(
  265    266   
                1576540098, 0_f64,
  266    267   
            )))
  267    268   
            .build();
  268    269   
        let mut http_response = ::aws_smithy_runtime_api::http::Response::try_from(::http::response::Builder::new()
  269    270   
        .header("Content-Type", "text/xml")
  270    271   
        .status(200)
  271    272   
                    .body(::aws_smithy_types::body::SdkBody::from("<DatetimeOffsetsResponse xmlns=\"https://example.com/\">\n    <DatetimeOffsetsResult>\n        <datetime>2019-12-17T00:48:18+01:00</datetime>\n    </DatetimeOffsetsResult>\n</DatetimeOffsetsResponse>\n"))

tmp-codegen-diff/codegen-client-test/aws_query/rust-client-codegen/src/operation/empty_input_and_empty_output.rs

@@ -196,196 +284,285 @@
  216    216   
        let params = crate::config::endpoint::Params::builder().build().map_err(|err| {
  217    217   
            ::aws_smithy_runtime_api::client::interceptors::error::ContextAttachedError::new("endpoint params could not be built", err)
  218    218   
        })?;
  219    219   
        cfg.interceptor_state()
  220    220   
            .store_put(::aws_smithy_runtime_api::client::endpoint::EndpointResolverParams::new(params));
  221    221   
        ::std::result::Result::Ok(())
  222    222   
    }
  223    223   
}
  224    224   
#[allow(unreachable_code, unused_variables)]
  225    225   
#[cfg(test)]
  226         -
mod empty_input_and_empty_output_request_test {
         226  +
mod empty_input_and_empty_output_test {
         227  +
  227    228   
    /// Empty input serializes no extra query params
  228    229   
    /// Test ID: QueryEmptyInputAndEmptyOutput
  229    230   
    #[::tokio::test]
  230         -
    #[allow(unused_mut)]
         231  +
    #[::tracing_test::traced_test]
  231    232   
    async fn query_empty_input_and_empty_output_request() {
  232    233   
        let (http_client, request_receiver) = ::aws_smithy_runtime::client::http::test_util::capture_request(None);
  233    234   
        let config_builder = crate::config::Config::builder().with_test_defaults().endpoint_url("https://example.com");
  234    235   
  235    236   
        let client = crate::Client::from_conf(config_builder.http_client(http_client).build());
  236    237   
        let result = client.empty_input_and_empty_output().send().await;
  237    238   
        let _ = dbg!(result);
  238    239   
        let http_request = request_receiver.expect_request();
  239    240   
        let expected_headers = [("Content-Type", "application/x-www-form-urlencoded")];
  240    241   
        ::aws_smithy_protocol_test::assert_ok(::aws_smithy_protocol_test::validate_headers(http_request.headers(), expected_headers));
  241    242   
        let body = http_request.body().bytes().expect("body should be strict");
  242    243   
        ::aws_smithy_protocol_test::assert_ok(::aws_smithy_protocol_test::validate_body(
  243    244   
            body,
  244    245   
            "Action=EmptyInputAndEmptyOutput&Version=2020-01-08",
  245    246   
            ::aws_smithy_protocol_test::MediaType::from("application/x-www-form-urlencoded"),
  246    247   
        ));
  247    248   
        let uri: ::http::Uri = http_request.uri().parse().expect("invalid URI sent");
  248    249   
        ::pretty_assertions::assert_eq!(http_request.method(), "POST", "method was incorrect");
  249    250   
        ::pretty_assertions::assert_eq!(uri.path(), "/", "path was incorrect");
  250    251   
    }
  251    252   
    /// Empty output
  252    253   
    /// Test ID: QueryEmptyInputAndEmptyOutput
  253    254   
    #[::tokio::test]
  254         -
    #[allow(unused_mut)]
         255  +
    #[::tracing_test::traced_test]
  255    256   
    async fn query_empty_input_and_empty_output_response() {
  256    257   
        let expected_output = crate::operation::empty_input_and_empty_output::EmptyInputAndEmptyOutputOutput::builder().build();
  257    258   
        let mut http_response = ::aws_smithy_runtime_api::http::Response::try_from(
  258    259   
            ::http::response::Builder::new()
  259    260   
                .status(200)
  260    261   
                .body(::aws_smithy_types::body::SdkBody::from(vec![]))
  261    262   
                .unwrap(),
  262    263   
        )
  263    264   
        .unwrap();
  264    265   
        use ::aws_smithy_runtime_api::client::runtime_plugin::RuntimePlugin;

tmp-codegen-diff/codegen-client-test/aws_query/rust-client-codegen/src/operation/endpoint_operation.rs

@@ -195,195 +260,261 @@
  215    215   
        let params = crate::config::endpoint::Params::builder().build().map_err(|err| {
  216    216   
            ::aws_smithy_runtime_api::client::interceptors::error::ContextAttachedError::new("endpoint params could not be built", err)
  217    217   
        })?;
  218    218   
        cfg.interceptor_state()
  219    219   
            .store_put(::aws_smithy_runtime_api::client::endpoint::EndpointResolverParams::new(params));
  220    220   
        ::std::result::Result::Ok(())
  221    221   
    }
  222    222   
}
  223    223   
#[allow(unreachable_code, unused_variables)]
  224    224   
#[cfg(test)]
  225         -
mod endpoint_operation_request_test {
         225  +
mod endpoint_operation_test {
         226  +
  226    227   
    /// Operations can prepend to the given host if they define the
  227    228   
    /// endpoint trait.
  228    229   
    /// Test ID: AwsQueryEndpointTrait
  229    230   
    #[::tokio::test]
  230         -
    #[allow(unused_mut)]
         231  +
    #[::tracing_test::traced_test]
  231    232   
    async fn aws_query_endpoint_trait_request() {
  232    233   
        let (http_client, request_receiver) = ::aws_smithy_runtime::client::http::test_util::capture_request(None);
  233    234   
        let config_builder = crate::config::Config::builder().with_test_defaults().endpoint_url("https://example.com");
  234    235   
  235    236   
        let client = crate::Client::from_conf(config_builder.http_client(http_client).build());
  236    237   
        let result = client.endpoint_operation().send().await;
  237    238   
        let _ = dbg!(result);
  238    239   
        let http_request = request_receiver.expect_request();
  239    240   
        let expected_headers = [("Content-Type", "application/x-www-form-urlencoded")];
  240    241   
        ::aws_smithy_protocol_test::assert_ok(::aws_smithy_protocol_test::validate_headers(http_request.headers(), expected_headers));

tmp-codegen-diff/codegen-client-test/aws_query/rust-client-codegen/src/operation/endpoint_with_host_label_operation.rs

@@ -216,216 +282,283 @@
  236    236   
        let params = crate::config::endpoint::Params::builder().build().map_err(|err| {
  237    237   
            ::aws_smithy_runtime_api::client::interceptors::error::ContextAttachedError::new("endpoint params could not be built", err)
  238    238   
        })?;
  239    239   
        cfg.interceptor_state()
  240    240   
            .store_put(::aws_smithy_runtime_api::client::endpoint::EndpointResolverParams::new(params));
  241    241   
        ::std::result::Result::Ok(())
  242    242   
    }
  243    243   
}
  244    244   
#[allow(unreachable_code, unused_variables)]
  245    245   
#[cfg(test)]
  246         -
mod endpoint_with_host_label_operation_request_test {
         246  +
mod endpoint_with_host_label_operation_test {
         247  +
  247    248   
    /// Operations can prepend to the given host if they define the
  248    249   
    /// endpoint trait, and can use the host label trait to define
  249    250   
    /// further customization based on user input.
  250    251   
    /// Test ID: AwsQueryEndpointTraitWithHostLabel
  251    252   
    #[::tokio::test]
  252         -
    #[allow(unused_mut)]
         253  +
    #[::tracing_test::traced_test]
  253    254   
    async fn aws_query_endpoint_trait_with_host_label_request() {
  254    255   
        let (http_client, request_receiver) = ::aws_smithy_runtime::client::http::test_util::capture_request(None);
  255    256   
        let config_builder = crate::config::Config::builder().with_test_defaults().endpoint_url("https://example.com");
  256    257   
  257    258   
        let client = crate::Client::from_conf(config_builder.http_client(http_client).build());
  258    259   
        let result = client
  259    260   
            .endpoint_with_host_label_operation()
  260    261   
            .set_label(::std::option::Option::Some("bar".to_owned()))
  261    262   
            .send()
  262    263   
            .await;

tmp-codegen-diff/codegen-client-test/aws_query/rust-client-codegen/src/operation/flattened_xml_map.rs

@@ -190,190 +254,255 @@
  210    210   
        let params = crate::config::endpoint::Params::builder().build().map_err(|err| {
  211    211   
            ::aws_smithy_runtime_api::client::interceptors::error::ContextAttachedError::new("endpoint params could not be built", err)
  212    212   
        })?;
  213    213   
        cfg.interceptor_state()
  214    214   
            .store_put(::aws_smithy_runtime_api::client::endpoint::EndpointResolverParams::new(params));
  215    215   
        ::std::result::Result::Ok(())
  216    216   
    }
  217    217   
}
  218    218   
#[allow(unreachable_code, unused_variables)]
  219    219   
#[cfg(test)]
  220         -
mod flattened_xml_map_request_test {
         220  +
mod flattened_xml_map_test {
         221  +
  221    222   
    /// Serializes flattened XML maps in responses
  222    223   
    /// Test ID: QueryQueryFlattenedXmlMap
  223    224   
    #[::tokio::test]
  224         -
    #[allow(unused_mut)]
         225  +
    #[::tracing_test::traced_test]
  225    226   
    async fn query_query_flattened_xml_map_response() {
  226    227   
        let expected_output = crate::operation::flattened_xml_map::FlattenedXmlMapOutput::builder()
  227    228   
            .set_my_map(::std::option::Option::Some({
  228    229   
                let mut ret = ::std::collections::HashMap::new();
  229    230   
                ret.insert(
  230    231   
                    "foo".to_owned(),
  231    232   
                    "Foo".parse::<crate::types::FooEnum>().expect("static value validated to member"),
  232    233   
                );
  233    234   
                ret.insert(
  234    235   
                    "baz".to_owned(),

tmp-codegen-diff/codegen-client-test/aws_query/rust-client-codegen/src/operation/flattened_xml_map_with_xml_name.rs

@@ -196,196 +260,261 @@
  216    216   
        let params = crate::config::endpoint::Params::builder().build().map_err(|err| {
  217    217   
            ::aws_smithy_runtime_api::client::interceptors::error::ContextAttachedError::new("endpoint params could not be built", err)
  218    218   
        })?;
  219    219   
        cfg.interceptor_state()
  220    220   
            .store_put(::aws_smithy_runtime_api::client::endpoint::EndpointResolverParams::new(params));
  221    221   
        ::std::result::Result::Ok(())
  222    222   
    }
  223    223   
}
  224    224   
#[allow(unreachable_code, unused_variables)]
  225    225   
#[cfg(test)]
  226         -
mod flattened_xml_map_with_xml_name_request_test {
         226  +
mod flattened_xml_map_with_xml_name_test {
         227  +
  227    228   
    /// Serializes flattened XML maps in responses that have xmlName on members
  228    229   
    /// Test ID: QueryQueryFlattenedXmlMapWithXmlName
  229    230   
    #[::tokio::test]
  230         -
    #[allow(unused_mut)]
         231  +
    #[::tracing_test::traced_test]
  231    232   
    async fn query_query_flattened_xml_map_with_xml_name_response() {
  232    233   
        let expected_output = crate::operation::flattened_xml_map_with_xml_name::FlattenedXmlMapWithXmlNameOutput::builder()
  233    234   
            .set_my_map(::std::option::Option::Some({
  234    235   
                let mut ret = ::std::collections::HashMap::new();
  235    236   
                ret.insert("a".to_owned(), "A".to_owned());
  236    237   
                ret.insert("b".to_owned(), "B".to_owned());
  237    238   
                ret
  238    239   
            }))
  239    240   
            .build();
  240    241   
        let mut http_response = ::aws_smithy_runtime_api::http::Response::try_from(::http::response::Builder::new()

tmp-codegen-diff/codegen-client-test/aws_query/rust-client-codegen/src/operation/flattened_xml_map_with_xml_namespace.rs

@@ -202,202 +266,267 @@
  222    222   
        let params = crate::config::endpoint::Params::builder().build().map_err(|err| {
  223    223   
            ::aws_smithy_runtime_api::client::interceptors::error::ContextAttachedError::new("endpoint params could not be built", err)
  224    224   
        })?;
  225    225   
        cfg.interceptor_state()
  226    226   
            .store_put(::aws_smithy_runtime_api::client::endpoint::EndpointResolverParams::new(params));
  227    227   
        ::std::result::Result::Ok(())
  228    228   
    }
  229    229   
}
  230    230   
#[allow(unreachable_code, unused_variables)]
  231    231   
#[cfg(test)]
  232         -
mod flattened_xml_map_with_xml_namespace_request_test {
         232  +
mod flattened_xml_map_with_xml_namespace_test {
         233  +
  233    234   
    /// Serializes flattened XML maps in responses that have xmlNamespace and xmlName on members
  234    235   
    /// Test ID: QueryQueryFlattenedXmlMapWithXmlNamespace
  235    236   
    #[::tokio::test]
  236         -
    #[allow(unused_mut)]
         237  +
    #[::tracing_test::traced_test]
  237    238   
    async fn query_query_flattened_xml_map_with_xml_namespace_response() {
  238    239   
        let expected_output = crate::operation::flattened_xml_map_with_xml_namespace::FlattenedXmlMapWithXmlNamespaceOutput::builder()
  239    240   
            .set_my_map(::std::option::Option::Some({
  240    241   
                let mut ret = ::std::collections::HashMap::new();
  241    242   
                ret.insert("a".to_owned(), "A".to_owned());
  242    243   
                ret.insert("b".to_owned(), "B".to_owned());
  243    244   
                ret
  244    245   
            }))
  245    246   
            .build();
  246    247   
        let mut http_response = ::aws_smithy_runtime_api::http::Response::try_from(::http::response::Builder::new()

tmp-codegen-diff/codegen-client-test/aws_query/rust-client-codegen/src/operation/fractional_seconds.rs

@@ -190,190 +254,255 @@
  210    210   
        let params = crate::config::endpoint::Params::builder().build().map_err(|err| {
  211    211   
            ::aws_smithy_runtime_api::client::interceptors::error::ContextAttachedError::new("endpoint params could not be built", err)
  212    212   
        })?;
  213    213   
        cfg.interceptor_state()
  214    214   
            .store_put(::aws_smithy_runtime_api::client::endpoint::EndpointResolverParams::new(params));
  215    215   
        ::std::result::Result::Ok(())
  216    216   
    }
  217    217   
}
  218    218   
#[allow(unreachable_code, unused_variables)]
  219    219   
#[cfg(test)]
  220         -
mod fractional_seconds_request_test {
         220  +
mod fractional_seconds_test {
         221  +
  221    222   
    /// Ensures that clients can correctly parse datetime timestamps with fractional seconds
  222    223   
    /// Test ID: AwsQueryDateTimeWithFractionalSeconds
  223    224   
    #[::tokio::test]
  224         -
    #[allow(unused_mut)]
         225  +
    #[::tracing_test::traced_test]
  225    226   
    async fn aws_query_date_time_with_fractional_seconds_response() {
  226    227   
        let expected_output = crate::operation::fractional_seconds::FractionalSecondsOutput::builder()
  227    228   
            .set_datetime(::std::option::Option::Some(::aws_smithy_types::DateTime::from_fractional_secs(
  228    229   
                946845296, 0.123_f64,
  229    230   
            )))
  230    231   
            .build();
  231    232   
        let mut http_response = ::aws_smithy_runtime_api::http::Response::try_from(::http::response::Builder::new()
  232    233   
        .header("Content-Type", "text/xml")
  233    234   
        .status(200)
  234    235   
                    .body(::aws_smithy_types::body::SdkBody::from("<FractionalSecondsResponse xmlns=\"https://example.com/\">\n    <FractionalSecondsResult>\n        <datetime>2000-01-02T20:34:56.123Z</datetime>\n    </FractionalSecondsResult>\n</FractionalSecondsResponse>\n"))

tmp-codegen-diff/codegen-client-test/aws_query/rust-client-codegen/src/operation/greeting_with_errors.rs

@@ -190,190 +370,371 @@
  210    210   
        let params = crate::config::endpoint::Params::builder().build().map_err(|err| {
  211    211   
            ::aws_smithy_runtime_api::client::interceptors::error::ContextAttachedError::new("endpoint params could not be built", err)
  212    212   
        })?;
  213    213   
        cfg.interceptor_state()
  214    214   
            .store_put(::aws_smithy_runtime_api::client::endpoint::EndpointResolverParams::new(params));
  215    215   
        ::std::result::Result::Ok(())
  216    216   
    }
  217    217   
}
  218    218   
#[allow(unreachable_code, unused_variables)]
  219    219   
#[cfg(test)]
  220         -
mod greeting_with_errors_request_test {
         220  +
mod greeting_with_errors_test {
         221  +
  221    222   
    /// Ensures that operations with errors successfully know how to deserialize the successful response
  222    223   
    /// Test ID: QueryGreetingWithErrors
  223    224   
    #[::tokio::test]
  224         -
    #[allow(unused_mut)]
         225  +
    #[::tracing_test::traced_test]
  225    226   
    async fn query_greeting_with_errors_response() {
  226    227   
        let expected_output = crate::operation::greeting_with_errors::GreetingWithErrorsOutput::builder()
  227    228   
            .set_greeting(::std::option::Option::Some("Hello".to_owned()))
  228    229   
            .build();
  229    230   
        let mut http_response = ::aws_smithy_runtime_api::http::Response::try_from(::http::response::Builder::new()
  230    231   
        .header("Content-Type", "text/xml")
  231    232   
        .status(200)
  232    233   
                    .body(::aws_smithy_types::body::SdkBody::from("<GreetingWithErrorsResponse xmlns=\"https://example.com/\">\n    <GreetingWithErrorsResult>\n        <greeting>Hello</greeting>\n    </GreetingWithErrorsResult>\n</GreetingWithErrorsResponse>\n"))
  233    234   
                    .unwrap()
  234    235   
                    ).unwrap();
  235    236   
        use ::aws_smithy_runtime_api::client::runtime_plugin::RuntimePlugin;
  236    237   
        use ::aws_smithy_runtime_api::client::ser_de::DeserializeResponse;
  237    238   
  238    239   
        let op = crate::operation::greeting_with_errors::GreetingWithErrors::new();
  239    240   
        let config = op.config().expect("the operation has config");
  240    241   
        let de = config
  241    242   
            .load::<::aws_smithy_runtime_api::client::ser_de::SharedResponseDeserializer>()
  242    243   
            .expect("the config must have a deserializer");
  243    244   
  244    245   
        let parsed = de.deserialize_streaming(&mut http_response);
  245    246   
        let parsed = parsed.unwrap_or_else(|| {
  246    247   
            let http_response =
  247    248   
                http_response.map(|body| ::aws_smithy_types::body::SdkBody::from(::bytes::Bytes::copy_from_slice(body.bytes().unwrap())));
  248    249   
            de.deserialize_nonstreaming(&http_response)
  249    250   
        });
  250    251   
        let parsed = parsed
  251    252   
            .expect("should be successful response")
  252    253   
            .downcast::<crate::operation::greeting_with_errors::GreetingWithErrorsOutput>()
  253    254   
            .unwrap();
  254    255   
        ::pretty_assertions::assert_eq!(parsed.greeting, expected_output.greeting, "Unexpected value for `greeting`");
  255    256   
    }
  256    257   
    /// Parses simple XML errors
  257    258   
    /// Test ID: QueryInvalidGreetingError
  258    259   
    #[::tokio::test]
  259         -
    #[allow(unused_mut)]
         260  +
    #[::tracing_test::traced_test]
  260    261   
    async fn query_invalid_greeting_error_response() {
  261    262   
        let expected_output = crate::types::error::InvalidGreeting::builder()
  262    263   
            .set_message(::std::option::Option::Some("Hi".to_owned()))
  263    264   
            .build();
  264    265   
        let mut http_response = ::aws_smithy_runtime_api::http::Response::try_from(::http::response::Builder::new()
  265    266   
        .header("Content-Type", "text/xml")
  266    267   
        .status(400)
  267    268   
                    .body(::aws_smithy_types::body::SdkBody::from("<ErrorResponse>\n   <Error>\n      <Type>Sender</Type>\n      <Code>InvalidGreeting</Code>\n      <Message>Hi</Message>\n   </Error>\n   <RequestId>foo-id</RequestId>\n</ErrorResponse>\n"))
  268    269   
                    .unwrap()
  269    270   
                    ).unwrap();
  270    271   
        use ::aws_smithy_runtime_api::client::runtime_plugin::RuntimePlugin;
  271    272   
        use ::aws_smithy_runtime_api::client::ser_de::DeserializeResponse;
  272    273   
  273    274   
        let op = crate::operation::greeting_with_errors::GreetingWithErrors::new();
  274    275   
        let config = op.config().expect("the operation has config");
  275    276   
        let de = config
  276    277   
            .load::<::aws_smithy_runtime_api::client::ser_de::SharedResponseDeserializer>()
  277    278   
            .expect("the config must have a deserializer");
  278    279   
  279    280   
        let parsed = de.deserialize_streaming(&mut http_response);
  280    281   
        let parsed = parsed.unwrap_or_else(|| {
  281    282   
            let http_response =
  282    283   
                http_response.map(|body| ::aws_smithy_types::body::SdkBody::from(::bytes::Bytes::copy_from_slice(body.bytes().unwrap())));
  283    284   
            de.deserialize_nonstreaming(&http_response)
  284    285   
        });
  285    286   
        let parsed = parsed.expect_err("should be error response");
  286    287   
        let parsed: &crate::operation::greeting_with_errors::GreetingWithErrorsError =
  287    288   
            parsed.as_operation_error().expect("operation error").downcast_ref().unwrap();
  288    289   
        if let crate::operation::greeting_with_errors::GreetingWithErrorsError::InvalidGreeting(parsed) = parsed {
  289    290   
            ::pretty_assertions::assert_eq!(parsed.message, expected_output.message, "Unexpected value for `message`");
  290    291   
        } else {
  291    292   
            panic!("wrong variant: Got: {:?}. Expected: {:?}", parsed, expected_output);
  292    293   
        }
  293    294   
    }
  294    295   
    /// Test ID: QueryComplexError
  295    296   
    #[::tokio::test]
  296         -
    #[allow(unused_mut)]
         297  +
    #[::tracing_test::traced_test]
  297    298   
    async fn query_complex_error_response() {
  298    299   
        let expected_output = crate::types::error::ComplexError::builder()
  299    300   
            .set_top_level(::std::option::Option::Some("Top level".to_owned()))
  300    301   
            .set_nested(::std::option::Option::Some(
  301    302   
                crate::types::ComplexNestedErrorData::builder()
  302    303   
                    .set_foo(::std::option::Option::Some("bar".to_owned()))
  303    304   
                    .build(),
  304    305   
            ))
  305    306   
            .build();
  306    307   
        let mut http_response = ::aws_smithy_runtime_api::http::Response::try_from(::http::response::Builder::new()
  307    308   
        .header("Content-Type", "text/xml")
  308    309   
        .status(400)
  309    310   
                    .body(::aws_smithy_types::body::SdkBody::from("<ErrorResponse>\n   <Error>\n      <Type>Sender</Type>\n      <Code>ComplexError</Code>\n      <TopLevel>Top level</TopLevel>\n      <Nested>\n          <Foo>bar</Foo>\n      </Nested>\n   </Error>\n   <RequestId>foo-id</RequestId>\n</ErrorResponse>\n"))
  310    311   
                    .unwrap()
  311    312   
                    ).unwrap();
  312    313   
        use ::aws_smithy_runtime_api::client::runtime_plugin::RuntimePlugin;
  313    314   
        use ::aws_smithy_runtime_api::client::ser_de::DeserializeResponse;
  314    315   
  315    316   
        let op = crate::operation::greeting_with_errors::GreetingWithErrors::new();
  316    317   
        let config = op.config().expect("the operation has config");
  317    318   
        let de = config
  318    319   
            .load::<::aws_smithy_runtime_api::client::ser_de::SharedResponseDeserializer>()
  319    320   
            .expect("the config must have a deserializer");
  320    321   
  321    322   
        let parsed = de.deserialize_streaming(&mut http_response);
  322    323   
        let parsed = parsed.unwrap_or_else(|| {
  323    324   
            let http_response =
  324    325   
                http_response.map(|body| ::aws_smithy_types::body::SdkBody::from(::bytes::Bytes::copy_from_slice(body.bytes().unwrap())));
  325    326   
            de.deserialize_nonstreaming(&http_response)
  326    327   
        });
  327    328   
        let parsed = parsed.expect_err("should be error response");
  328    329   
        let parsed: &crate::operation::greeting_with_errors::GreetingWithErrorsError =
  329    330   
            parsed.as_operation_error().expect("operation error").downcast_ref().unwrap();
  330    331   
        if let crate::operation::greeting_with_errors::GreetingWithErrorsError::ComplexError(parsed) = parsed {
  331    332   
            ::pretty_assertions::assert_eq!(parsed.top_level, expected_output.top_level, "Unexpected value for `top_level`");
  332    333   
            ::pretty_assertions::assert_eq!(parsed.nested, expected_output.nested, "Unexpected value for `nested`");
  333    334   
        } else {
  334    335   
            panic!("wrong variant: Got: {:?}. Expected: {:?}", parsed, expected_output);
  335    336   
        }
  336    337   
    }
  337    338   
    /// Parses customized XML errors
  338    339   
    /// Test ID: QueryCustomizedError
  339    340   
    #[::tokio::test]
  340         -
    #[allow(unused_mut)]
         341  +
    #[::tracing_test::traced_test]
  341    342   
    async fn query_customized_error_response() {
  342    343   
        let expected_output = crate::types::error::CustomCodeError::builder()
  343    344   
            .set_message(::std::option::Option::Some("Hi".to_owned()))
  344    345   
            .build();
  345    346   
        let mut http_response = ::aws_smithy_runtime_api::http::Response::try_from(::http::response::Builder::new()
  346    347   
        .header("Content-Type", "text/xml")
  347    348   
        .status(402)
  348    349   
                    .body(::aws_smithy_types::body::SdkBody::from("<ErrorResponse>\n   <Error>\n      <Type>Sender</Type>\n      <Code>Customized</Code>\n      <Message>Hi</Message>\n   </Error>\n   <RequestId>foo-id</RequestId>\n</ErrorResponse>\n"))
  349    350   
                    .unwrap()
  350    351   
                    ).unwrap();

tmp-codegen-diff/codegen-client-test/aws_query/rust-client-codegen/src/operation/host_with_path_operation.rs

@@ -196,196 +260,261 @@
  216    216   
        let params = crate::config::endpoint::Params::builder().build().map_err(|err| {
  217    217   
            ::aws_smithy_runtime_api::client::interceptors::error::ContextAttachedError::new("endpoint params could not be built", err)
  218    218   
        })?;
  219    219   
        cfg.interceptor_state()
  220    220   
            .store_put(::aws_smithy_runtime_api::client::endpoint::EndpointResolverParams::new(params));
  221    221   
        ::std::result::Result::Ok(())
  222    222   
    }
  223    223   
}
  224    224   
#[allow(unreachable_code, unused_variables)]
  225    225   
#[cfg(test)]
  226         -
mod host_with_path_operation_request_test {
         226  +
mod host_with_path_operation_test {
         227  +
  227    228   
    /// Custom endpoints supplied by users can have paths
  228    229   
    /// Test ID: QueryHostWithPath
  229    230   
    #[::tokio::test]
  230         -
    #[allow(unused_mut)]
         231  +
    #[::tracing_test::traced_test]
  231    232   
    async fn query_host_with_path_request() {
  232    233   
        let (http_client, request_receiver) = ::aws_smithy_runtime::client::http::test_util::capture_request(None);
  233    234   
        let config_builder = crate::config::Config::builder()
  234    235   
            .with_test_defaults()
  235    236   
            .endpoint_url("https://example.com/custom");
  236    237   
  237    238   
        let client = crate::Client::from_conf(config_builder.http_client(http_client).build());
  238    239   
        let result = client.host_with_path_operation().send().await;
  239    240   
        let _ = dbg!(result);
  240    241   
        let http_request = request_receiver.expect_request();

tmp-codegen-diff/codegen-client-test/aws_query/rust-client-codegen/src/operation/ignores_wrapping_xml_name.rs

@@ -196,196 +260,261 @@
  216    216   
        let params = crate::config::endpoint::Params::builder().build().map_err(|err| {
  217    217   
            ::aws_smithy_runtime_api::client::interceptors::error::ContextAttachedError::new("endpoint params could not be built", err)
  218    218   
        })?;
  219    219   
        cfg.interceptor_state()
  220    220   
            .store_put(::aws_smithy_runtime_api::client::endpoint::EndpointResolverParams::new(params));
  221    221   
        ::std::result::Result::Ok(())
  222    222   
    }
  223    223   
}
  224    224   
#[allow(unreachable_code, unused_variables)]
  225    225   
#[cfg(test)]
  226         -
mod ignores_wrapping_xml_name_request_test {
         226  +
mod ignores_wrapping_xml_name_test {
         227  +
  227    228   
    /// The xmlName trait on the output structure is ignored in AWS Query
  228    229   
    /// Test ID: QueryIgnoresWrappingXmlName
  229    230   
    #[::tokio::test]
  230         -
    #[allow(unused_mut)]
         231  +
    #[::tracing_test::traced_test]
  231    232   
    async fn query_ignores_wrapping_xml_name_response() {
  232    233   
        let expected_output = crate::operation::ignores_wrapping_xml_name::IgnoresWrappingXmlNameOutput::builder()
  233    234   
            .set_foo(::std::option::Option::Some("bar".to_owned()))
  234    235   
            .build();
  235    236   
        let mut http_response = ::aws_smithy_runtime_api::http::Response::try_from(::http::response::Builder::new()
  236    237   
        .header("Content-Type", "text/xml")
  237    238   
        .status(200)
  238    239   
                    .body(::aws_smithy_types::body::SdkBody::from("<IgnoresWrappingXmlNameResponse xmlns=\"https://example.com/\">\n    <IgnoresWrappingXmlNameResult>\n        <foo>bar</foo>\n    </IgnoresWrappingXmlNameResult>\n</IgnoresWrappingXmlNameResponse>\n"))
  239    240   
                    .unwrap()
  240    241   
                    ).unwrap();

tmp-codegen-diff/codegen-client-test/aws_query/rust-client-codegen/src/operation/nested_structures.rs

@@ -193,193 +257,258 @@
  213    213   
        let params = crate::config::endpoint::Params::builder().build().map_err(|err| {
  214    214   
            ::aws_smithy_runtime_api::client::interceptors::error::ContextAttachedError::new("endpoint params could not be built", err)
  215    215   
        })?;
  216    216   
        cfg.interceptor_state()
  217    217   
            .store_put(::aws_smithy_runtime_api::client::endpoint::EndpointResolverParams::new(params));
  218    218   
        ::std::result::Result::Ok(())
  219    219   
    }
  220    220   
}
  221    221   
#[allow(unreachable_code, unused_variables)]
  222    222   
#[cfg(test)]
  223         -
mod nested_structures_request_test {
         223  +
mod nested_structures_test {
         224  +
  224    225   
    /// Serializes nested structures using dots
  225    226   
    /// Test ID: NestedStructures
  226    227   
    #[::tokio::test]
  227         -
    #[allow(unused_mut)]
         228  +
    #[::tracing_test::traced_test]
  228    229   
    async fn nested_structures_request() {
  229    230   
        let (http_client, request_receiver) = ::aws_smithy_runtime::client::http::test_util::capture_request(None);
  230    231   
        let config_builder = crate::config::Config::builder().with_test_defaults().endpoint_url("https://example.com");
  231    232   
  232    233   
        let client = crate::Client::from_conf(config_builder.http_client(http_client).build());
  233    234   
        let result = client
  234    235   
            .nested_structures()
  235    236   
            .set_nested(::std::option::Option::Some(
  236    237   
                crate::types::StructArg::builder()
  237    238   
                    .set_string_arg(::std::option::Option::Some("foo".to_owned()))

tmp-codegen-diff/codegen-client-test/aws_query/rust-client-codegen/src/operation/no_input_and_no_output.rs

@@ -190,190 +278,279 @@
  210    210   
        let params = crate::config::endpoint::Params::builder().build().map_err(|err| {
  211    211   
            ::aws_smithy_runtime_api::client::interceptors::error::ContextAttachedError::new("endpoint params could not be built", err)
  212    212   
        })?;
  213    213   
        cfg.interceptor_state()
  214    214   
            .store_put(::aws_smithy_runtime_api::client::endpoint::EndpointResolverParams::new(params));
  215    215   
        ::std::result::Result::Ok(())
  216    216   
    }
  217    217   
}
  218    218   
#[allow(unreachable_code, unused_variables)]
  219    219   
#[cfg(test)]
  220         -
mod no_input_and_no_output_request_test {
         220  +
mod no_input_and_no_output_test {
         221  +
  221    222   
    /// No input serializes no additional query params
  222    223   
    /// Test ID: QueryNoInputAndNoOutput
  223    224   
    #[::tokio::test]
  224         -
    #[allow(unused_mut)]
         225  +
    #[::tracing_test::traced_test]
  225    226   
    async fn query_no_input_and_no_output_request() {
  226    227   
        let (http_client, request_receiver) = ::aws_smithy_runtime::client::http::test_util::capture_request(None);
  227    228   
        let config_builder = crate::config::Config::builder().with_test_defaults().endpoint_url("https://example.com");
  228    229   
  229    230   
        let client = crate::Client::from_conf(config_builder.http_client(http_client).build());
  230    231   
        let result = client.no_input_and_no_output().send().await;
  231    232   
        let _ = dbg!(result);
  232    233   
        let http_request = request_receiver.expect_request();
  233    234   
        let expected_headers = [("Content-Type", "application/x-www-form-urlencoded")];
  234    235   
        ::aws_smithy_protocol_test::assert_ok(::aws_smithy_protocol_test::validate_headers(http_request.headers(), expected_headers));
  235    236   
        let body = http_request.body().bytes().expect("body should be strict");
  236    237   
        ::aws_smithy_protocol_test::assert_ok(::aws_smithy_protocol_test::validate_body(
  237    238   
            body,
  238    239   
            "Action=NoInputAndNoOutput&Version=2020-01-08",
  239    240   
            ::aws_smithy_protocol_test::MediaType::from("application/x-www-form-urlencoded"),
  240    241   
        ));
  241    242   
        let uri: ::http::Uri = http_request.uri().parse().expect("invalid URI sent");
  242    243   
        ::pretty_assertions::assert_eq!(http_request.method(), "POST", "method was incorrect");
  243    244   
        ::pretty_assertions::assert_eq!(uri.path(), "/", "path was incorrect");
  244    245   
    }
  245    246   
    /// Empty output. Note that no assertion is made on the output body itself.
  246    247   
    /// Test ID: QueryNoInputAndNoOutput
  247    248   
    #[::tokio::test]
  248         -
    #[allow(unused_mut)]
         249  +
    #[::tracing_test::traced_test]
  249    250   
    async fn query_no_input_and_no_output_response() {
  250    251   
        let expected_output = crate::operation::no_input_and_no_output::NoInputAndNoOutputOutput::builder().build();
  251    252   
        let mut http_response = ::aws_smithy_runtime_api::http::Response::try_from(
  252    253   
            ::http::response::Builder::new()
  253    254   
                .status(200)
  254    255   
                .body(::aws_smithy_types::body::SdkBody::from(vec![]))
  255    256   
                .unwrap(),
  256    257   
        )
  257    258   
        .unwrap();
  258    259   
        use ::aws_smithy_runtime_api::client::runtime_plugin::RuntimePlugin;

tmp-codegen-diff/codegen-client-test/aws_query/rust-client-codegen/src/operation/no_input_and_output.rs

@@ -190,190 +278,279 @@
  210    210   
        let params = crate::config::endpoint::Params::builder().build().map_err(|err| {
  211    211   
            ::aws_smithy_runtime_api::client::interceptors::error::ContextAttachedError::new("endpoint params could not be built", err)
  212    212   
        })?;
  213    213   
        cfg.interceptor_state()
  214    214   
            .store_put(::aws_smithy_runtime_api::client::endpoint::EndpointResolverParams::new(params));
  215    215   
        ::std::result::Result::Ok(())
  216    216   
    }
  217    217   
}
  218    218   
#[allow(unreachable_code, unused_variables)]
  219    219   
#[cfg(test)]
  220         -
mod no_input_and_output_request_test {
         220  +
mod no_input_and_output_test {
         221  +
  221    222   
    /// No input serializes no payload
  222    223   
    /// Test ID: QueryNoInputAndOutput
  223    224   
    #[::tokio::test]
  224         -
    #[allow(unused_mut)]
         225  +
    #[::tracing_test::traced_test]
  225    226   
    async fn query_no_input_and_output_request() {
  226    227   
        let (http_client, request_receiver) = ::aws_smithy_runtime::client::http::test_util::capture_request(None);
  227    228   
        let config_builder = crate::config::Config::builder().with_test_defaults().endpoint_url("https://example.com");
  228    229   
  229    230   
        let client = crate::Client::from_conf(config_builder.http_client(http_client).build());
  230    231   
        let result = client.no_input_and_output().send().await;
  231    232   
        let _ = dbg!(result);
  232    233   
        let http_request = request_receiver.expect_request();
  233    234   
        let expected_headers = [("Content-Type", "application/x-www-form-urlencoded")];
  234    235   
        ::aws_smithy_protocol_test::assert_ok(::aws_smithy_protocol_test::validate_headers(http_request.headers(), expected_headers));
  235    236   
        let body = http_request.body().bytes().expect("body should be strict");
  236    237   
        ::aws_smithy_protocol_test::assert_ok(::aws_smithy_protocol_test::validate_body(
  237    238   
            body,
  238    239   
            "Action=NoInputAndOutput&Version=2020-01-08",
  239    240   
            ::aws_smithy_protocol_test::MediaType::from("application/x-www-form-urlencoded"),
  240    241   
        ));
  241    242   
        let uri: ::http::Uri = http_request.uri().parse().expect("invalid URI sent");
  242    243   
        ::pretty_assertions::assert_eq!(http_request.method(), "POST", "method was incorrect");
  243    244   
        ::pretty_assertions::assert_eq!(uri.path(), "/", "path was incorrect");
  244    245   
    }
  245    246   
    /// Empty output
  246    247   
    /// Test ID: QueryNoInputAndOutput
  247    248   
    #[::tokio::test]
  248         -
    #[allow(unused_mut)]
         249  +
    #[::tracing_test::traced_test]
  249    250   
    async fn query_no_input_and_output_response() {
  250    251   
        let expected_output = crate::operation::no_input_and_output::NoInputAndOutputOutput::builder().build();
  251    252   
        let mut http_response = ::aws_smithy_runtime_api::http::Response::try_from(
  252    253   
            ::http::response::Builder::new()
  253    254   
                .status(200)
  254    255   
                .body(::aws_smithy_types::body::SdkBody::from(vec![]))
  255    256   
                .unwrap(),
  256    257   
        )
  257    258   
        .unwrap();
  258    259   
        use ::aws_smithy_runtime_api::client::runtime_plugin::RuntimePlugin;