mirror of https://github.com/minio/minio.git
				
				
				
			
		
			
				
	
	
		
			202 lines
		
	
	
		
			6.5 KiB
		
	
	
	
		
			Go
		
	
	
	
			
		
		
	
	
			202 lines
		
	
	
		
			6.5 KiB
		
	
	
	
		
			Go
		
	
	
	
| /*
 | |
|  * MinIO 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 cmd
 | |
| 
 | |
| import (
 | |
| 	"context"
 | |
| 	"reflect"
 | |
| 	"testing"
 | |
| )
 | |
| 
 | |
| // Tests caclculating disk count.
 | |
| func TestDiskCount(t *testing.T) {
 | |
| 	testCases := []struct {
 | |
| 		disks     []StorageAPI
 | |
| 		diskCount int
 | |
| 	}{
 | |
| 		// Test case - 1
 | |
| 		{
 | |
| 			disks:     []StorageAPI{&xlStorage{}, &xlStorage{}, &xlStorage{}, &xlStorage{}},
 | |
| 			diskCount: 4,
 | |
| 		},
 | |
| 		// Test case - 2
 | |
| 		{
 | |
| 			disks:     []StorageAPI{nil, &xlStorage{}, &xlStorage{}, &xlStorage{}},
 | |
| 			diskCount: 3,
 | |
| 		},
 | |
| 	}
 | |
| 	for i, testCase := range testCases {
 | |
| 		cdiskCount := diskCount(testCase.disks)
 | |
| 		if cdiskCount != testCase.diskCount {
 | |
| 			t.Errorf("Test %d: Expected %d, got %d", i+1, testCase.diskCount, cdiskCount)
 | |
| 		}
 | |
| 	}
 | |
| }
 | |
| 
 | |
| // Test for reduceErrs, reduceErr reduces collection
 | |
| // of errors into a single maximal error with in the list.
 | |
| func TestReduceErrs(t *testing.T) {
 | |
| 	// List all of all test cases to validate various cases of reduce errors.
 | |
| 	testCases := []struct {
 | |
| 		errs        []error
 | |
| 		ignoredErrs []error
 | |
| 		err         error
 | |
| 	}{
 | |
| 		// Validate if have reduced properly.
 | |
| 		{[]error{
 | |
| 			errDiskNotFound,
 | |
| 			errDiskNotFound,
 | |
| 			errDiskFull,
 | |
| 		}, []error{}, errErasureReadQuorum},
 | |
| 		// Validate if have no consensus.
 | |
| 		{[]error{
 | |
| 			errDiskFull,
 | |
| 			errDiskNotFound,
 | |
| 			nil, nil,
 | |
| 		}, []error{}, errErasureReadQuorum},
 | |
| 		// Validate if have consensus and errors ignored.
 | |
| 		{[]error{
 | |
| 			errVolumeNotFound,
 | |
| 			errVolumeNotFound,
 | |
| 			errVolumeNotFound,
 | |
| 			errVolumeNotFound,
 | |
| 			errVolumeNotFound,
 | |
| 			errDiskNotFound,
 | |
| 			errDiskNotFound,
 | |
| 		}, []error{errDiskNotFound}, errVolumeNotFound},
 | |
| 		{[]error{}, []error{}, errErasureReadQuorum},
 | |
| 		{[]error{errFileNotFound, errFileNotFound, errFileNotFound,
 | |
| 			errFileNotFound, errFileNotFound, nil, nil, nil, nil, nil},
 | |
| 			nil, nil},
 | |
| 	}
 | |
| 	// Validates list of all the testcases for returning valid errors.
 | |
| 	for i, testCase := range testCases {
 | |
| 		gotErr := reduceReadQuorumErrs(context.Background(), testCase.errs, testCase.ignoredErrs, 5)
 | |
| 		if gotErr != testCase.err {
 | |
| 			t.Errorf("Test %d : expected %s, got %s", i+1, testCase.err, gotErr)
 | |
| 		}
 | |
| 		gotNewErr := reduceWriteQuorumErrs(context.Background(), testCase.errs, testCase.ignoredErrs, 6)
 | |
| 		if gotNewErr != errErasureWriteQuorum {
 | |
| 			t.Errorf("Test %d : expected %s, got %s", i+1, errErasureWriteQuorum, gotErr)
 | |
| 		}
 | |
| 	}
 | |
| }
 | |
| 
 | |
| // TestHashOrder - test order of ints in array
 | |
| func TestHashOrder(t *testing.T) {
 | |
| 	testCases := []struct {
 | |
| 		objectName  string
 | |
| 		hashedOrder []int
 | |
| 	}{
 | |
| 		// cases which should pass the test.
 | |
| 		// passing in valid object name.
 | |
| 		{"object", []int{14, 15, 16, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13}},
 | |
| 		{"The Shining Script <v1>.pdf", []int{16, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15}},
 | |
| 		{"Cost Benefit Analysis (2009-2010).pptx", []int{15, 16, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14}},
 | |
| 		{"117Gn8rfHL2ACARPAhaFd0AGzic9pUbIA/5OCn5A", []int{3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 1, 2}},
 | |
| 		{"SHØRT", []int{11, 12, 13, 14, 15, 16, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10}},
 | |
| 		{"There are far too many object names, and far too few bucket names!", []int{15, 16, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14}},
 | |
| 		{"a/b/c/", []int{3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 1, 2}},
 | |
| 		{"/a/b/c", []int{6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 1, 2, 3, 4, 5}},
 | |
| 		{string([]byte{0xff, 0xfe, 0xfd}), []int{15, 16, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14}},
 | |
| 	}
 | |
| 
 | |
| 	// Tests hashing order to be consistent.
 | |
| 	for i, testCase := range testCases {
 | |
| 		hashedOrder := hashOrder(testCase.objectName, 16)
 | |
| 		if !reflect.DeepEqual(testCase.hashedOrder, hashedOrder) {
 | |
| 			t.Errorf("Test case %d: Expected \"%v\" but failed \"%v\"", i+1, testCase.hashedOrder, hashedOrder)
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	// Tests hashing order to fail for when order is '-1'.
 | |
| 	if hashedOrder := hashOrder("This will fail", -1); hashedOrder != nil {
 | |
| 		t.Errorf("Test: Expect \"nil\" but failed \"%#v\"", hashedOrder)
 | |
| 	}
 | |
| 
 | |
| 	if hashedOrder := hashOrder("This will fail", 0); hashedOrder != nil {
 | |
| 		t.Errorf("Test: Expect \"nil\" but failed \"%#v\"", hashedOrder)
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func TestShuffleDisks(t *testing.T) {
 | |
| 	ctx, cancel := context.WithCancel(context.Background())
 | |
| 	defer cancel()
 | |
| 
 | |
| 	nDisks := 16
 | |
| 	disks, err := getRandomDisks(nDisks)
 | |
| 	if err != nil {
 | |
| 		t.Fatal(err)
 | |
| 	}
 | |
| 	objLayer, _, err := initObjectLayer(ctx, mustGetZoneEndpoints(disks...))
 | |
| 	if err != nil {
 | |
| 		removeRoots(disks)
 | |
| 		t.Fatal(err)
 | |
| 	}
 | |
| 	defer removeRoots(disks)
 | |
| 	z := objLayer.(*erasureServerPools)
 | |
| 	testShuffleDisks(t, z)
 | |
| }
 | |
| 
 | |
| // Test shuffleDisks which returns shuffled slice of disks for their actual distribution.
 | |
| func testShuffleDisks(t *testing.T, z *erasureServerPools) {
 | |
| 	disks := z.serverPools[0].GetDisks(0)()
 | |
| 	distribution := []int{16, 14, 12, 10, 8, 6, 4, 2, 1, 3, 5, 7, 9, 11, 13, 15}
 | |
| 	shuffledDisks := shuffleDisks(disks, distribution)
 | |
| 	// From the "distribution" above you can notice that:
 | |
| 	// 1st data block is in the 9th disk (i.e distribution index 8)
 | |
| 	// 2nd data block is in the 8th disk (i.e distribution index 7) and so on.
 | |
| 	if shuffledDisks[0] != disks[8] ||
 | |
| 		shuffledDisks[1] != disks[7] ||
 | |
| 		shuffledDisks[2] != disks[9] ||
 | |
| 		shuffledDisks[3] != disks[6] ||
 | |
| 		shuffledDisks[4] != disks[10] ||
 | |
| 		shuffledDisks[5] != disks[5] ||
 | |
| 		shuffledDisks[6] != disks[11] ||
 | |
| 		shuffledDisks[7] != disks[4] ||
 | |
| 		shuffledDisks[8] != disks[12] ||
 | |
| 		shuffledDisks[9] != disks[3] ||
 | |
| 		shuffledDisks[10] != disks[13] ||
 | |
| 		shuffledDisks[11] != disks[2] ||
 | |
| 		shuffledDisks[12] != disks[14] ||
 | |
| 		shuffledDisks[13] != disks[1] ||
 | |
| 		shuffledDisks[14] != disks[15] ||
 | |
| 		shuffledDisks[15] != disks[0] {
 | |
| 		t.Errorf("shuffleDisks returned incorrect order.")
 | |
| 	}
 | |
| }
 | |
| 
 | |
| // TestEvalDisks tests the behavior of evalDisks
 | |
| func TestEvalDisks(t *testing.T) {
 | |
| 	ctx, cancel := context.WithCancel(context.Background())
 | |
| 	defer cancel()
 | |
| 
 | |
| 	nDisks := 16
 | |
| 	disks, err := getRandomDisks(nDisks)
 | |
| 	if err != nil {
 | |
| 		t.Fatal(err)
 | |
| 	}
 | |
| 	objLayer, _, err := initObjectLayer(ctx, mustGetZoneEndpoints(disks...))
 | |
| 	if err != nil {
 | |
| 		removeRoots(disks)
 | |
| 		t.Fatal(err)
 | |
| 	}
 | |
| 	defer removeRoots(disks)
 | |
| 	z := objLayer.(*erasureServerPools)
 | |
| 	testShuffleDisks(t, z)
 | |
| }
 |