From 167414b012ad0472b75cff50c29d1ba35289e43f Mon Sep 17 00:00:00 2001 From: Ian Milligan Date: Mon, 21 Nov 2016 13:47:29 -0800 Subject: [PATCH] Add s3_buckets test Create a new S3 bucket and test that it exists on each node. --- tests/s3_buckets.erl | 38 ++++++++++++++++++++++++++++++++++++++ 1 file changed, 38 insertions(+) create mode 100644 tests/s3_buckets.erl diff --git a/tests/s3_buckets.erl b/tests/s3_buckets.erl new file mode 100644 index 000000000..28092dba1 --- /dev/null +++ b/tests/s3_buckets.erl @@ -0,0 +1,38 @@ +-module(s3_buckets). + +-behaviour(riak_test). + +-export([confirm/0]). + +-define(BUCKET, <<"bucket">>). +-define(USERNAME, "user"). + +confirm() -> + Cluster = [Node1, _Node2] = rt:build_cluster(2), + create_bucket(Node1), + lists:foreach(fun (Node) -> + rt:wait_until(fun () -> verify_bucket(Node) end) + end, Cluster), + pass. + +create_bucket(Node) -> + rpc:call(Node, riak_s3_bucket, create, [?BUCKET, ?USERNAME]). + +verify_bucket(Node) -> + case rpc:call(Node, riak_s3_bucket, get, [?BUCKET]) of + not_found -> + not_found; + {error, Reason} -> + Reason; + Bucket -> + BucketName = riak_s3_bucket:get_name(Bucket), + User = riak_s3_bucket:get_username(Bucket), + if BucketName == ?BUCKET andalso User == ?USERNAME -> + true; + BucketName /= ?BUCKET -> + {incorrect_bucket_name, BucketName}; + User /= ?USERNAME -> + {incorrect_user, User} + end + end. +