go-ceph/rgw/admin/radosgw_test.go

157 lines
3.4 KiB
Go

package admin
import (
"fmt"
"net/http"
"net/http/httputil"
"os"
"reflect"
"testing"
"time"
"github.com/aws/aws-sdk-go/aws"
"github.com/aws/aws-sdk-go/aws/awserr"
"github.com/aws/aws-sdk-go/aws/credentials"
"github.com/aws/aws-sdk-go/aws/session"
"github.com/aws/aws-sdk-go/service/s3"
tsuite "github.com/stretchr/testify/suite"
)
type RadosGWTestSuite struct {
tsuite.Suite
endpoint string
accessKey string
secretKey string
bucketTestName string
}
type debugHTTPClient struct {
client HTTPClient
}
func newDebugHTTPClient(client HTTPClient) *debugHTTPClient {
return &debugHTTPClient{client}
}
func (c *debugHTTPClient) Do(req *http.Request) (*http.Response, error) {
dump, err := httputil.DumpRequestOut(req, true)
if err != nil {
return nil, err
}
fmt.Printf("\n%s\n", string(dump))
resp, err := c.client.Do(req)
if err != nil {
return nil, err
}
defer resp.Body.Close()
dump, err = httputil.DumpResponse(resp, true)
if err != nil {
return nil, err
}
fmt.Printf("\n%s\n", string(dump))
return resp, nil
}
func TestRadosGWTestSuite(t *testing.T) {
tsuite.Run(t, new(RadosGWTestSuite))
}
// S3Agent wraps the s3.S3 structure to allow for wrapper methods
type S3Agent struct {
Client *s3.S3
}
func newS3Agent(accessKey, secretKey, endpoint string, debug bool) (*S3Agent, error) {
const cephRegion = "us-east-1"
logLevel := aws.LogOff
if debug {
logLevel = aws.LogDebug
}
client := http.Client{
Timeout: time.Second * 15,
}
sess, err := session.NewSession(
aws.NewConfig().
WithRegion(cephRegion).
WithCredentials(credentials.NewStaticCredentials(accessKey, secretKey, "")).
WithEndpoint(endpoint).
WithS3ForcePathStyle(true).
WithMaxRetries(5).
WithDisableSSL(true).
WithHTTPClient(&client).
WithLogLevel(logLevel),
)
if err != nil {
return nil, err
}
svc := s3.New(sess)
return &S3Agent{
Client: svc,
}, nil
}
func (s *S3Agent) createBucket(name string) error {
bucketInput := &s3.CreateBucketInput{
Bucket: &name,
}
_, err := s.Client.CreateBucket(bucketInput)
if err != nil {
if aerr, ok := err.(awserr.Error); ok {
switch aerr.Code() {
case s3.ErrCodeBucketAlreadyExists:
return nil
case s3.ErrCodeBucketAlreadyOwnedByYou:
return nil
}
}
return fmt.Errorf("failed to create bucket %q. %w", name, err)
}
return nil
}
func (suite *RadosGWTestSuite) SetupConnection() {
suite.accessKey = "2262XNX11FZRR44XWIRD"
suite.secretKey = "rmtuS1Uj1bIC08QFYGW18GfSHAbkPqdsuYynNudw"
hostname := os.Getenv("HOSTNAME")
endpoint := hostname
if hostname != "test_ceph_aio" {
endpoint = "test_ceph_a"
}
suite.endpoint = "http://" + endpoint
suite.bucketTestName = "test"
}
func TestNew(t *testing.T) {
type args struct {
endpoint string
accessKey string
secretKey string
}
tests := []struct {
name string
args args
want *API
wantErr error
}{
{"no endpoint", args{}, nil, errNoEndpoint},
{"no accessKey", args{endpoint: "http://192.168.0.1"}, nil, errNoAccessKey},
{"no secretKey", args{endpoint: "http://192.168.0.1", accessKey: "foo"}, nil, errNoSecretKey},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
got, err := New(tt.args.endpoint, tt.args.accessKey, tt.args.secretKey, nil)
if tt.wantErr != err {
t.Errorf("New() error = %v, wantErr %v", err, tt.wantErr)
return
}
if !reflect.DeepEqual(got, tt.want) {
t.Errorf("New() = %v, want %v", got, tt.want)
}
})
}
}