diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/garage/Cargo.toml | 1 | ||||
-rw-r--r-- | src/garage/tests/common/mod.rs | 14 | ||||
-rw-r--r-- | src/garage/tests/common/util.rs | 13 | ||||
-rw-r--r-- | src/garage/tests/simple.rs | 12 |
4 files changed, 31 insertions, 9 deletions
diff --git a/src/garage/Cargo.toml b/src/garage/Cargo.toml index d6034bbd..eaf35358 100644 --- a/src/garage/Cargo.toml +++ b/src/garage/Cargo.toml @@ -56,4 +56,5 @@ netapp = "0.3.0" aws-sdk-s3 = "0.6" http = "0.2" +rand = "0.8" static_init = "1.0" diff --git a/src/garage/tests/common/mod.rs b/src/garage/tests/common/mod.rs index b5f7f0b2..c40fde0f 100644 --- a/src/garage/tests/common/mod.rs +++ b/src/garage/tests/common/mod.rs @@ -7,6 +7,7 @@ pub mod macros; pub mod client; pub mod ext; pub mod garage; +pub mod util; const REGION: Region = Region::from_static("garage-integ-test"); @@ -23,20 +24,27 @@ impl Context { Context { garage, client } } - pub fn create_bucket(&self, name: &str) { + /// Create an unique bucket with a random suffix. + /// + /// Return the created bucket full name. + pub fn create_bucket(&self, name: &str) -> String { + let bucket_name = format!("{}-{}", name, util::random_id(6)); + self.garage .command() - .args(["bucket", "create", name]) + .args(["bucket", "create", &bucket_name]) .quiet() .expect_success_status("Could not create bucket"); self.garage .command() .args(["bucket", "allow"]) .args(["--owner", "--read", "--write"]) - .arg(name) + .arg(&bucket_name) .args(["--key", &self.garage.key.name]) .quiet() .expect_success_status("Could not allow key for bucket"); + + bucket_name } } diff --git a/src/garage/tests/common/util.rs b/src/garage/tests/common/util.rs new file mode 100644 index 00000000..49c72879 --- /dev/null +++ b/src/garage/tests/common/util.rs @@ -0,0 +1,13 @@ +pub fn random_id(len: usize) -> String { + use rand::distributions::Slice; + use rand::Rng; + + static ALPHABET: &[u8] = b"abcdefghijklmnopqrstuvwxyz0123456789."; + + let rng = rand::thread_rng(); + rng.sample_iter(Slice::new(ALPHABET).unwrap()) + .map(|b| char::from(*b)) + .filter(|c| c.is_ascii_lowercase() || c.is_ascii_digit()) + .take(len) + .collect() +} diff --git a/src/garage/tests/simple.rs b/src/garage/tests/simple.rs index a627c770..bf88d8c2 100644 --- a/src/garage/tests/simple.rs +++ b/src/garage/tests/simple.rs @@ -5,13 +5,13 @@ async fn test_simple() { use aws_sdk_s3::ByteStream; let ctx = common::context(); - ctx.create_bucket("test-simple"); + let bucket = ctx.create_bucket("test-simple"); let data = ByteStream::from_static(b"Hello world!"); ctx.client .put_object() - .bucket("test-simple") + .bucket(&bucket) .key("test") .body(data) .send() @@ -21,7 +21,7 @@ async fn test_simple() { let res = ctx .client .get_object() - .bucket("test-simple") + .bucket(&bucket) .key("test") .send() .await @@ -35,13 +35,13 @@ async fn test_simple_2() { use aws_sdk_s3::ByteStream; let ctx = common::context(); - ctx.create_bucket("test-simple-2"); + let bucket = ctx.create_bucket("test-simple-2"); let data = ByteStream::from_static(b"Hello world!"); ctx.client .put_object() - .bucket("test-simple-2") + .bucket(&bucket) .key("test") .body(data) .send() @@ -51,7 +51,7 @@ async fn test_simple_2() { let res = ctx .client .get_object() - .bucket("test-simple-2") + .bucket(&bucket) .key("test") .send() .await |