aboutsummaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rw-r--r--src/garage/Cargo.toml1
-rw-r--r--src/garage/tests/common/mod.rs14
-rw-r--r--src/garage/tests/common/util.rs13
-rw-r--r--src/garage/tests/simple.rs12
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