mirror of
https://github.com/octoleo/restic.git
synced 2024-06-10 12:52:21 +00:00
338 lines
9.8 KiB
Go
338 lines
9.8 KiB
Go
/*
|
|
* Minio Go Library for Amazon S3 Compatible Cloud Storage
|
|
* (C) 2015, 2016, 2017 Minio, Inc.
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
package minio
|
|
|
|
import (
|
|
"bytes"
|
|
"io"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"net/url"
|
|
"os"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/minio/minio-go/pkg/credentials"
|
|
"github.com/minio/minio-go/pkg/policy"
|
|
)
|
|
|
|
type customReader struct{}
|
|
|
|
func (c *customReader) Read(p []byte) (n int, err error) {
|
|
return 0, nil
|
|
}
|
|
|
|
func (c *customReader) Size() (n int64) {
|
|
return 10
|
|
}
|
|
|
|
// Tests getReaderSize() for various Reader types.
|
|
func TestGetReaderSize(t *testing.T) {
|
|
var reader io.Reader
|
|
size, err := getReaderSize(reader)
|
|
if err != nil {
|
|
t.Fatal("Error:", err)
|
|
}
|
|
if size != -1 {
|
|
t.Fatal("Reader shouldn't have any length.")
|
|
}
|
|
|
|
bytesReader := bytes.NewReader([]byte("Hello World"))
|
|
size, err = getReaderSize(bytesReader)
|
|
if err != nil {
|
|
t.Fatal("Error:", err)
|
|
}
|
|
if size != int64(len("Hello World")) {
|
|
t.Fatalf("Reader length doesn't match got: %v, want: %v", size, len("Hello World"))
|
|
}
|
|
|
|
size, err = getReaderSize(new(customReader))
|
|
if err != nil {
|
|
t.Fatal("Error:", err)
|
|
}
|
|
if size != int64(10) {
|
|
t.Fatalf("Reader length doesn't match got: %v, want: %v", size, 10)
|
|
}
|
|
|
|
stringsReader := strings.NewReader("Hello World")
|
|
size, err = getReaderSize(stringsReader)
|
|
if err != nil {
|
|
t.Fatal("Error:", err)
|
|
}
|
|
if size != int64(len("Hello World")) {
|
|
t.Fatalf("Reader length doesn't match got: %v, want: %v", size, len("Hello World"))
|
|
}
|
|
|
|
// Create request channel.
|
|
reqCh := make(chan getRequest, 1)
|
|
// Create response channel.
|
|
resCh := make(chan getResponse, 1)
|
|
// Create done channel.
|
|
doneCh := make(chan struct{})
|
|
|
|
objectInfo := ObjectInfo{Size: 10}
|
|
// Create the first request.
|
|
firstReq := getRequest{
|
|
isReadOp: false, // Perform only a HEAD object to get objectInfo.
|
|
isFirstReq: true,
|
|
}
|
|
// Create the expected response.
|
|
firstRes := getResponse{
|
|
objectInfo: objectInfo,
|
|
}
|
|
// Send the expected response.
|
|
resCh <- firstRes
|
|
|
|
// Test setting size on the first request.
|
|
objectReaderFirstReq := newObject(reqCh, resCh, doneCh)
|
|
defer objectReaderFirstReq.Close()
|
|
// Not checking the response here...just that the reader size is correct.
|
|
_, err = objectReaderFirstReq.doGetRequest(firstReq)
|
|
if err != nil {
|
|
t.Fatal("Error:", err)
|
|
}
|
|
|
|
// Validate that the reader size is the objectInfo size.
|
|
size, err = getReaderSize(objectReaderFirstReq)
|
|
if err != nil {
|
|
t.Fatal("Error:", err)
|
|
}
|
|
if size != int64(10) {
|
|
t.Fatalf("Reader length doesn't match got: %d, wanted %d", size, objectInfo.Size)
|
|
}
|
|
|
|
fileReader, err := ioutil.TempFile(os.TempDir(), "prefix")
|
|
if err != nil {
|
|
t.Fatal("Error:", err)
|
|
}
|
|
defer fileReader.Close()
|
|
defer os.RemoveAll(fileReader.Name())
|
|
|
|
size, err = getReaderSize(fileReader)
|
|
if err != nil {
|
|
t.Fatal("Error:", err)
|
|
}
|
|
if size == -1 {
|
|
t.Fatal("Reader length for file cannot be -1.")
|
|
}
|
|
|
|
// Verify for standard input, output and error file descriptors.
|
|
size, err = getReaderSize(os.Stdin)
|
|
if err != nil {
|
|
t.Fatal("Error:", err)
|
|
}
|
|
if size != -1 {
|
|
t.Fatal("Stdin should have length of -1.")
|
|
}
|
|
size, err = getReaderSize(os.Stdout)
|
|
if err != nil {
|
|
t.Fatal("Error:", err)
|
|
}
|
|
if size != -1 {
|
|
t.Fatal("Stdout should have length of -1.")
|
|
}
|
|
size, err = getReaderSize(os.Stderr)
|
|
if err != nil {
|
|
t.Fatal("Error:", err)
|
|
}
|
|
if size != -1 {
|
|
t.Fatal("Stderr should have length of -1.")
|
|
}
|
|
file, err := os.Open(os.TempDir())
|
|
if err != nil {
|
|
t.Fatal("Error:", err)
|
|
}
|
|
defer file.Close()
|
|
_, err = getReaderSize(file)
|
|
if err == nil {
|
|
t.Fatal("Input file as directory should throw an error.")
|
|
}
|
|
}
|
|
|
|
// Tests valid hosts for location.
|
|
func TestValidBucketLocation(t *testing.T) {
|
|
s3Hosts := []struct {
|
|
bucketLocation string
|
|
endpoint string
|
|
}{
|
|
{"us-east-1", "s3.amazonaws.com"},
|
|
{"unknown", "s3.amazonaws.com"},
|
|
{"ap-southeast-1", "s3-ap-southeast-1.amazonaws.com"},
|
|
}
|
|
for _, s3Host := range s3Hosts {
|
|
endpoint := getS3Endpoint(s3Host.bucketLocation)
|
|
if endpoint != s3Host.endpoint {
|
|
t.Fatal("Error: invalid bucket location", endpoint)
|
|
}
|
|
}
|
|
}
|
|
|
|
// Tests error response structure.
|
|
func TestErrorResponse(t *testing.T) {
|
|
var err error
|
|
err = ErrorResponse{
|
|
Code: "Testing",
|
|
}
|
|
errResp := ToErrorResponse(err)
|
|
if errResp.Code != "Testing" {
|
|
t.Fatal("Type conversion failed, we have an empty struct.")
|
|
}
|
|
|
|
// Test http response decoding.
|
|
var httpResponse *http.Response
|
|
// Set empty variables
|
|
httpResponse = nil
|
|
var bucketName, objectName string
|
|
|
|
// Should fail with invalid argument.
|
|
err = httpRespToErrorResponse(httpResponse, bucketName, objectName)
|
|
errResp = ToErrorResponse(err)
|
|
if errResp.Code != "InvalidArgument" {
|
|
t.Fatal("Empty response input should return invalid argument.")
|
|
}
|
|
}
|
|
|
|
// Tests signature type.
|
|
func TestSignatureType(t *testing.T) {
|
|
clnt := Client{}
|
|
if !clnt.overrideSignerType.IsV4() {
|
|
t.Fatal("Error")
|
|
}
|
|
clnt.overrideSignerType = credentials.SignatureV2
|
|
if !clnt.overrideSignerType.IsV2() {
|
|
t.Fatal("Error")
|
|
}
|
|
if clnt.overrideSignerType.IsV4() {
|
|
t.Fatal("Error")
|
|
}
|
|
clnt.overrideSignerType = credentials.SignatureV4
|
|
if !clnt.overrideSignerType.IsV4() {
|
|
t.Fatal("Error")
|
|
}
|
|
}
|
|
|
|
// Tests bucket policy types.
|
|
func TestBucketPolicyTypes(t *testing.T) {
|
|
want := map[string]bool{
|
|
"none": true,
|
|
"readonly": true,
|
|
"writeonly": true,
|
|
"readwrite": true,
|
|
"invalid": false,
|
|
}
|
|
for bucketPolicy, ok := range want {
|
|
if policy.BucketPolicy(bucketPolicy).IsValidBucketPolicy() != ok {
|
|
t.Fatal("Error")
|
|
}
|
|
}
|
|
}
|
|
|
|
// Tests optimal part size.
|
|
func TestPartSize(t *testing.T) {
|
|
_, _, _, err := optimalPartInfo(5000000000000000000)
|
|
if err == nil {
|
|
t.Fatal("Error: should fail")
|
|
}
|
|
totalPartsCount, partSize, lastPartSize, err := optimalPartInfo(5497558138880)
|
|
if err != nil {
|
|
t.Fatal("Error: ", err)
|
|
}
|
|
if totalPartsCount != 9103 {
|
|
t.Fatalf("Error: expecting total parts count of 9987: got %v instead", totalPartsCount)
|
|
}
|
|
if partSize != 603979776 {
|
|
t.Fatalf("Error: expecting part size of 550502400: got %v instead", partSize)
|
|
}
|
|
if lastPartSize != 134217728 {
|
|
t.Fatalf("Error: expecting last part size of 241172480: got %v instead", lastPartSize)
|
|
}
|
|
_, partSize, _, err = optimalPartInfo(5000000000)
|
|
if err != nil {
|
|
t.Fatal("Error:", err)
|
|
}
|
|
if partSize != minPartSize {
|
|
t.Fatalf("Error: expecting part size of %v: got %v instead", minPartSize, partSize)
|
|
}
|
|
totalPartsCount, partSize, lastPartSize, err = optimalPartInfo(-1)
|
|
if err != nil {
|
|
t.Fatal("Error:", err)
|
|
}
|
|
if totalPartsCount != 9103 {
|
|
t.Fatalf("Error: expecting total parts count of 9987: got %v instead", totalPartsCount)
|
|
}
|
|
if partSize != 603979776 {
|
|
t.Fatalf("Error: expecting part size of 550502400: got %v instead", partSize)
|
|
}
|
|
if lastPartSize != 134217728 {
|
|
t.Fatalf("Error: expecting last part size of 241172480: got %v instead", lastPartSize)
|
|
}
|
|
}
|
|
|
|
// TestMakeTargetURL - testing makeTargetURL()
|
|
func TestMakeTargetURL(t *testing.T) {
|
|
testCases := []struct {
|
|
addr string
|
|
secure bool
|
|
bucketName string
|
|
objectName string
|
|
bucketLocation string
|
|
queryValues map[string][]string
|
|
expectedURL url.URL
|
|
expectedErr error
|
|
}{
|
|
// Test 1
|
|
{"localhost:9000", false, "", "", "", nil, url.URL{Host: "localhost:9000", Scheme: "http", Path: "/"}, nil},
|
|
// Test 2
|
|
{"localhost", true, "", "", "", nil, url.URL{Host: "localhost", Scheme: "https", Path: "/"}, nil},
|
|
// Test 3
|
|
{"localhost:9000", true, "mybucket", "", "", nil, url.URL{Host: "localhost:9000", Scheme: "https", Path: "/mybucket/"}, nil},
|
|
// Test 4, testing against google storage API
|
|
{"storage.googleapis.com", true, "mybucket", "", "", nil, url.URL{Host: "mybucket.storage.googleapis.com", Scheme: "https", Path: "/"}, nil},
|
|
// Test 5, testing against AWS S3 API
|
|
{"s3.amazonaws.com", true, "mybucket", "myobject", "", nil, url.URL{Host: "mybucket.s3.amazonaws.com", Scheme: "https", Path: "/myobject"}, nil},
|
|
// Test 6
|
|
{"localhost:9000", false, "mybucket", "myobject", "", nil, url.URL{Host: "localhost:9000", Scheme: "http", Path: "/mybucket/myobject"}, nil},
|
|
// Test 7, testing with query
|
|
{"localhost:9000", false, "mybucket", "myobject", "", map[string][]string{"param": []string{"val"}}, url.URL{Host: "localhost:9000", Scheme: "http", Path: "/mybucket/myobject", RawQuery: "param=val"}, nil},
|
|
// Test 8, testing with port 80
|
|
{"localhost:80", false, "mybucket", "myobject", "", nil, url.URL{Host: "localhost", Scheme: "http", Path: "/mybucket/myobject"}, nil},
|
|
// Test 9, testing with port 443
|
|
{"localhost:443", true, "mybucket", "myobject", "", nil, url.URL{Host: "localhost", Scheme: "https", Path: "/mybucket/myobject"}, nil},
|
|
}
|
|
|
|
for i, testCase := range testCases {
|
|
// Initialize a Minio client
|
|
c, _ := New(testCase.addr, "foo", "bar", testCase.secure)
|
|
u, err := c.makeTargetURL(testCase.bucketName, testCase.objectName, testCase.bucketLocation, testCase.queryValues)
|
|
// Check the returned error
|
|
if testCase.expectedErr == nil && err != nil {
|
|
t.Fatalf("Test %d: Should succeed but failed with err = %v", i+1, err)
|
|
}
|
|
if testCase.expectedErr != nil && err == nil {
|
|
t.Fatalf("Test %d: Should fail but succeeded", i+1)
|
|
}
|
|
if err == nil {
|
|
// Check if the returned url is equal to what we expect
|
|
if u.String() != testCase.expectedURL.String() {
|
|
t.Fatalf("Test %d: Mismatched target url: expected = `%v`, found = `%v`",
|
|
i+1, testCase.expectedURL.String(), u.String())
|
|
}
|
|
}
|
|
}
|
|
}
|