72 lines
1.7 KiB
Go
72 lines
1.7 KiB
Go
package alicloudoss
|
|
|
|
import (
|
|
"fmt"
|
|
"math/rand"
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
|
|
log "github.com/hashicorp/go-hclog"
|
|
"github.com/hashicorp/vault/helper/logging"
|
|
"github.com/hashicorp/vault/physical"
|
|
"github.com/tyrannosaurus-becks/aliyun-oss-go-sdk/oss"
|
|
)
|
|
|
|
func TestAliCloudOSSBackend(t *testing.T) {
|
|
|
|
// ex. http://oss-us-east-1.aliyuncs.com
|
|
endpoint := os.Getenv("ALICLOUD_OSS_ENDPOINT")
|
|
accessKeyID := os.Getenv("ALICLOUD_ACCESS_KEY")
|
|
accessKeySecret := os.Getenv("ALICLOUD_SECRET_KEY")
|
|
|
|
if endpoint == "" || accessKeyID == "" || accessKeySecret == "" {
|
|
t.SkipNow()
|
|
}
|
|
|
|
conn, err := oss.New(endpoint, accessKeyID, accessKeySecret)
|
|
if err != nil {
|
|
t.Fatalf("unable to create test client: %s", err)
|
|
}
|
|
|
|
var randInt = rand.New(rand.NewSource(time.Now().UnixNano())).Int()
|
|
bucket := fmt.Sprintf("vault-alibaba-testacc-%d", randInt)
|
|
|
|
err = conn.CreateBucket(bucket)
|
|
if err != nil {
|
|
t.Fatalf("unable to create test bucket: %s", err)
|
|
}
|
|
|
|
defer func() {
|
|
// Gotta list all the objects and delete them
|
|
// before being able to delete the bucket
|
|
b, err := conn.Bucket(bucket)
|
|
listResp, err := b.ListObjects()
|
|
|
|
objects := []string{}
|
|
for _, object := range listResp.Objects {
|
|
objects = append(objects, object.Key)
|
|
}
|
|
|
|
b.DeleteObjects(objects)
|
|
|
|
err = conn.DeleteBucket(bucket)
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
}()
|
|
|
|
logger := logging.NewVaultLogger(log.Debug)
|
|
|
|
// This uses the same logic to find the Alibaba credentials as we did at the beginning of the test
|
|
b, err := NewAliCloudOSSBackend(map[string]string{
|
|
"bucket": bucket,
|
|
}, logger)
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
|
|
physical.ExerciseBackend(t, b)
|
|
physical.ExerciseBackend_ListPrefix(t, b)
|
|
}
|