go-ceph/rgw/admin/user_bucket_test.go
Anoop C S 72ac947edf rgw/admin: Make API.ListUserBuckets and API.ListUserBucketsWithStat stable
Signed-off-by: Anoop C S <anoopcs@cryptolab.net>
2022-12-06 14:20:29 +00:00

64 lines
2.0 KiB
Go

package admin
import (
"context"
"errors"
"net/http"
"testing"
"github.com/stretchr/testify/assert"
)
func (suite *RadosGWTestSuite) TestUserBucket() {
suite.SetupConnection()
co, err := New(suite.endpoint, suite.accessKey, suite.secretKey, newDebugHTTPClient(http.DefaultClient))
assert.NoError(suite.T(), err)
s3, err := newS3Agent(suite.accessKey, suite.secretKey, suite.endpoint, true)
assert.NoError(suite.T(), err)
err = s3.createBucket(suite.bucketTestName)
assert.NoError(suite.T(), err)
suite.T().Run("list empty user's buckets", func(t *testing.T) {
_, err := co.ListUsersBuckets(context.Background(), "")
assert.Error(suite.T(), err)
assert.True(suite.T(), errors.Is(err, errMissingUserID), err)
})
suite.T().Run("list user's buckets", func(t *testing.T) {
buckets, err := co.ListUsersBuckets(context.Background(), "admin")
assert.NoError(suite.T(), err)
assert.Equal(suite.T(), 1, len(buckets))
})
suite.T().Run("list unknown user's buckets", func(t *testing.T) {
buckets, err := co.ListUsersBuckets(context.Background(), "foo")
assert.NoError(suite.T(), err)
assert.Equal(suite.T(), 0, len(buckets))
})
suite.T().Run("list empty user's buckets with stat", func(t *testing.T) {
_, err := co.ListUsersBucketsWithStat(context.Background(), "")
assert.Error(suite.T(), err)
assert.True(suite.T(), errors.Is(err, errMissingUserID), err)
})
suite.T().Run("list user's buckets with stat", func(t *testing.T) {
buckets, err := co.ListUsersBucketsWithStat(context.Background(), "admin")
assert.NoError(suite.T(), err)
assert.Equal(suite.T(), 1, len(buckets))
b := buckets[0]
assert.NotNil(suite.T(), b)
assert.Equal(suite.T(), suite.bucketTestName, b.Bucket)
assert.Equal(suite.T(), "admin", b.Owner)
assert.NotNil(suite.T(), b.BucketQuota.MaxSize)
})
suite.T().Run("list unknown user's buckets with stat", func(t *testing.T) {
buckets, err := co.ListUsersBucketsWithStat(context.Background(), "foo")
assert.NoError(suite.T(), err)
assert.Equal(suite.T(), 0, len(buckets))
})
}