mirror of https://github.com/k3s-io/k3s
363 lines
12 KiB
Go
363 lines
12 KiB
Go
/*
|
|
Copyright 2015 The Kubernetes Authors.
|
|
|
|
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 iscsi
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"reflect"
|
|
"testing"
|
|
|
|
"k8s.io/kubernetes/pkg/util/mount"
|
|
"k8s.io/kubernetes/pkg/volume"
|
|
)
|
|
|
|
func TestGetDevicePrefixRefCount(t *testing.T) {
|
|
fm := &mount.FakeMounter{
|
|
MountPoints: []mount.MountPoint{
|
|
{Device: "/dev/sdb",
|
|
Path: "/127.0.0.1:3260-iqn.2014-12.com.example:test.tgt00-lun-0"},
|
|
{Device: "/dev/sdb",
|
|
Path: "/127.0.0.1:3260-iqn.2014-12.com.example:test.tgt00-lun-1"},
|
|
{Device: "/dev/sdb",
|
|
Path: "/127.0.0.1:3260-iqn.2014-12.com.example:test.tgt00-lun-2"},
|
|
{Device: "/dev/sdb",
|
|
Path: "/127.0.0.1:3260-iqn.2014-12.com.example:test.tgt00-lun-3"},
|
|
},
|
|
}
|
|
|
|
tests := []struct {
|
|
devicePrefix string
|
|
expectedRefs int
|
|
}{
|
|
{
|
|
"/127.0.0.1:3260-iqn.2014-12.com.example:test.tgt00",
|
|
4,
|
|
},
|
|
}
|
|
|
|
for i, test := range tests {
|
|
if refs, err := getDevicePrefixRefCount(fm, test.devicePrefix); err != nil || test.expectedRefs != refs {
|
|
t.Errorf("%d. GetDevicePrefixRefCount(%s) = %d, %v; expected %d, nil", i, test.devicePrefix, refs, err, test.expectedRefs)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestExtractDeviceAndPrefix(t *testing.T) {
|
|
devicePath := "127.0.0.1:3260-iqn.2014-12.com.example:test.tgt00"
|
|
mountPrefix := "/var/lib/kubelet/plugins/kubernetes.io/iscsi/iface-default/" + devicePath
|
|
lun := "-lun-0"
|
|
device, prefix, err := extractDeviceAndPrefix(mountPrefix + lun)
|
|
if err != nil || device != (devicePath+lun) || prefix != mountPrefix {
|
|
t.Errorf("extractDeviceAndPrefix: expected %s and %s, got %v %s and %s", devicePath+lun, mountPrefix, err, device, prefix)
|
|
}
|
|
}
|
|
|
|
func TestExtractIface(t *testing.T) {
|
|
ifaceName := "default"
|
|
devicePath := "127.0.0.1:3260-iqn.2014-12.com.example:test.tgt00-lun-0"
|
|
iface, found := extractIface("/var/lib/kubelet/plugins/kubernetes.io/iscsi/iface-" + ifaceName + "/" + devicePath)
|
|
if !found || iface != ifaceName {
|
|
t.Errorf("extractIface: expected %s and %t, got %s and %t", ifaceName, true, iface, found)
|
|
}
|
|
iface, found = extractIface("/var/lib/kubelet/plugins/kubernetes.io/iscsi/" + devicePath)
|
|
if found || iface != "" {
|
|
t.Errorf("extractIface: expected %s and %t, got %s and %t", "", false, iface, found)
|
|
}
|
|
}
|
|
|
|
func TestExtractPortalAndIqn(t *testing.T) {
|
|
devicePath := "127.0.0.1:3260-iqn.2014-12.com.example:test.tgt00-lun-0"
|
|
portal, iqn, err := extractPortalAndIqn(devicePath)
|
|
if err != nil || portal != "127.0.0.1:3260" || iqn != "iqn.2014-12.com.example:test.tgt00" {
|
|
t.Errorf("extractPortalAndIqn: got %v %s %s", err, portal, iqn)
|
|
}
|
|
devicePath = "127.0.0.1:3260-eui.02004567A425678D-lun-0"
|
|
portal, iqn, err = extractPortalAndIqn(devicePath)
|
|
if err != nil || portal != "127.0.0.1:3260" || iqn != "eui.02004567A425678D" {
|
|
t.Errorf("extractPortalAndIqn: got %v %s %s", err, portal, iqn)
|
|
}
|
|
}
|
|
|
|
func TestRemoveDuplicate(t *testing.T) {
|
|
dupPortals := []string{"127.0.0.1:3260", "127.0.0.1:3260", "127.0.0.100:3260"}
|
|
portals := removeDuplicate(dupPortals)
|
|
want := []string{"127.0.0.1:3260", "127.0.0.100:3260"}
|
|
if reflect.DeepEqual(portals, want) == false {
|
|
t.Errorf("removeDuplicate: want: %s, got: %s", want, portals)
|
|
}
|
|
}
|
|
|
|
func fakeOsStat(devicePath string) (fi os.FileInfo, err error) {
|
|
var cmd os.FileInfo
|
|
return cmd, nil
|
|
}
|
|
|
|
func fakeFilepathGlob(devicePath string) (globs []string, err error) {
|
|
return []string{devicePath}, nil
|
|
}
|
|
|
|
func fakeFilepathGlob2(devicePath string) (globs []string, err error) {
|
|
return []string{
|
|
"/dev/disk/by-path/pci-0000:00:00.0-ip-127.0.0.1:3260-iqn.2014-12.com.example:test.tgt00-lun-0",
|
|
}, nil
|
|
}
|
|
|
|
func TestExtractTransportname(t *testing.T) {
|
|
fakeIscsiadmOutput := []string{
|
|
"# BEGIN RECORD 2.0-873\n" +
|
|
"iface.iscsi_ifacename = default\n" +
|
|
"iface.transport_name = tcp\n" +
|
|
"iface.initiatorname = <empty>\n" +
|
|
"# END RECORD",
|
|
"# BEGIN RECORD 2.0-873\n" +
|
|
"iface.iscsi_ifacename = default\n" +
|
|
"iface.transport_name = cxgb4i\n" +
|
|
"iface.initiatorname = <empty>\n" +
|
|
"# END RECORD",
|
|
"# BEGIN RECORD 2.0-873\n" +
|
|
"iface.iscsi_ifacename = default\n" +
|
|
"iface.transport_name = <empty>\n" +
|
|
"iface.initiatorname = <empty>\n" +
|
|
"# END RECORD",
|
|
"# BEGIN RECORD 2.0-873\n" +
|
|
"iface.iscsi_ifacename = default\n" +
|
|
"iface.initiatorname = <empty>\n" +
|
|
"# END RECORD"}
|
|
transportName := extractTransportname(fakeIscsiadmOutput[0])
|
|
if transportName != "tcp" {
|
|
t.Errorf("extractTransportname: Could not extract correct iface.transport_name 'tcp', got %s", transportName)
|
|
}
|
|
transportName = extractTransportname(fakeIscsiadmOutput[1])
|
|
if transportName != "cxgb4i" {
|
|
t.Errorf("extractTransportname: Could not extract correct iface.transport_name 'cxgb4i', got %s", transportName)
|
|
}
|
|
transportName = extractTransportname(fakeIscsiadmOutput[2])
|
|
if transportName != "tcp" {
|
|
t.Errorf("extractTransportname: Could not extract correct iface.transport_name 'tcp', got %s", transportName)
|
|
}
|
|
transportName = extractTransportname(fakeIscsiadmOutput[3])
|
|
if transportName != "" {
|
|
t.Errorf("extractTransportname: Could not extract correct iface.transport_name '', got %s", transportName)
|
|
}
|
|
}
|
|
|
|
func TestWaitForPathToExist(t *testing.T) {
|
|
devicePath := []string{"/dev/disk/by-path/ip-127.0.0.1:3260-iqn.2014-12.com.example:test.tgt00-lun-0",
|
|
"/dev/disk/by-path/pci-*-ip-127.0.0.1:3260-iqn.2014-12.com.example:test.tgt00-lun-0"}
|
|
fpath := "/dev/disk/by-path/pci-0000:00:00.0-ip-127.0.0.1:3260-iqn.2014-12.com.example:test.tgt00-lun-0"
|
|
|
|
exist := waitForPathToExistInternal(&devicePath[0], 1, "tcp", fakeOsStat, filepath.Glob)
|
|
if exist == false {
|
|
t.Errorf("waitForPathToExist: could not find path %s", devicePath[0])
|
|
}
|
|
exist = waitForPathToExistInternal(&devicePath[0], 1, "fake_iface", fakeOsStat, filepath.Glob)
|
|
if exist != false {
|
|
t.Errorf("waitForPathToExist: wrong code path called for %s", devicePath[0])
|
|
}
|
|
|
|
exist = waitForPathToExistInternal(&devicePath[1], 1, "fake_iface", os.Stat, fakeFilepathGlob)
|
|
if exist == false {
|
|
t.Errorf("waitForPathToExist: could not find path %s", devicePath[1])
|
|
}
|
|
exist = waitForPathToExistInternal(&devicePath[1], 1, "tcp", os.Stat, fakeFilepathGlob)
|
|
if exist != false {
|
|
t.Errorf("waitForPathToExist: wrong code path called for %s", devicePath[1])
|
|
}
|
|
|
|
exist = waitForPathToExistInternal(&devicePath[1], 1, "fake_iface", os.Stat, fakeFilepathGlob2)
|
|
if devicePath[1] != fpath {
|
|
t.Errorf("waitForPathToExist: wrong code path called for %s", devicePath[1])
|
|
}
|
|
}
|
|
|
|
func TestParseIscsiadmShow(t *testing.T) {
|
|
fakeIscsiadmOutput1 := "# BEGIN RECORD 2.0-873\n" +
|
|
"iface.iscsi_ifacename = default\n" +
|
|
"iface.transport_name = tcp\n" +
|
|
"iface.initiatorname = <empty>\n" +
|
|
"iface.mtu = 0\n" +
|
|
"# END RECORD"
|
|
|
|
fakeIscsiadmOutput2 := "# BEGIN RECORD 2.0-873\n" +
|
|
"iface.iscsi_ifacename = default\n" +
|
|
"iface.transport_name = cxgb4i\n" +
|
|
"iface.initiatorname = <empty>\n" +
|
|
"iface.mtu = 0\n" +
|
|
"# END RECORD"
|
|
|
|
fakeIscsiadmOutput3 := "# BEGIN RECORD 2.0-873\n" +
|
|
"iface.iscsi_ifacename = custom\n" +
|
|
"iface.transport_name = <empty>\n" +
|
|
"iface.initiatorname = <empty>\n" +
|
|
"iface.mtu = 0\n" +
|
|
"# END RECORD"
|
|
|
|
fakeIscsiadmOutput4 := "iface.iscsi_ifacename=error"
|
|
fakeIscsiadmOutput5 := "iface.iscsi_ifacename + error"
|
|
|
|
expectedIscsiadmOutput1 := map[string]string{
|
|
"iface.transport_name": "tcp",
|
|
"iface.mtu": "0"}
|
|
|
|
expectedIscsiadmOutput2 := map[string]string{
|
|
"iface.transport_name": "cxgb4i",
|
|
"iface.mtu": "0"}
|
|
|
|
expectedIscsiadmOutput3 := map[string]string{
|
|
"iface.mtu": "0"}
|
|
|
|
params, _ := parseIscsiadmShow(fakeIscsiadmOutput1)
|
|
if !reflect.DeepEqual(params, expectedIscsiadmOutput1) {
|
|
t.Errorf("parseIscsiadmShow: Fail to parse iface record: %s", params)
|
|
}
|
|
params, _ = parseIscsiadmShow(fakeIscsiadmOutput2)
|
|
if !reflect.DeepEqual(params, expectedIscsiadmOutput2) {
|
|
t.Errorf("parseIscsiadmShow: Fail to parse iface record: %s", params)
|
|
}
|
|
params, _ = parseIscsiadmShow(fakeIscsiadmOutput3)
|
|
if !reflect.DeepEqual(params, expectedIscsiadmOutput3) {
|
|
t.Errorf("parseIscsiadmShow: Fail to parse iface record: %s", params)
|
|
}
|
|
_, err := parseIscsiadmShow(fakeIscsiadmOutput4)
|
|
if err == nil {
|
|
t.Errorf("parseIscsiadmShow: Fail to handle invalid record: iface %s", fakeIscsiadmOutput4)
|
|
}
|
|
_, err = parseIscsiadmShow(fakeIscsiadmOutput5)
|
|
if err == nil {
|
|
t.Errorf("parseIscsiadmShow: Fail to handle invalid record: iface %s", fakeIscsiadmOutput5)
|
|
}
|
|
}
|
|
|
|
func TestClonedIface(t *testing.T) {
|
|
cmdCount := 0
|
|
fakeExec := mount.NewFakeExec(func(cmd string, args ...string) ([]byte, error) {
|
|
cmdCount++
|
|
if cmd != "iscsiadm" {
|
|
t.Errorf("iscsiadm command expected, got %q", cmd)
|
|
}
|
|
switch cmdCount {
|
|
case 1:
|
|
// iscsiadm -m iface -I <iface> -o show
|
|
return []byte("iface.ipaddress = <empty>\niface.transport_name = tcp\niface.initiatorname = <empty>\n"), nil
|
|
|
|
case 2:
|
|
// iscsiadm -m iface -I <newIface> -o new
|
|
return []byte("New interface 192.168.1.10:pv0001 added"), nil
|
|
case 3:
|
|
// iscsiadm -m iface -I <newIface> -o update -n <key> -v <val>
|
|
return []byte(""), nil
|
|
case 4:
|
|
return []byte(""), nil
|
|
}
|
|
return nil, fmt.Errorf("Unexpected exec call nr %d: %s", cmdCount, cmd)
|
|
})
|
|
plugins := []volume.VolumePlugin{
|
|
&iscsiPlugin{
|
|
host: nil,
|
|
},
|
|
}
|
|
plugin := plugins[0]
|
|
fakeMounter := iscsiDiskMounter{
|
|
iscsiDisk: &iscsiDisk{
|
|
plugin: plugin.(*iscsiPlugin)},
|
|
exec: fakeExec,
|
|
}
|
|
newIface := "192.168.1.10:pv0001"
|
|
cloneIface(fakeMounter, newIface)
|
|
if cmdCount != 4 {
|
|
t.Errorf("expected 4 CombinedOutput() calls, got %d", cmdCount)
|
|
}
|
|
|
|
}
|
|
|
|
func TestClonedIfaceShowError(t *testing.T) {
|
|
cmdCount := 0
|
|
fakeExec := mount.NewFakeExec(func(cmd string, args ...string) ([]byte, error) {
|
|
cmdCount++
|
|
if cmd != "iscsiadm" {
|
|
t.Errorf("iscsiadm command expected, got %q", cmd)
|
|
}
|
|
// iscsiadm -m iface -I <iface> -o show, return test error
|
|
return []byte(""), errors.New("test error")
|
|
})
|
|
plugins := []volume.VolumePlugin{
|
|
&iscsiPlugin{
|
|
host: nil,
|
|
},
|
|
}
|
|
plugin := plugins[0]
|
|
fakeMounter := iscsiDiskMounter{
|
|
iscsiDisk: &iscsiDisk{
|
|
plugin: plugin.(*iscsiPlugin)},
|
|
exec: fakeExec,
|
|
}
|
|
newIface := "192.168.1.10:pv0001"
|
|
cloneIface(fakeMounter, newIface)
|
|
if cmdCount != 1 {
|
|
t.Errorf("expected 1 CombinedOutput() calls, got %d", cmdCount)
|
|
}
|
|
|
|
}
|
|
|
|
func TestClonedIfaceUpdateError(t *testing.T) {
|
|
cmdCount := 0
|
|
fakeExec := mount.NewFakeExec(func(cmd string, args ...string) ([]byte, error) {
|
|
cmdCount++
|
|
if cmd != "iscsiadm" {
|
|
t.Errorf("iscsiadm command expected, got %q", cmd)
|
|
}
|
|
switch cmdCount {
|
|
case 1:
|
|
// iscsiadm -m iface -I <iface> -o show
|
|
return []byte("iface.ipaddress = <empty>\niface.transport_name = tcp\niface.initiatorname = <empty>\n"), nil
|
|
|
|
case 2:
|
|
// iscsiadm -m iface -I <newIface> -o new
|
|
return []byte("New interface 192.168.1.10:pv0001 added"), nil
|
|
case 3:
|
|
// iscsiadm -m iface -I <newIface> -o update -n <key> -v <val>
|
|
return []byte(""), nil
|
|
case 4:
|
|
return []byte(""), errors.New("test error")
|
|
case 5:
|
|
// iscsiadm -m iface -I <newIface> -o delete
|
|
return []byte(""), nil
|
|
}
|
|
return nil, fmt.Errorf("Unexpected exec call nr %d: %s", cmdCount, cmd)
|
|
})
|
|
plugins := []volume.VolumePlugin{
|
|
&iscsiPlugin{
|
|
host: nil,
|
|
},
|
|
}
|
|
plugin := plugins[0]
|
|
fakeMounter := iscsiDiskMounter{
|
|
iscsiDisk: &iscsiDisk{
|
|
plugin: plugin.(*iscsiPlugin)},
|
|
exec: fakeExec,
|
|
}
|
|
newIface := "192.168.1.10:pv0001"
|
|
cloneIface(fakeMounter, newIface)
|
|
if cmdCount != 5 {
|
|
t.Errorf("expected 5 CombinedOutput() calls, got %d", cmdCount)
|
|
}
|
|
|
|
}
|