From e66378d2544f9e39acfac18143cfac4afa8ae7ac Mon Sep 17 00:00:00 2001
From: Arun Raghavan <arun@asymptotic.io>
Date: Tue, 18 Oct 2022 10:18:03 +0200
Subject: [PATCH] aws: Add a test for s3src/s3sink

This does rely on AWS credentials being provided in the environment, but
the test will be ignored if those are missing.
---
 net/aws/Cargo.toml  |  7 +++++
 net/aws/tests/s3.rs | 77 +++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 84 insertions(+)
 create mode 100644 net/aws/tests/s3.rs

diff --git a/net/aws/Cargo.toml b/net/aws/Cargo.toml
index a094a7a1..7f349a41 100644
--- a/net/aws/Cargo.toml
+++ b/net/aws/Cargo.toml
@@ -41,6 +41,13 @@ base32 = "0.4"
 backoff = { version = "0.4", features = [ "futures", "tokio" ] }
 gio = { git = "https://github.com/gtk-rs/gtk-rs-core.git", package = "gio" }
 
+[dev-dependencies]
+chrono = { version = "0.4", features = [ "alloc" ] }
+env_logger = "0.9"
+gst-check = { package = "gstreamer-check", git = "https://gitlab.freedesktop.org/gstreamer/gstreamer-rs", features = ["v1_18"] }
+rand = "0.8"
+test-with = { version = "0.8", default-features = false }
+
 [lib]
 name = "gstaws"
 crate-type = ["cdylib", "rlib"]
diff --git a/net/aws/tests/s3.rs b/net/aws/tests/s3.rs
new file mode 100644
index 00000000..511295ae
--- /dev/null
+++ b/net/aws/tests/s3.rs
@@ -0,0 +1,77 @@
+// Copyright (C) 2022, Asymptotic Inc.
+//      Author: Arun Raghavan <arun@asymptotic.io>
+//
+// This Source Code Form is subject to the terms of the Mozilla Public License, v2.0.
+// If a copy of the MPL was not distributed with this file, You can obtain one at
+// <https://mozilla.org/MPL/2.0/>.
+//
+// SPDX-License-Identifier: MPL-2.0
+//
+
+use gst::prelude::*;
+
+const DEFAULT_S3_REGION: &str = "us-west-2";
+
+fn init() {
+    use std::sync::Once;
+    static INIT: Once = Once::new();
+
+    INIT.call_once(|| {
+        gst::init().unwrap();
+        gstaws::plugin_register_static().unwrap();
+    });
+}
+
+#[test_with::env(AWS_ACCESS_KEY_ID)]
+#[test_with::env(AWS_SECRET_ACCESS_KEY)]
+#[tokio::test]
+async fn test_s3() {
+    init();
+    // Makes it easier to get AWS SDK logs if needed
+    env_logger::init();
+
+    let region = std::env::var("AWS_REGION").unwrap_or_else(|_| DEFAULT_S3_REGION.to_string());
+    let bucket =
+        std::env::var("AWS_S3_BUCKET").unwrap_or_else(|_| "gst-plugins-rs-tests".to_string());
+    let key = format!("s3-test-{:?}.txt", chrono::Utc::now());
+    let uri = format!("s3://{}/{}/{}", region, bucket, key);
+    let content = "Hello, world!\n".as_bytes();
+
+    // Manually add the element so we can configure it before it goes to PLAYING
+    let mut h1 = gst_check::Harness::new_empty();
+    // Need to add_parse() because the Harness API / Rust bindings aren't conducive to creating and
+    // adding an element manually
+    h1.add_parse(format!("awss3sink uri={}", uri).as_str());
+
+    h1.set_src_caps(gst::Caps::builder("text/plain").build());
+    h1.play();
+
+    h1.push(gst::Buffer::from_slice(content)).unwrap();
+    h1.push_event(gst::event::Eos::new());
+
+    let mut h2 = gst_check::Harness::new("awss3src");
+    h2.element().unwrap().set_property("uri", uri.clone());
+    h2.play();
+
+    let buf = h2.pull_until_eos().unwrap().unwrap();
+    assert_eq!(
+        content,
+        buf.into_mapped_buffer_readable().unwrap().as_slice()
+    );
+
+    let region_provider = aws_config::meta::region::RegionProviderChain::first_try(
+        aws_sdk_s3::Region::new(region.clone()),
+    )
+    .or_default_provider();
+
+    let config = aws_config::from_env().region(region_provider).load().await;
+    let client = aws_sdk_s3::Client::new(&config);
+
+    client
+        .delete_object()
+        .bucket(bucket)
+        .key(key)
+        .send()
+        .await
+        .unwrap();
+}