355 lines
8.5 KiB
Go
355 lines
8.5 KiB
Go
/*
|
|
* Minio Cloud Storage, (C) 2016 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 (
|
|
"flag"
|
|
"net/http"
|
|
"os"
|
|
"runtime"
|
|
"testing"
|
|
|
|
"github.com/minio/cli"
|
|
)
|
|
|
|
func TestGetListenIPs(t *testing.T) {
|
|
testCases := []struct {
|
|
addr string
|
|
port string
|
|
shouldPass bool
|
|
}{
|
|
{"localhost", "9000", true},
|
|
{"", "9000", true},
|
|
{"", "", false},
|
|
}
|
|
for _, test := range testCases {
|
|
var addr string
|
|
// Please keep this we need to do this because
|
|
// of odd https://play.golang.org/p/4dMPtM6Wdd
|
|
// implementation issue.
|
|
if test.port != "" {
|
|
addr = test.addr + ":" + test.port
|
|
}
|
|
hosts, port, err := getListenIPs(&http.Server{
|
|
Addr: addr,
|
|
})
|
|
if !test.shouldPass && err == nil {
|
|
t.Fatalf("Test should fail but succeeded %s", err)
|
|
}
|
|
if test.shouldPass && err != nil {
|
|
t.Fatalf("Test should succeed but failed %s", err)
|
|
}
|
|
if test.shouldPass {
|
|
if port != test.port {
|
|
t.Errorf("Test expected %s, got %s", test.port, port)
|
|
}
|
|
if len(hosts) == 0 {
|
|
t.Errorf("Test unexpected value hosts cannot be empty %#v", test)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestFinalizeEndpoints(t *testing.T) {
|
|
testCases := []struct {
|
|
tls bool
|
|
addr string
|
|
}{
|
|
{false, ":80"},
|
|
{true, ":80"},
|
|
{false, "localhost:80"},
|
|
{true, "localhost:80"},
|
|
}
|
|
|
|
for i, test := range testCases {
|
|
endPoints := finalizeEndpoints(test.tls, &http.Server{Addr: test.addr})
|
|
if len(endPoints) <= 0 {
|
|
t.Errorf("Test case %d returned with no API end points for %s",
|
|
i+1, test.addr)
|
|
}
|
|
}
|
|
}
|
|
|
|
// Tests all the expected input disks for function checkSufficientDisks.
|
|
func TestCheckSufficientDisks(t *testing.T) {
|
|
var xlDisks []string
|
|
if runtime.GOOS == "windows" {
|
|
xlDisks = []string{
|
|
"C:\\mnt\\backend1",
|
|
"C:\\mnt\\backend2",
|
|
"C:\\mnt\\backend3",
|
|
"C:\\mnt\\backend4",
|
|
"C:\\mnt\\backend5",
|
|
"C:\\mnt\\backend6",
|
|
"C:\\mnt\\backend7",
|
|
"C:\\mnt\\backend8",
|
|
"C:\\mnt\\backend9",
|
|
"C:\\mnt\\backend10",
|
|
"C:\\mnt\\backend11",
|
|
"C:\\mnt\\backend12",
|
|
"C:\\mnt\\backend13",
|
|
"C:\\mnt\\backend14",
|
|
"C:\\mnt\\backend15",
|
|
"C:\\mnt\\backend16",
|
|
"C:\\mnt\\backend17",
|
|
}
|
|
} else {
|
|
xlDisks = []string{
|
|
"/mnt/backend1",
|
|
"/mnt/backend2",
|
|
"/mnt/backend3",
|
|
"/mnt/backend4",
|
|
"/mnt/backend5",
|
|
"/mnt/backend6",
|
|
"/mnt/backend7",
|
|
"/mnt/backend8",
|
|
"/mnt/backend9",
|
|
"/mnt/backend10",
|
|
"/mnt/backend11",
|
|
"/mnt/backend12",
|
|
"/mnt/backend13",
|
|
"/mnt/backend14",
|
|
"/mnt/backend15",
|
|
"/mnt/backend16",
|
|
"/mnt/backend17",
|
|
}
|
|
}
|
|
// List of test cases fo sufficient disk verification.
|
|
testCases := []struct {
|
|
disks []string
|
|
expectedErr error
|
|
}{
|
|
// Even number of disks '6'.
|
|
{
|
|
xlDisks[0:6],
|
|
nil,
|
|
},
|
|
// Even number of disks '12'.
|
|
{
|
|
xlDisks[0:12],
|
|
nil,
|
|
},
|
|
// Even number of disks '16'.
|
|
{
|
|
xlDisks[0:16],
|
|
nil,
|
|
},
|
|
// Larger than maximum number of disks > 16.
|
|
{
|
|
xlDisks,
|
|
errXLMaxDisks,
|
|
},
|
|
// Lesser than minimum number of disks < 6.
|
|
{
|
|
xlDisks[0:3],
|
|
errXLMinDisks,
|
|
},
|
|
// Odd number of disks, not divisible by '2'.
|
|
{
|
|
append(xlDisks[0:10], xlDisks[11]),
|
|
errXLNumDisks,
|
|
},
|
|
}
|
|
|
|
// Validates different variations of input disks.
|
|
for i, testCase := range testCases {
|
|
endpoints, err := parseStorageEndpoints(testCase.disks)
|
|
if err != nil {
|
|
t.Fatalf("Unexpected error %s", err)
|
|
}
|
|
if checkSufficientDisks(endpoints) != testCase.expectedErr {
|
|
t.Errorf("Test %d expected to pass for disks %s", i+1, testCase.disks)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestParseStorageEndpoints(t *testing.T) {
|
|
testCases := []struct {
|
|
globalMinioHost string
|
|
host string
|
|
expectedErr error
|
|
}{
|
|
{"", "http://localhost/export", nil},
|
|
{"testhost", "http://localhost/export", errInvalidArgument},
|
|
{"", "http://localhost:9000/export", errInvalidArgument},
|
|
{"testhost", "http://localhost:9000/export", nil},
|
|
}
|
|
for i, test := range testCases {
|
|
globalMinioHost = test.globalMinioHost
|
|
_, err := parseStorageEndpoints([]string{test.host})
|
|
if err != test.expectedErr {
|
|
t.Errorf("Test %d : got %v, expected %v", i+1, err, test.expectedErr)
|
|
}
|
|
}
|
|
// Should be reset back to "" so that we don't affect other tests.
|
|
globalMinioHost = ""
|
|
}
|
|
|
|
func TestCheckEndpointsSyntax(t *testing.T) {
|
|
var testCases []string
|
|
if runtime.GOOS == "windows" {
|
|
testCases = []string{
|
|
"\\export",
|
|
"D:\\export",
|
|
"D:\\",
|
|
"D:",
|
|
"\\",
|
|
}
|
|
} else {
|
|
testCases = []string{
|
|
"/export",
|
|
}
|
|
}
|
|
testCasesCommon := []string{
|
|
"export",
|
|
"http://localhost/export",
|
|
"https://localhost/export",
|
|
}
|
|
testCases = append(testCases, testCasesCommon...)
|
|
for _, disk := range testCases {
|
|
eps, err := parseStorageEndpoints([]string{disk})
|
|
if err != nil {
|
|
t.Error(disk, err)
|
|
continue
|
|
}
|
|
// This will fatalIf() if endpoint is invalid.
|
|
checkEndpointsSyntax(eps, []string{disk})
|
|
}
|
|
}
|
|
|
|
func TestCheckServerSyntax(t *testing.T) {
|
|
app := cli.NewApp()
|
|
app.Commands = []cli.Command{serverCmd}
|
|
serverFlagSet := flag.NewFlagSet("server", 0)
|
|
cli.NewContext(app, serverFlagSet, nil)
|
|
disksGen := func(n int) []string {
|
|
disks, err := getRandomDisks(n)
|
|
if err != nil {
|
|
t.Fatalf("Unable to initialie disks %s", err)
|
|
}
|
|
return disks
|
|
}
|
|
testCases := [][]string{
|
|
disksGen(1),
|
|
disksGen(4),
|
|
disksGen(8),
|
|
disksGen(16),
|
|
}
|
|
for i, disks := range testCases {
|
|
err := serverFlagSet.Parse(disks)
|
|
if err != nil {
|
|
t.Errorf("Test %d failed to parse arguments %s", i+1, disks)
|
|
}
|
|
defer removeRoots(disks)
|
|
endpoints, err := parseStorageEndpoints(disks)
|
|
if err != nil {
|
|
t.Fatalf("Test %d : Unexpected error %s", i+1, err)
|
|
}
|
|
checkEndpointsSyntax(endpoints, disks)
|
|
_, err = initStorageDisks(endpoints, nil)
|
|
if err != nil {
|
|
t.Errorf("Test %d : disk init failed : %s", i+1, err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestIsDistributedSetup(t *testing.T) {
|
|
var testCases []struct {
|
|
disks []string
|
|
result bool
|
|
}
|
|
if runtime.GOOS == "windows" {
|
|
testCases = []struct {
|
|
disks []string
|
|
result bool
|
|
}{
|
|
{[]string{`http://4.4.4.4/c:\mnt\disk1`, `http://4.4.4.4/c:\mnt\disk2`}, true},
|
|
{[]string{`http://4.4.4.4/c:\mnt\disk1`, `http://127.0.0.1/c:\mnt\disk2`}, true},
|
|
{[]string{`c:\mnt\disk1`, `c:\mnt\disk2`}, false},
|
|
}
|
|
} else {
|
|
testCases = []struct {
|
|
disks []string
|
|
result bool
|
|
}{
|
|
{[]string{"http://4.4.4.4/mnt/disk1", "http://4.4.4.4/mnt/disk2"}, true},
|
|
{[]string{"http://4.4.4.4/mnt/disk1", "http://127.0.0.1/mnt/disk2"}, true},
|
|
{[]string{"/mnt/disk1", "/mnt/disk2"}, false},
|
|
}
|
|
}
|
|
for i, test := range testCases {
|
|
endpoints, err := parseStorageEndpoints(test.disks)
|
|
if err != nil {
|
|
t.Fatalf("Test %d: Unexpected error: %s", i+1, err)
|
|
}
|
|
res := isDistributedSetup(endpoints)
|
|
if res != test.result {
|
|
t.Errorf("Test %d: expected result %t but received %t", i+1, test.result, res)
|
|
}
|
|
}
|
|
|
|
// Test cases when globalMinioHost is set
|
|
globalMinioHost = "testhost"
|
|
testCases = []struct {
|
|
disks []string
|
|
result bool
|
|
}{
|
|
{[]string{"http://127.0.0.1:9001/mnt/disk1", "http://127.0.0.1:9002/mnt/disk2", "http://127.0.0.1:9003/mnt/disk3", "http://127.0.0.1:9004/mnt/disk4"}, true},
|
|
{[]string{"/mnt/disk1", "/mnt/disk2"}, false},
|
|
}
|
|
|
|
for i, test := range testCases {
|
|
endpoints, err := parseStorageEndpoints(test.disks)
|
|
if err != nil {
|
|
t.Fatalf("Test %d: Unexpected error: %s", i+1, err)
|
|
}
|
|
res := isDistributedSetup(endpoints)
|
|
if res != test.result {
|
|
t.Errorf("Test %d: expected result %t but received %t", i+1, test.result, res)
|
|
}
|
|
}
|
|
// Reset so that we don't affect other tests.
|
|
globalMinioHost = ""
|
|
}
|
|
|
|
func TestInitServerConfig(t *testing.T) {
|
|
ctx := &cli.Context{}
|
|
root, err := newTestConfig("us-east-1")
|
|
if err != nil {
|
|
t.Fatal("Failed to set up test config")
|
|
}
|
|
defer removeAll(root)
|
|
|
|
testCases := []struct {
|
|
envVar string
|
|
val string
|
|
}{
|
|
{"MINIO_MAXCONN", "10"},
|
|
{"MINIO_CACHE_SIZE", "42MB"},
|
|
{"MINIO_CACHE_EXPIRY", "2h45m"},
|
|
{"MINIO_ACCESS_KEY", "abcd1"},
|
|
{"MINIO_SECRET_KEY", "abcd12345"},
|
|
}
|
|
for i, test := range testCases {
|
|
tErr := os.Setenv(test.envVar, test.val)
|
|
if tErr != nil {
|
|
t.Fatalf("Test %d failed with %v", i+1, tErr)
|
|
}
|
|
initServerConfig(ctx)
|
|
}
|
|
}
|