This is to ensure that there are no projects that try to import `minio/minio/pkg` into their own repo. Any such common packages should go to `https://github.com/minio/pkg`
248 lines
7 KiB
Go
248 lines
7 KiB
Go
// Copyright (c) 2015-2021 MinIO, Inc.
|
|
//
|
|
// This file is part of MinIO Object Storage stack
|
|
//
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU Affero General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// This program is distributed in the hope that it will be useful
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU Affero General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU Affero General Public License
|
|
// along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
package cmd
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"net/http"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/minio/minio/internal/bucket/lifecycle"
|
|
xhttp "github.com/minio/minio/internal/http"
|
|
)
|
|
|
|
// TestParseRestoreObjStatus tests parseRestoreObjStatus
|
|
func TestParseRestoreObjStatus(t *testing.T) {
|
|
testCases := []struct {
|
|
restoreHdr string
|
|
expectedStatus restoreObjStatus
|
|
expectedErr error
|
|
}{
|
|
{
|
|
// valid: represents a restored object, 'pending' expiry.
|
|
restoreHdr: "ongoing-request=false, expiry-date=Fri, 21 Dec 2012 00:00:00 GMT",
|
|
expectedStatus: restoreObjStatus{
|
|
ongoing: false,
|
|
expiry: time.Date(2012, 12, 21, 0, 0, 0, 0, time.UTC),
|
|
},
|
|
expectedErr: nil,
|
|
},
|
|
{
|
|
// valid: represents an ongoing restore object request.
|
|
restoreHdr: "ongoing-request=true",
|
|
expectedStatus: restoreObjStatus{
|
|
ongoing: true,
|
|
},
|
|
expectedErr: nil,
|
|
},
|
|
{
|
|
// invalid; ongoing restore object request can't have expiry set on it.
|
|
restoreHdr: "ongoing-request=true, expiry-date=Fri, 21 Dec 2012 00:00:00 GMT",
|
|
expectedStatus: restoreObjStatus{},
|
|
expectedErr: errRestoreHDRMalformed,
|
|
},
|
|
{
|
|
// invalid; completed restore object request must have expiry set on it.
|
|
restoreHdr: "ongoing-request=false",
|
|
expectedStatus: restoreObjStatus{},
|
|
expectedErr: errRestoreHDRMalformed,
|
|
},
|
|
}
|
|
for i, tc := range testCases {
|
|
actual, err := parseRestoreObjStatus(tc.restoreHdr)
|
|
if err != tc.expectedErr {
|
|
t.Fatalf("Test %d: got %v expected %v", i+1, err, tc.expectedErr)
|
|
}
|
|
if actual != tc.expectedStatus {
|
|
t.Fatalf("Test %d: got %v expected %v", i+1, actual, tc.expectedStatus)
|
|
}
|
|
}
|
|
}
|
|
|
|
// TestRestoreObjStatusRoundTrip restoreObjStatus roundtrip
|
|
func TestRestoreObjStatusRoundTrip(t *testing.T) {
|
|
testCases := []restoreObjStatus{
|
|
ongoingRestoreObj(),
|
|
completedRestoreObj(time.Now().UTC()),
|
|
}
|
|
for i, tc := range testCases {
|
|
actual, err := parseRestoreObjStatus(tc.String())
|
|
if err != nil {
|
|
t.Fatalf("Test %d: parse restore object failed: %v", i+1, err)
|
|
}
|
|
if actual.ongoing != tc.ongoing || actual.expiry.Format(http.TimeFormat) != tc.expiry.Format(http.TimeFormat) {
|
|
t.Fatalf("Test %d: got %v expected %v", i+1, actual, tc)
|
|
}
|
|
}
|
|
}
|
|
|
|
// TestRestoreObjOnDisk tests restoreObjStatus' OnDisk method
|
|
func TestRestoreObjOnDisk(t *testing.T) {
|
|
testCases := []struct {
|
|
restoreStatus restoreObjStatus
|
|
ondisk bool
|
|
}{
|
|
{
|
|
// restore in progress
|
|
restoreStatus: ongoingRestoreObj(),
|
|
ondisk: false,
|
|
},
|
|
{
|
|
// restore completed but expired
|
|
restoreStatus: completedRestoreObj(time.Now().Add(-time.Hour)),
|
|
ondisk: false,
|
|
},
|
|
{
|
|
// restore completed
|
|
restoreStatus: completedRestoreObj(time.Now().Add(time.Hour)),
|
|
ondisk: true,
|
|
},
|
|
}
|
|
|
|
for i, tc := range testCases {
|
|
if actual := tc.restoreStatus.OnDisk(); actual != tc.ondisk {
|
|
t.Fatalf("Test %d: expected %v but got %v", i+1, tc.ondisk, actual)
|
|
}
|
|
}
|
|
}
|
|
|
|
// TestIsRestoredObjectOnDisk tests isRestoredObjectOnDisk helper function
|
|
func TestIsRestoredObjectOnDisk(t *testing.T) {
|
|
testCases := []struct {
|
|
meta map[string]string
|
|
ondisk bool
|
|
}{
|
|
{
|
|
// restore in progress
|
|
meta: map[string]string{
|
|
xhttp.AmzRestore: ongoingRestoreObj().String(),
|
|
},
|
|
ondisk: false,
|
|
},
|
|
{
|
|
// restore completed
|
|
meta: map[string]string{
|
|
xhttp.AmzRestore: completedRestoreObj(time.Now().Add(time.Hour)).String(),
|
|
},
|
|
ondisk: true,
|
|
},
|
|
{
|
|
// restore completed but expired
|
|
meta: map[string]string{
|
|
xhttp.AmzRestore: completedRestoreObj(time.Now().Add(-time.Hour)).String(),
|
|
},
|
|
ondisk: false,
|
|
},
|
|
}
|
|
|
|
for i, tc := range testCases {
|
|
if actual := isRestoredObjectOnDisk(tc.meta); actual != tc.ondisk {
|
|
t.Fatalf("Test %d: expected %v but got %v for %v", i+1, tc.ondisk, actual, tc.meta)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestObjectIsRemote(t *testing.T) {
|
|
fi := newFileInfo("object", 8, 8)
|
|
fi.Erasure.Index = 1
|
|
if !fi.IsValid() {
|
|
t.Fatalf("unable to get xl meta")
|
|
}
|
|
|
|
testCases := []struct {
|
|
meta map[string]string
|
|
remote bool
|
|
}{
|
|
{
|
|
// restore in progress
|
|
meta: map[string]string{
|
|
xhttp.AmzRestore: ongoingRestoreObj().String(),
|
|
},
|
|
remote: true,
|
|
},
|
|
{
|
|
// restore completed
|
|
meta: map[string]string{
|
|
xhttp.AmzRestore: completedRestoreObj(time.Now().Add(time.Hour)).String(),
|
|
},
|
|
remote: false,
|
|
},
|
|
{
|
|
// restore completed but expired
|
|
meta: map[string]string{
|
|
xhttp.AmzRestore: completedRestoreObj(time.Now().Add(-time.Hour)).String(),
|
|
},
|
|
remote: true,
|
|
},
|
|
{
|
|
// restore never initiated
|
|
meta: map[string]string{},
|
|
remote: true,
|
|
},
|
|
}
|
|
for i, tc := range testCases {
|
|
// Set transition status to complete
|
|
fi.TransitionStatus = lifecycle.TransitionComplete
|
|
fi.Metadata = tc.meta
|
|
if got := fi.IsRemote(); got != tc.remote {
|
|
t.Fatalf("Test %d.a: expected %v got %v", i+1, tc.remote, got)
|
|
}
|
|
oi := fi.ToObjectInfo("bucket", "object")
|
|
if got := oi.IsRemote(); got != tc.remote {
|
|
t.Fatalf("Test %d.b: expected %v got %v", i+1, tc.remote, got)
|
|
}
|
|
}
|
|
// Reset transition status; An object that's not transitioned is not remote.
|
|
fi.TransitionStatus = ""
|
|
fi.Metadata = nil
|
|
if got := fi.IsRemote(); got != false {
|
|
t.Fatalf("Expected object not to be remote but got %v", got)
|
|
}
|
|
}
|
|
|
|
func TestValidateTransitionTier(t *testing.T) {
|
|
globalTierConfigMgr = NewTierConfigMgr()
|
|
testCases := []struct {
|
|
xml []byte
|
|
expectedErr error
|
|
}{
|
|
{
|
|
// non-existent storage-class
|
|
xml: []byte(`<LifecycleConfiguration xmlns="http://s3.amazonaws.com/doc/2006-03-01/"><Rule><ID>rule</ID><Prefix /><Status>Enabled</Status><Transition><Days>1</Days><StorageClass>"NONEXISTENT"</StorageClass></Transition></Rule></LifecycleConfiguration>`),
|
|
expectedErr: errInvalidStorageClass,
|
|
},
|
|
{
|
|
// no transition rule
|
|
xml: []byte(`<LifecycleConfiguration xmlns="http://s3.amazonaws.com/doc/2006-03-01/"><Rule><ID>rule</ID><Prefix /><Status>Enabled</Status><Expiration><Days>1</Days></Expiration></Rule></LifecycleConfiguration>`),
|
|
expectedErr: nil,
|
|
},
|
|
}
|
|
for i, tc := range testCases {
|
|
lc, err := lifecycle.ParseLifecycleConfig(bytes.NewReader(tc.xml))
|
|
if err != nil {
|
|
t.Fatalf("Test %d: Failed to parse lifecycle config %v", i+1, err)
|
|
}
|
|
|
|
err = validateTransitionTier(context.Background(), lc)
|
|
if err != tc.expectedErr {
|
|
t.Fatalf("Test %d: Expected %v but got %v", i+1, tc.expectedErr, err)
|
|
}
|
|
}
|
|
}
|