/* * 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 "testing" // Tests initialization of remote controller clients. func TestInitRemoteControllerClients(t *testing.T) { rootPath, err := newTestConfig("us-east-1") if err != nil { t.Fatal("Unable to initialize config", err) } defer removeAll(rootPath) testCases := []struct { srvCmdConfig serverCmdConfig totalClients int }{ // Test - 1 no allocation if server config is not distributed XL. { srvCmdConfig: serverCmdConfig{ isDistXL: false, }, totalClients: 0, }, // Test - 2 two clients allocated with 4 disks with 2 disks on same node each. { srvCmdConfig: serverCmdConfig{ isDistXL: true, disks: []string{ "10.1.10.1:/mnt/disk1", "10.1.10.1:/mnt/disk2", "10.1.10.2:/mnt/disk3", "10.1.10.2:/mnt/disk4", }, }, totalClients: 2, }, // Test - 3 4 clients allocated with 4 disks with 1 disk on each node. { srvCmdConfig: serverCmdConfig{ isDistXL: true, disks: []string{ "10.1.10.1:/mnt/disk1", "10.1.10.2:/mnt/disk2", "10.1.10.3:/mnt/disk3", "10.1.10.4:/mnt/disk4", }, }, totalClients: 4, }, // Test - 4 2 clients allocated with 4 disks with 1 disk ignored. { srvCmdConfig: serverCmdConfig{ isDistXL: true, disks: []string{ "10.1.10.1:/mnt/disk1", "10.1.10.2:/mnt/disk2", "10.1.10.3:/mnt/disk3", "10.1.10.4:/mnt/disk4", }, ignoredDisks: []string{ "10.1.10.1:/mnt/disk1", }, }, totalClients: 3, }, } // Evaluate and validate all test cases. for i, testCase := range testCases { rclients := initRemoteControllerClients(testCase.srvCmdConfig) if len(rclients) != testCase.totalClients { t.Errorf("Test %d, Expected %d, got %d RPC clients.", i+1, testCase.totalClients, len(rclients)) } } }