mirror of https://github.com/minio/minio.git
				
				
				
			
		
			
				
	
	
		
			472 lines
		
	
	
		
			12 KiB
		
	
	
	
		
			Go
		
	
	
	
			
		
		
	
	
			472 lines
		
	
	
		
			12 KiB
		
	
	
	
		
			Go
		
	
	
	
| /*
 | |
|  * Minio Cloud Storage, (C) 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 cmd
 | |
| 
 | |
| import (
 | |
| 	"bytes"
 | |
| 	"errors"
 | |
| 	"fmt"
 | |
| 	"io"
 | |
| 	"net"
 | |
| 	"net/rpc"
 | |
| 	"runtime"
 | |
| 	"testing"
 | |
| 
 | |
| 	"golang.org/x/crypto/blake2b"
 | |
| )
 | |
| 
 | |
| // Tests the construction of canonical string by the
 | |
| // Stringer method for StorageAPI.
 | |
| func TestStorageCanonicalStrings(t *testing.T) {
 | |
| 	testCases := []struct {
 | |
| 		storageAPI    StorageAPI
 | |
| 		canonicalPath string
 | |
| 	}{
 | |
| 		// Canonicalized name as unix path.
 | |
| 		{
 | |
| 			storageAPI: &posix{
 | |
| 				diskPath: "/tmp",
 | |
| 			},
 | |
| 			canonicalPath: "/tmp",
 | |
| 		},
 | |
| 		// Canonicalized name as windows path.
 | |
| 		{
 | |
| 			storageAPI: &posix{
 | |
| 				diskPath: "C:/tmp",
 | |
| 			},
 | |
| 			canonicalPath: "C:/tmp",
 | |
| 		},
 | |
| 		// Canonicalized name as unix path.
 | |
| 		{
 | |
| 			storageAPI: &networkStorage{
 | |
| 				rpcClient: newAuthRPCClient(authConfig{
 | |
| 					accessKey:        "",
 | |
| 					secretKey:        "",
 | |
| 					serverAddr:       "localhost:9000",
 | |
| 					serviceEndpoint:  "/tmp",
 | |
| 					secureConn:       false,
 | |
| 					serviceName:      "Storage",
 | |
| 					disableReconnect: true,
 | |
| 				}),
 | |
| 			},
 | |
| 			canonicalPath: "http://localhost:9000/tmp",
 | |
| 		},
 | |
| 		// Canonicalized name as non TLS.
 | |
| 		{
 | |
| 			storageAPI: &networkStorage{
 | |
| 				rpcClient: newAuthRPCClient(authConfig{
 | |
| 					accessKey:        "",
 | |
| 					secretKey:        "",
 | |
| 					serverAddr:       "localhost:9000",
 | |
| 					serviceEndpoint:  "C:/tmp",
 | |
| 					secureConn:       false,
 | |
| 					serviceName:      "Storage",
 | |
| 					disableReconnect: true,
 | |
| 				}),
 | |
| 			},
 | |
| 			canonicalPath: "http://localhost:9000/C:/tmp",
 | |
| 		},
 | |
| 		// Canonicalized name as TLS.
 | |
| 		{
 | |
| 			storageAPI: &networkStorage{
 | |
| 				rpcClient: newAuthRPCClient(authConfig{
 | |
| 					accessKey:        "",
 | |
| 					secretKey:        "",
 | |
| 					serverAddr:       "localhost:9000",
 | |
| 					serviceEndpoint:  "C:/tmp",
 | |
| 					secureConn:       true,
 | |
| 					serviceName:      "Storage",
 | |
| 					disableReconnect: true,
 | |
| 				}),
 | |
| 			},
 | |
| 			canonicalPath: "https://localhost:9000/C:/tmp",
 | |
| 		},
 | |
| 	}
 | |
| 
 | |
| 	// Validate all the test cases.
 | |
| 	for i, testCase := range testCases {
 | |
| 		p := testCase.storageAPI
 | |
| 		if p.String() != testCase.canonicalPath {
 | |
| 			t.Errorf("Test %d: Expected %s, got %s", i+1, testCase.canonicalPath, p.String())
 | |
| 		}
 | |
| 	}
 | |
| }
 | |
| 
 | |
| // Tests storage error transformation.
 | |
| func TestStorageErr(t *testing.T) {
 | |
| 	unknownErr := errors.New("Unknown error")
 | |
| 	testCases := []struct {
 | |
| 		expectedErr error
 | |
| 		err         error
 | |
| 	}{
 | |
| 		{
 | |
| 			expectedErr: nil,
 | |
| 			err:         nil,
 | |
| 		},
 | |
| 		{
 | |
| 			expectedErr: io.EOF,
 | |
| 			err:         fmt.Errorf("%s", io.EOF.Error()),
 | |
| 		},
 | |
| 		{
 | |
| 			expectedErr: io.ErrUnexpectedEOF,
 | |
| 			err:         fmt.Errorf("%s", io.ErrUnexpectedEOF.Error()),
 | |
| 		},
 | |
| 		{
 | |
| 			expectedErr: errDiskNotFoundFromNetError,
 | |
| 			err:         &net.OpError{},
 | |
| 		},
 | |
| 		{
 | |
| 			expectedErr: errDiskNotFoundFromRPCShutdown,
 | |
| 			err:         rpc.ErrShutdown,
 | |
| 		},
 | |
| 		{
 | |
| 			expectedErr: errUnexpected,
 | |
| 			err:         fmt.Errorf("%s", errUnexpected.Error()),
 | |
| 		},
 | |
| 		{
 | |
| 			expectedErr: errDiskFull,
 | |
| 			err:         fmt.Errorf("%s", errDiskFull.Error()),
 | |
| 		},
 | |
| 		{
 | |
| 			expectedErr: errVolumeNotFound,
 | |
| 			err:         fmt.Errorf("%s", errVolumeNotFound.Error()),
 | |
| 		},
 | |
| 		{
 | |
| 			expectedErr: errVolumeExists,
 | |
| 			err:         fmt.Errorf("%s", errVolumeExists.Error()),
 | |
| 		},
 | |
| 		{
 | |
| 			expectedErr: errFileNotFound,
 | |
| 			err:         fmt.Errorf("%s", errFileNotFound.Error()),
 | |
| 		},
 | |
| 		{
 | |
| 			expectedErr: errFileAccessDenied,
 | |
| 			err:         fmt.Errorf("%s", errFileAccessDenied.Error()),
 | |
| 		},
 | |
| 		{
 | |
| 			expectedErr: errIsNotRegular,
 | |
| 			err:         fmt.Errorf("%s", errIsNotRegular.Error()),
 | |
| 		},
 | |
| 		{
 | |
| 			expectedErr: errVolumeNotEmpty,
 | |
| 			err:         fmt.Errorf("%s", errVolumeNotEmpty.Error()),
 | |
| 		},
 | |
| 		{
 | |
| 			expectedErr: errVolumeAccessDenied,
 | |
| 			err:         fmt.Errorf("%s", errVolumeAccessDenied.Error()),
 | |
| 		},
 | |
| 		{
 | |
| 			expectedErr: errCorruptedFormat,
 | |
| 			err:         fmt.Errorf("%s", errCorruptedFormat.Error()),
 | |
| 		},
 | |
| 		{
 | |
| 			expectedErr: errUnformattedDisk,
 | |
| 			err:         fmt.Errorf("%s", errUnformattedDisk.Error()),
 | |
| 		},
 | |
| 		{
 | |
| 			expectedErr: errFileNameTooLong,
 | |
| 			err:         fmt.Errorf("%s", errFileNameTooLong.Error()),
 | |
| 		},
 | |
| 		{
 | |
| 			expectedErr: errInvalidAccessKeyID,
 | |
| 			err:         fmt.Errorf("%s", errInvalidAccessKeyID.Error()),
 | |
| 		},
 | |
| 		{
 | |
| 			expectedErr: errAuthentication,
 | |
| 			err:         fmt.Errorf("%s", errAuthentication.Error()),
 | |
| 		},
 | |
| 		{
 | |
| 			expectedErr: errServerVersionMismatch,
 | |
| 			err:         fmt.Errorf("%s", errServerVersionMismatch.Error()),
 | |
| 		},
 | |
| 		{
 | |
| 			expectedErr: errServerTimeMismatch,
 | |
| 			err:         fmt.Errorf("%s", errServerTimeMismatch.Error()),
 | |
| 		},
 | |
| 		{
 | |
| 			expectedErr: unknownErr,
 | |
| 			err:         unknownErr,
 | |
| 		},
 | |
| 	}
 | |
| 	for i, testCase := range testCases {
 | |
| 		resultErr := toStorageErr(testCase.err)
 | |
| 		if testCase.expectedErr != resultErr {
 | |
| 			t.Errorf("Test %d: Expected %s, got %s", i+1, testCase.expectedErr, resultErr)
 | |
| 		}
 | |
| 	}
 | |
| }
 | |
| 
 | |
| // API suite container common to both FS and XL.
 | |
| type TestRPCStorageSuite struct {
 | |
| 	serverType  string
 | |
| 	testServer  TestServer
 | |
| 	remoteDisks []StorageAPI
 | |
| }
 | |
| 
 | |
| // Setting up the test suite.
 | |
| // Starting the Test server with temporary FS backend.
 | |
| func (s *TestRPCStorageSuite) SetUpSuite(t *testing.T) {
 | |
| 	s.testServer = StartTestStorageRPCServer(t, s.serverType, 1)
 | |
| 	listenAddress := s.testServer.Server.Listener.Addr().String()
 | |
| 
 | |
| 	for _, ep := range s.testServer.Disks {
 | |
| 		// Eventhough s.testServer.Disks is EndpointList, we would need a URLEndpointType here.
 | |
| 		endpoint := ep
 | |
| 		if endpoint.Type() == PathEndpointType {
 | |
| 			endpoint.Scheme = "http"
 | |
| 		}
 | |
| 		endpoint.Host = listenAddress
 | |
| 		storageDisk := newStorageRPC(endpoint)
 | |
| 		s.remoteDisks = append(s.remoteDisks, storageDisk)
 | |
| 	}
 | |
| }
 | |
| 
 | |
| // No longer used with gocheck, but used in explicit teardown code in
 | |
| // each test function. Called implicitly by after all tests are run.
 | |
| func (s *TestRPCStorageSuite) TearDownSuite(t *testing.T) {
 | |
| 	s.testServer.Stop()
 | |
| }
 | |
| 
 | |
| func TestRPCStorageClient(t *testing.T) {
 | |
| 	// Setup code
 | |
| 	s := &TestRPCStorageSuite{serverType: "XL"}
 | |
| 	s.SetUpSuite(t)
 | |
| 
 | |
| 	// Run the test.
 | |
| 	s.testRPCStorageClient(t)
 | |
| 
 | |
| 	// Teardown code
 | |
| 	s.TearDownSuite(t)
 | |
| }
 | |
| 
 | |
| func (s *TestRPCStorageSuite) testRPCStorageClient(t *testing.T) {
 | |
| 	// TODO - Fix below tests to run on windows.
 | |
| 	if runtime.GOOS == globalWindowsOSName {
 | |
| 		return
 | |
| 	}
 | |
| 	s.testRPCStorageDisksInfo(t)
 | |
| 	s.testRPCStorageVolOps(t)
 | |
| 	s.testRPCStorageFileOps(t)
 | |
| 	s.testRPCStorageListDir(t)
 | |
| }
 | |
| 
 | |
| // Test storage disks info.
 | |
| func (s *TestRPCStorageSuite) testRPCStorageDisksInfo(t *testing.T) {
 | |
| 	for _, storageDisk := range s.remoteDisks {
 | |
| 		diskInfo, err := storageDisk.DiskInfo()
 | |
| 		if err != nil {
 | |
| 			t.Error("Unable to initiate DiskInfo", err)
 | |
| 		}
 | |
| 		if diskInfo.Total == 0 {
 | |
| 			t.Error("Invalid diskInfo total")
 | |
| 		}
 | |
| 		if storageDisk.String() == "" {
 | |
| 			t.Error("String representation of storageAPI should not be empty")
 | |
| 		}
 | |
| 	}
 | |
| }
 | |
| 
 | |
| // Test storage vol operations.
 | |
| func (s *TestRPCStorageSuite) testRPCStorageVolOps(t *testing.T) {
 | |
| 	for _, storageDisk := range s.remoteDisks {
 | |
| 		numVols := 0
 | |
| 		err := storageDisk.MakeVol("myvol")
 | |
| 		if err != nil {
 | |
| 			t.Error("Unable to initiate MakeVol", err)
 | |
| 		}
 | |
| 		numVols++
 | |
| 		volInfo, err := storageDisk.StatVol("myvol")
 | |
| 		if err != nil {
 | |
| 			t.Error("Unable to initiate StatVol", err)
 | |
| 		}
 | |
| 		if volInfo.Name != "myvol" {
 | |
| 			t.Errorf("Expected `myvol` found %s instead", volInfo.Name)
 | |
| 		}
 | |
| 		if volInfo.Created.IsZero() {
 | |
| 			t.Error("Expected created time to be non zero")
 | |
| 		}
 | |
| 
 | |
| 		for i := 0; i < 10; i++ {
 | |
| 			err = storageDisk.MakeVol(fmt.Sprintf("myvol-%d", i))
 | |
| 			if err != nil {
 | |
| 				t.Error("Unable to initiate MakeVol", err)
 | |
| 			}
 | |
| 			numVols++
 | |
| 		}
 | |
| 		vols, err := storageDisk.ListVols()
 | |
| 		if err != nil {
 | |
| 			t.Error("Unable to initiate ListVol")
 | |
| 		}
 | |
| 		if len(vols) != numVols {
 | |
| 			t.Errorf("Expected %d volumes but found only %d", numVols, len(vols))
 | |
| 		}
 | |
| 
 | |
| 		for i := 0; i < 10; i++ {
 | |
| 			err = storageDisk.DeleteVol(fmt.Sprintf("myvol-%d", i))
 | |
| 			if err != nil {
 | |
| 				t.Error("Unable to initiate DeleteVol", err)
 | |
| 			}
 | |
| 		}
 | |
| 
 | |
| 		err = storageDisk.DeleteVol("myvol")
 | |
| 		if err != nil {
 | |
| 			t.Error("Unable to initiate DeleteVol", err)
 | |
| 		}
 | |
| 		vols, err = storageDisk.ListVols()
 | |
| 		if err != nil {
 | |
| 			t.Error("Unable to initiate ListVol")
 | |
| 		}
 | |
| 		if len(vols) > 0 {
 | |
| 			t.Errorf("Expected no volumes but found %d", len(vols))
 | |
| 		}
 | |
| 	}
 | |
| }
 | |
| 
 | |
| // Tests all file operations.
 | |
| func (s *TestRPCStorageSuite) testRPCStorageFileOps(t *testing.T) {
 | |
| 	for _, storageDisk := range s.remoteDisks {
 | |
| 		err := storageDisk.MakeVol("myvol")
 | |
| 		if err != nil {
 | |
| 			t.Error("Unable to initiate MakeVol", err)
 | |
| 		}
 | |
| 		err = storageDisk.PrepareFile("myvol", "file1", int64(len([]byte("Hello, world"))))
 | |
| 		if err != nil {
 | |
| 			t.Error("Unable to initiate AppendFile", err)
 | |
| 		}
 | |
| 		err = storageDisk.AppendFile("myvol", "file1", []byte("Hello, world"))
 | |
| 		if err != nil {
 | |
| 			t.Error("Unable to initiate AppendFile", err)
 | |
| 		}
 | |
| 		fi, err := storageDisk.StatFile("myvol", "file1")
 | |
| 		if err != nil {
 | |
| 			t.Error("Unable to initiate StatFile", err)
 | |
| 		}
 | |
| 		if fi.Name != "file1" {
 | |
| 			t.Errorf("Expected `file1` but got %s", fi.Name)
 | |
| 		}
 | |
| 		if fi.Volume != "myvol" {
 | |
| 			t.Errorf("Expected `myvol` but got %s", fi.Volume)
 | |
| 		}
 | |
| 		if fi.Size != 12 {
 | |
| 			t.Errorf("Expected 12 but got %d", fi.Size)
 | |
| 		}
 | |
| 		if !fi.Mode.IsRegular() {
 | |
| 			t.Error("Expected file to be regular found", fi.Mode)
 | |
| 		}
 | |
| 		if fi.ModTime.IsZero() {
 | |
| 			t.Error("Expected created time to be non zero")
 | |
| 		}
 | |
| 		buf, err := storageDisk.ReadAll("myvol", "file1")
 | |
| 		if err != nil {
 | |
| 			t.Error("Unable to initiate ReadAll", err)
 | |
| 		}
 | |
| 		if !bytes.Equal(buf, []byte("Hello, world")) {
 | |
| 			t.Errorf("Expected `Hello, world`, got %s", string(buf))
 | |
| 		}
 | |
| 		buf1 := make([]byte, 5)
 | |
| 		n, err := storageDisk.ReadFile("myvol", "file1", 4, buf1, nil)
 | |
| 		if err != nil {
 | |
| 			t.Error("Unable to initiate ReadFile", err)
 | |
| 		}
 | |
| 		if n != 5 {
 | |
| 			t.Errorf("Expected `5`, got %d", n)
 | |
| 		}
 | |
| 		if !bytes.Equal(buf[4:9], buf1) {
 | |
| 			t.Errorf("Expected %s, got %s", string(buf[4:9]), string(buf1))
 | |
| 		}
 | |
| 
 | |
| 		blakeHash := func(b []byte) []byte {
 | |
| 			k := blake2b.Sum512(b)
 | |
| 			return k[:]
 | |
| 		}
 | |
| 		verifier := NewBitrotVerifier(BLAKE2b512, blakeHash(buf))
 | |
| 		buf2 := make([]byte, 2)
 | |
| 		n, err = storageDisk.ReadFile("myvol", "file1", 1, buf2, verifier)
 | |
| 		if err != nil {
 | |
| 			t.Error("Error in ReadFile with bitrot verification", err)
 | |
| 		}
 | |
| 		if n != 2 {
 | |
| 			t.Errorf("Expected `2`, got %d", n)
 | |
| 		}
 | |
| 		if !bytes.Equal(buf[1:3], buf2) {
 | |
| 			t.Errorf("Expected %s, got %s", string(buf[1:3]), string(buf2))
 | |
| 		}
 | |
| 
 | |
| 		err = storageDisk.RenameFile("myvol", "file1", "myvol", "file2")
 | |
| 		if err != nil {
 | |
| 			t.Error("Unable to initiate RenameFile", err)
 | |
| 		}
 | |
| 		err = storageDisk.DeleteFile("myvol", "file2")
 | |
| 		if err != nil {
 | |
| 			t.Error("Unable to initiate DeleteFile", err)
 | |
| 		}
 | |
| 		err = storageDisk.DeleteVol("myvol")
 | |
| 		if err != nil {
 | |
| 			t.Error("Unable to initiate DeleteVol", err)
 | |
| 		}
 | |
| 	}
 | |
| }
 | |
| 
 | |
| // Tests for ListDirHandler.
 | |
| func (s *TestRPCStorageSuite) testRPCStorageListDir(t *testing.T) {
 | |
| 	for _, storageDisk := range s.remoteDisks {
 | |
| 		err := storageDisk.MakeVol("myvol")
 | |
| 		if err != nil {
 | |
| 			t.Error("Unable to initiate MakeVol", err)
 | |
| 		}
 | |
| 		dirCount := 10
 | |
| 		for i := 0; i < dirCount; i++ {
 | |
| 			err = storageDisk.MakeVol(fmt.Sprintf("myvol/mydir-%d", i))
 | |
| 			if err != nil {
 | |
| 				t.Error("Unable to initiate MakeVol", err)
 | |
| 			}
 | |
| 		}
 | |
| 		dirs, err := storageDisk.ListDir("myvol", "")
 | |
| 		if err != nil {
 | |
| 			t.Error(err)
 | |
| 		}
 | |
| 		if len(dirs) != dirCount {
 | |
| 			t.Errorf("Expected %d directories but found only %d", dirCount, len(dirs))
 | |
| 		}
 | |
| 		for i := 0; i < dirCount; i++ {
 | |
| 			err = storageDisk.DeleteVol(fmt.Sprintf("myvol/mydir-%d", i))
 | |
| 			if err != nil {
 | |
| 				t.Error("Unable to initiate DeleteVol", err)
 | |
| 			}
 | |
| 		}
 | |
| 		dirs, err = storageDisk.ListDir("myvol", "")
 | |
| 		if err != nil {
 | |
| 			t.Error(err)
 | |
| 		}
 | |
| 		if len(dirs) != 0 {
 | |
| 			t.Errorf("Expected no directories but found %d", dirCount)
 | |
| 		}
 | |
| 
 | |
| 		err = storageDisk.DeleteVol("myvol")
 | |
| 		if err != nil {
 | |
| 			t.Error("Unable to initiate DeleteVol", err)
 | |
| 		}
 | |
| 		vols, err := storageDisk.ListVols()
 | |
| 		if err != nil {
 | |
| 			t.Error(err)
 | |
| 		}
 | |
| 		if len(vols) != 0 {
 | |
| 			t.Errorf("Expected no volumes but found %d", dirCount)
 | |
| 		}
 | |
| 	}
 | |
| }
 |