From 74fea521244f50f74dff143fbad5cb21f79a73c0 Mon Sep 17 00:00:00 2001
From: Arun Jangra <arunjangra1001@gmail.com>
Date: Fri, 26 Jul 2024 18:26:03 +0530
Subject: [PATCH] update : test_put_and_get_data_s3 test

---
 crates/orchestrator/src/tests/data_storage/mod.rs | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/crates/orchestrator/src/tests/data_storage/mod.rs b/crates/orchestrator/src/tests/data_storage/mod.rs
index 1fdb73f3..cf67a001 100644
--- a/crates/orchestrator/src/tests/data_storage/mod.rs
+++ b/crates/orchestrator/src/tests/data_storage/mod.rs
@@ -5,6 +5,7 @@ use crate::data_storage::{DataStorage, DataStorageConfig};
 use bytes::Bytes;
 use rstest::rstest;
 use serde_json::json;
+use utils::env_utils::get_env_var_or_panic;
 
 #[rstest]
 #[tokio::test]
@@ -15,6 +16,7 @@ async fn test_put_and_get_data_s3() -> color_eyre::Result<()> {
 
     let config = AWSS3Config::new_from_env();
     let s3_client = AWSS3::new(config).await;
+    s3_client.build_test_bucket(&get_env_var_or_panic("AWS_S3_BUCKET_NAME")).await.unwrap();
 
     let mock_data = json!(
         {