go-ceph/rgw/admin/link_test.go

//go:build ceph_preview
// +build ceph_preview

package admin

import (
	"context"
	"net/http"
	"testing"

	"github.com/stretchr/testify/assert"
)

func (suite *RadosGWTestSuite) TestLink() {
	suite.SetupConnection()
	co, err := New(suite.endpoint, suite.accessKey, suite.secretKey, newDebugHTTPClient(http.DefaultClient))
	assert.NoError(suite.T(), err)

	suite.T().Run("create test user1", func(t *testing.T) {
		user, err := co.CreateUser(context.Background(), User{ID: "test-user1", DisplayName: "test-user1", Email: "test1@example.com"})
		assert.NoError(suite.T(), err)
		assert.Equal(suite.T(), "test-user1", user.ID)
		assert.Zero(suite.T(), len(user.Caps))
	})

	suite.T().Run("create test bucket", func(t *testing.T) {
		s3, err := newS3Agent(suite.accessKey, suite.secretKey, suite.endpoint, true)
		assert.NoError(t, err)

		err = s3.createBucket(suite.bucketTestName)
		assert.NoError(t, err)
	})

	suite.T().Run("create test user2", func(t *testing.T) {
		user, err := co.CreateUser(context.Background(), User{ID: "test-user2", DisplayName: "test-user2", Email: "test2@example.com"})
		assert.NoError(suite.T(), err)
		assert.Equal(suite.T(), "test-user2", user.ID)
		assert.Zero(suite.T(), len(user.Caps))
	})

	suite.T().Run("link test-user2", func(t *testing.T) {
		bucket, err := co.GetBucketInfo(context.Background(), Bucket{Bucket: suite.bucketTestName})
		assert.NoError(t, err)

		err = co.LinkBucket(context.Background(), BucketLinkInput{
			Bucket:   suite.bucketTestName,
			BucketID: bucket.ID,
			UID:      "test-user2",
		})
		assert.NoError(t, err)

		bucket, err = co.GetBucketInfo(context.Background(), Bucket{Bucket: suite.bucketTestName})
		assert.NoError(t, err)
		assert.Equal(t, bucket.Owner, "test-user2")
	})

	suite.T().Run("unlink test-user2", func(t *testing.T) {
		bucket, err := co.GetBucketInfo(context.Background(), Bucket{Bucket: suite.bucketTestName})
		assert.NoError(t, err)

		err = co.UnlinkBucket(context.Background(), BucketLinkInput{
			Bucket: suite.bucketTestName,
			UID:    bucket.Owner,
		})
		assert.NoError(t, err)
	})

	suite.T().Run("remove bucket", func(t *testing.T) {
		err := co.RemoveBucket(context.Background(), Bucket{Bucket: suite.bucketTestName})
		assert.NoError(suite.T(), err)
	})

	suite.T().Run("delete test user1", func(t *testing.T) {
		err := co.RemoveUser(context.Background(), User{ID: "test-user1"})
		assert.NoError(suite.T(), err)
	})

	suite.T().Run("delete test user2", func(t *testing.T) {
		err := co.RemoveUser(context.Background(), User{ID: "test-user2"})
		assert.NoError(suite.T(), err)
	})
}