mirror of
https://github.com/genuinetools/reg.git
synced 2024-09-12 06:11:03 -04:00
246 lines
5.8 KiB
Go
246 lines
5.8 KiB
Go
![]() |
package swift
|
||
|
|
||
|
import (
|
||
|
"io/ioutil"
|
||
|
"os"
|
||
|
"reflect"
|
||
|
"strconv"
|
||
|
"strings"
|
||
|
"testing"
|
||
|
|
||
|
"github.com/ncw/swift/swifttest"
|
||
|
|
||
|
"github.com/docker/distribution/context"
|
||
|
storagedriver "github.com/docker/distribution/registry/storage/driver"
|
||
|
"github.com/docker/distribution/registry/storage/driver/testsuites"
|
||
|
|
||
|
"gopkg.in/check.v1"
|
||
|
)
|
||
|
|
||
|
// Hook up gocheck into the "go test" runner.
|
||
|
func Test(t *testing.T) { check.TestingT(t) }
|
||
|
|
||
|
var swiftDriverConstructor func(prefix string) (*Driver, error)
|
||
|
|
||
|
func init() {
|
||
|
var (
|
||
|
username string
|
||
|
password string
|
||
|
authURL string
|
||
|
tenant string
|
||
|
tenantID string
|
||
|
domain string
|
||
|
domainID string
|
||
|
tenantDomain string
|
||
|
tenantDomainID string
|
||
|
trustID string
|
||
|
container string
|
||
|
region string
|
||
|
AuthVersion int
|
||
|
endpointType string
|
||
|
insecureSkipVerify bool
|
||
|
secretKey string
|
||
|
accessKey string
|
||
|
containerKey bool
|
||
|
tempURLMethods []string
|
||
|
|
||
|
swiftServer *swifttest.SwiftServer
|
||
|
err error
|
||
|
)
|
||
|
username = os.Getenv("SWIFT_USERNAME")
|
||
|
password = os.Getenv("SWIFT_PASSWORD")
|
||
|
authURL = os.Getenv("SWIFT_AUTH_URL")
|
||
|
tenant = os.Getenv("SWIFT_TENANT_NAME")
|
||
|
tenantID = os.Getenv("SWIFT_TENANT_ID")
|
||
|
domain = os.Getenv("SWIFT_DOMAIN_NAME")
|
||
|
domainID = os.Getenv("SWIFT_DOMAIN_ID")
|
||
|
tenantDomain = os.Getenv("SWIFT_DOMAIN_NAME")
|
||
|
tenantDomainID = os.Getenv("SWIFT_DOMAIN_ID")
|
||
|
trustID = os.Getenv("SWIFT_TRUST_ID")
|
||
|
container = os.Getenv("SWIFT_CONTAINER_NAME")
|
||
|
region = os.Getenv("SWIFT_REGION_NAME")
|
||
|
AuthVersion, _ = strconv.Atoi(os.Getenv("SWIFT_AUTH_VERSION"))
|
||
|
endpointType = os.Getenv("SWIFT_ENDPOINT_TYPE")
|
||
|
insecureSkipVerify, _ = strconv.ParseBool(os.Getenv("SWIFT_INSECURESKIPVERIFY"))
|
||
|
secretKey = os.Getenv("SWIFT_SECRET_KEY")
|
||
|
accessKey = os.Getenv("SWIFT_ACCESS_KEY")
|
||
|
containerKey, _ = strconv.ParseBool(os.Getenv("SWIFT_TEMPURL_CONTAINERKEY"))
|
||
|
tempURLMethods = strings.Split(os.Getenv("SWIFT_TEMPURL_METHODS"), ",")
|
||
|
|
||
|
if username == "" || password == "" || authURL == "" || container == "" {
|
||
|
if swiftServer, err = swifttest.NewSwiftServer("localhost"); err != nil {
|
||
|
panic(err)
|
||
|
}
|
||
|
username = "swifttest"
|
||
|
password = "swifttest"
|
||
|
authURL = swiftServer.AuthURL
|
||
|
container = "test"
|
||
|
}
|
||
|
|
||
|
prefix, err := ioutil.TempDir("", "driver-")
|
||
|
if err != nil {
|
||
|
panic(err)
|
||
|
}
|
||
|
defer os.Remove(prefix)
|
||
|
|
||
|
swiftDriverConstructor = func(root string) (*Driver, error) {
|
||
|
parameters := Parameters{
|
||
|
username,
|
||
|
password,
|
||
|
authURL,
|
||
|
tenant,
|
||
|
tenantID,
|
||
|
domain,
|
||
|
domainID,
|
||
|
tenantDomain,
|
||
|
tenantDomainID,
|
||
|
trustID,
|
||
|
region,
|
||
|
AuthVersion,
|
||
|
container,
|
||
|
root,
|
||
|
endpointType,
|
||
|
insecureSkipVerify,
|
||
|
defaultChunkSize,
|
||
|
secretKey,
|
||
|
accessKey,
|
||
|
containerKey,
|
||
|
tempURLMethods,
|
||
|
}
|
||
|
|
||
|
return New(parameters)
|
||
|
}
|
||
|
|
||
|
driverConstructor := func() (storagedriver.StorageDriver, error) {
|
||
|
return swiftDriverConstructor(prefix)
|
||
|
}
|
||
|
|
||
|
testsuites.RegisterSuite(driverConstructor, testsuites.NeverSkip)
|
||
|
}
|
||
|
|
||
|
func TestEmptyRootList(t *testing.T) {
|
||
|
validRoot, err := ioutil.TempDir("", "driver-")
|
||
|
if err != nil {
|
||
|
t.Fatalf("unexpected error creating temporary directory: %v", err)
|
||
|
}
|
||
|
defer os.Remove(validRoot)
|
||
|
|
||
|
rootedDriver, err := swiftDriverConstructor(validRoot)
|
||
|
if err != nil {
|
||
|
t.Fatalf("unexpected error creating rooted driver: %v", err)
|
||
|
}
|
||
|
|
||
|
emptyRootDriver, err := swiftDriverConstructor("")
|
||
|
if err != nil {
|
||
|
t.Fatalf("unexpected error creating empty root driver: %v", err)
|
||
|
}
|
||
|
|
||
|
slashRootDriver, err := swiftDriverConstructor("/")
|
||
|
if err != nil {
|
||
|
t.Fatalf("unexpected error creating slash root driver: %v", err)
|
||
|
}
|
||
|
|
||
|
filename := "/test"
|
||
|
contents := []byte("contents")
|
||
|
ctx := context.Background()
|
||
|
err = rootedDriver.PutContent(ctx, filename, contents)
|
||
|
if err != nil {
|
||
|
t.Fatalf("unexpected error creating content: %v", err)
|
||
|
}
|
||
|
|
||
|
keys, err := emptyRootDriver.List(ctx, "/")
|
||
|
for _, path := range keys {
|
||
|
if !storagedriver.PathRegexp.MatchString(path) {
|
||
|
t.Fatalf("unexpected string in path: %q != %q", path, storagedriver.PathRegexp)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
keys, err = slashRootDriver.List(ctx, "/")
|
||
|
for _, path := range keys {
|
||
|
if !storagedriver.PathRegexp.MatchString(path) {
|
||
|
t.Fatalf("unexpected string in path: %q != %q", path, storagedriver.PathRegexp)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// Create an object with a path nested under the existing object
|
||
|
err = rootedDriver.PutContent(ctx, filename+"/file1", contents)
|
||
|
if err != nil {
|
||
|
t.Fatalf("unexpected error creating content: %v", err)
|
||
|
}
|
||
|
|
||
|
err = rootedDriver.Delete(ctx, filename)
|
||
|
if err != nil {
|
||
|
t.Fatalf("failed to delete: %v", err)
|
||
|
}
|
||
|
|
||
|
keys, err = rootedDriver.List(ctx, "/")
|
||
|
if err != nil {
|
||
|
t.Fatalf("failed to list objects after deletion: %v", err)
|
||
|
}
|
||
|
|
||
|
if len(keys) != 0 {
|
||
|
t.Fatal("delete did not remove nested objects")
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func TestFilenameChunking(t *testing.T) {
|
||
|
// Test valid input and sizes
|
||
|
input := []string{"a", "b", "c", "d", "e"}
|
||
|
expecteds := [][][]string{
|
||
|
{
|
||
|
{"a"},
|
||
|
{"b"},
|
||
|
{"c"},
|
||
|
{"d"},
|
||
|
{"e"},
|
||
|
},
|
||
|
{
|
||
|
{"a", "b"},
|
||
|
{"c", "d"},
|
||
|
{"e"},
|
||
|
},
|
||
|
{
|
||
|
{"a", "b", "c"},
|
||
|
{"d", "e"},
|
||
|
},
|
||
|
{
|
||
|
{"a", "b", "c", "d"},
|
||
|
{"e"},
|
||
|
},
|
||
|
{
|
||
|
{"a", "b", "c", "d", "e"},
|
||
|
},
|
||
|
{
|
||
|
{"a", "b", "c", "d", "e"},
|
||
|
},
|
||
|
}
|
||
|
for i, expected := range expecteds {
|
||
|
actual, err := chunkFilenames(input, i+1)
|
||
|
if !reflect.DeepEqual(actual, expected) {
|
||
|
t.Fatalf("chunk %v didn't match expected value %v", actual, expected)
|
||
|
}
|
||
|
if err != nil {
|
||
|
t.Fatalf("unexpected error chunking filenames: %v", err)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// Test nil input
|
||
|
actual, err := chunkFilenames(nil, 5)
|
||
|
if len(actual) != 0 {
|
||
|
t.Fatal("chunks were returned when passed nil")
|
||
|
}
|
||
|
if err != nil {
|
||
|
t.Fatalf("unexpected error chunking filenames: %v", err)
|
||
|
}
|
||
|
|
||
|
// Test 0 and < 0 sizes
|
||
|
actual, err = chunkFilenames(nil, 0)
|
||
|
if err == nil {
|
||
|
t.Fatal("expected error for size = 0")
|
||
|
}
|
||
|
actual, err = chunkFilenames(nil, -1)
|
||
|
if err == nil {
|
||
|
t.Fatal("expected error for size = -1")
|
||
|
}
|
||
|
}
|