Rust S3 unit tests (#4299)

* Rust S3 unit tests

* Finish delete test.

Author: DavidSouther

rust_dev_preview/s3/Cargo.toml

@@ -26,8 +26,12 @@ aws-config = { git = "https://github.com/awslabs/aws-sdk-rust", branch = "main"
 aws-endpoint = { git = "https://github.com/awslabs/aws-sdk-rust", branch = "main" }
 # snippet-end:[s3.rust.s3-object-lambda-cargo.toml]
 aws-sdk-s3 = { git = "https://github.com/awslabs/aws-sdk-rust", branch = "main" }
+aws-smithy-client = { git = "https://github.com/awslabs/aws-sdk-rust", branch = "main", features = [
+  "test-util",
+] }
 aws-smithy-http = { git = "https://github.com/awslabs/aws-sdk-rust", branch = "main", features = ["rt-tokio"] }
 aws-smithy-types = { git = "https://github.com/awslabs/aws-sdk-rust", branch = "main" }
+sdk-examples-test-utils = { path = "../test-utils" }
 bytes = "0.4.12"
 futures-util = { version = "0.3.21", features = ["alloc"] }
 http = "0.2.8"

rust_dev_preview/s3/src/bin/copy-object.rs

@@ -4,7 +4,10 @@
  */
 
 use aws_config::meta::region::RegionProviderChain;
-use aws_sdk_s3::{Client, Error, Region, PKG_VERSION};
+use aws_sdk_s3::{
+    error::CopyObjectError, output::CopyObjectOutput, types::SdkError, Client, Error, Region,
+    PKG_VERSION,
+};
 use structopt::StructOpt;
 
 #[derive(Debug, StructOpt)]
@@ -42,7 +45,7 @@ async fn cp_object(
     destination_bucket: &str,
     source_object: &str,
     destination_object: &str,
-) -> Result<(), Error> {
+) -> Result<CopyObjectOutput, SdkError<CopyObjectError>> {
     // Create source of object:
     //   source-bucket-name/source-object-name
     let mut source_bucket_and_object: String = "".to_owned();
@@ -56,14 +59,37 @@ async fn cp_object(
         .bucket(destination_bucket)
         .key(destination_object)
         .send()
-        .await?;
-
-    println!("Object copied.");
-
-    Ok(())
+        .await
 }
 // snippet-end:[bin.rust.copy-object]
 
+#[cfg(test)]
+mod test_cp_object {
+    use sdk_examples_test_utils::single_shot_client;
+
+    use crate::cp_object;
+
+    #[tokio::test]
+    async fn test_cp_object() {
+        let client = single_shot_client! {
+            sdk: aws_sdk_s3,
+            status: 200,
+            response: r#""#
+        };
+
+        let response = cp_object(
+            &client,
+            "source_bucket",
+            "destination_bucket",
+            "source_object",
+            "destination_object",
+        )
+        .await;
+
+        assert!(response.is_ok(), "{response:?}");
+    }
+}
+
 /// Copies an object from one Amazon S3 bucket to another.
 /// # Arguments
 ///
@@ -113,5 +139,9 @@ async fn main() -> Result<(), Error> {
     let shared_config = aws_config::from_env().region(region_provider).load().await;
     let client = Client::new(&shared_config);
 
-    cp_object(&client, &source, &destination, &key, &new_name).await
+    cp_object(&client, &source, &destination, &key, &new_name).await?;
+
+    println!("Object copied.");
+
+    Ok(())
 }

rust_dev_preview/s3/src/bin/create-bucket.rs

@@ -4,7 +4,10 @@
  */
 
 use aws_config::meta::region::RegionProviderChain;
+use aws_sdk_s3::error::CreateBucketError;
 use aws_sdk_s3::model::{BucketLocationConstraint, CreateBucketConfiguration};
+use aws_sdk_s3::output::CreateBucketOutput;
+use aws_sdk_s3::types::SdkError;
 use aws_sdk_s3::{Client, Error, Region, PKG_VERSION};
 use structopt::StructOpt;
 
@@ -25,7 +28,11 @@ struct Opt {
 
 // Creates a bucket.
 // snippet-start:[s3.rust.create-bucket]
-async fn make_bucket(client: &Client, bucket: &str, region: &str) -> Result<(), Error> {
+async fn make_bucket(
+    client: &Client,
+    bucket: &str,
+    region: &str,
+) -> Result<CreateBucketOutput, SdkError<CreateBucketError>> {
     let constraint = BucketLocationConstraint::from(region);
     let cfg = CreateBucketConfiguration::builder()
         .location_constraint(constraint)
@@ -36,13 +43,33 @@ async fn make_bucket(client: &Client, bucket: &str, region: &str) -> Result<(),
         .create_bucket_configuration(cfg)
         .bucket(bucket)
         .send()
-        .await?;
-    println!("Created bucket.");
-
-    Ok(())
+        .await
 }
 // snippet-end:[s3.rust.create-bucket]
 
+#[cfg(test)]
+mod test_make_bucket {
+    use sdk_examples_test_utils::single_shot_client;
+
+    use crate::make_bucket;
+
+    #[tokio::test]
+    async fn test_make_bucket() {
+        let client = single_shot_client! {
+            sdk: aws_sdk_s3,
+            status: 200,
+            headers: vec![("Location", "/test_bucket")],
+            response: r#""#
+        };
+
+        let res = make_bucket(&client, "bucket", "region").await;
+
+        assert!(res.is_ok(), "{res:?}");
+
+        assert_eq!(res.unwrap().location(), Some("/test_bucket"))
+    }
+}
+
 /// Creates an Amazon S3 bucket in the Region.
 /// # Arguments
 ///
@@ -83,5 +110,8 @@ async fn main() -> Result<(), Error> {
     let shared_config = aws_config::from_env().region(region_provider).load().await;
     let client = Client::new(&shared_config);
 
-    make_bucket(&client, &bucket, r_str).await
+    make_bucket(&client, &bucket, r_str).await?;
+    println!("Created bucket.");
+
+    Ok(())
 }

rust_dev_preview/s3/src/bin/s3-multipart-upload.rs

@@ -144,7 +144,7 @@ async fn run_example() -> Result<(), Error> {
         .unwrap();
     // snippet-end:[rust.example_code.s3.complete_multipart_upload]
 
-    let data: GetObjectOutput = s3_service::download_object(&client, &bucket_name, &key).await;
+    let data: GetObjectOutput = s3_service::download_object(&client, &bucket_name, &key).await?;
     let data_length: u64 = data.content_length().try_into().unwrap();
     if file.metadata().unwrap().len() == data_length {
         println!("Data lengths match.");