use s3ql::{bucket::*, read::*, region, s3_client, transact::*};
use std::collections::HashMap;
fn client() -> rusoto_s3::S3Client {
let region = region("us-east-1".to_owned(), "http://localhost:4566".to_owned());
s3_client(region)
}
pub const BUCKET: &'static str = "transactObjectsBucket";
pub const UPDATE_BUCKET: &'static str = "updateTransactObjectsBucket";
#[tokio::test]
async fn insert_object() {
let s3 = client();
let bucket = s3.create_s3_bucket(BUCKET.to_string(), None).await;
assert!(bucket.is_ok());
let mut map = HashMap::new();
map.insert("tx-time".to_string(), "2007-19-01T11:12:00-000".to_string());
let body = "{\"hello\": \"world\"}";
let insert = s3.insert_s3_object(
BUCKET.to_string(),
Some(map),
"key".to_string(),
Some(body.to_string()),
None,
);
assert!(insert.await.is_ok());
assert!(s3
.has_s3_object(BUCKET.to_string(), "key".to_string(), None, None, None)
.await
.is_ok());
}
#[tokio::test]
async fn update_object_body() {
let s3 = client();
let bucket = s3.create_s3_bucket(BUCKET.to_string(), None).await;
assert!(bucket.is_ok());
let body = "{\"hello\": \"world\"}";
let insert = s3.insert_s3_object(
BUCKET.to_string(),
None,
"key".to_string(),
Some(body.to_string()),
None,
);
assert!(insert.await.is_ok());
assert!(s3
.has_s3_object(BUCKET.to_string(), "key".to_string(), None, None, None)
.await
.is_ok());
let read_obj = s3
.read_s3_object_body(BUCKET.to_string(), "key".to_string(), None, None, None)
.await;
assert!(read_obj.is_some());
assert_eq!(read_obj.unwrap(), "{\"hello\": \"world\"}");
let update = s3.update_s3_object_body(
BUCKET.to_string(),
"key".to_string(),
"this is a new body".to_string(),
None,
);
assert!(update.await.is_ok());
let read_obj = s3
.read_s3_object_body(BUCKET.to_string(), "key".to_string(), None, None, None)
.await;
assert!(read_obj.is_some());
assert_eq!(read_obj.unwrap(), "this is a new body");
}
#[tokio::test]
async fn update_object_meta() {
let s3 = client();
let bucket = s3.create_s3_bucket(UPDATE_BUCKET.to_string(), None).await;
assert!(bucket.is_ok());
let insert = s3.insert_s3_object(
UPDATE_BUCKET.to_string(),
None,
"meta-key".to_string(),
None,
None,
);
assert!(insert.await.is_ok());
assert!(s3
.has_s3_object(
UPDATE_BUCKET.to_string(),
"meta-key".to_string(),
None,
None,
None
)
.await
.is_ok());
let read_obj = s3
.read_s3_object(
UPDATE_BUCKET.to_string(),
"meta-key".to_string(),
None,
None,
None,
)
.await;
assert!(read_obj.is_ok());
assert_eq!(read_obj.unwrap().metadata, Some(HashMap::new()));
let mut map = HashMap::new();
map.insert("tx-time".to_string(), "2007-19-01T11:12:00-000".to_string());
let update = s3.update_s3_object_metadata(
UPDATE_BUCKET.to_string(),
"meta-key".to_string(),
Some(map),
None,
);
assert!(update.await.is_ok());
let read_obj = s3
.read_s3_object(
UPDATE_BUCKET.to_string(),
"meta-key".to_string(),
None,
None,
None,
)
.await;
assert!(read_obj.is_ok());
assert_eq!(
read_obj.unwrap().metadata.unwrap()["tx-time"],
"2007-19-01T11:12:00-000"
);
}