mirror of https://github.com/k3s-io/k3s
291 lines
8.0 KiB
Go
291 lines
8.0 KiB
Go
// +build linux
|
|
|
|
/*
|
|
Copyright 2016 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 util
|
|
|
|
import (
|
|
"errors"
|
|
"os"
|
|
"reflect"
|
|
"regexp"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
type mockOsIOHandler struct{}
|
|
|
|
func (handler *mockOsIOHandler) ReadFile(filename string) ([]byte, error) {
|
|
portPattern := regexp.MustCompile("^/sys/class/iscsi_host/(host\\d)/device/session\\d/connection\\d:0/iscsi_connection/connection\\d:0/(?:persistent_)?port$")
|
|
if portPattern.MatchString(filename) {
|
|
return []byte("3260"), nil
|
|
}
|
|
addressPattern := regexp.MustCompile("^/sys/class/iscsi_host/(host\\d)/device/session\\d/connection\\d:0/iscsi_connection/connection\\d:0/(?:persistent_)?address$")
|
|
matches := addressPattern.FindStringSubmatch(filename)
|
|
if nil != matches {
|
|
switch matches[1] {
|
|
case "host2":
|
|
return []byte("10.0.0.1"), nil
|
|
case "host3":
|
|
return []byte("10.0.0.2"), nil
|
|
}
|
|
}
|
|
targetNamePattern := regexp.MustCompile("^/sys/class/iscsi_host/(host\\d)/device/session\\d/iscsi_session/session\\d/targetname$")
|
|
matches = targetNamePattern.FindStringSubmatch(filename)
|
|
if nil != matches {
|
|
switch matches[1] {
|
|
case "host2":
|
|
return []byte("target1"), nil
|
|
case "host3":
|
|
return []byte("target2"), nil
|
|
}
|
|
}
|
|
return nil, errors.New("Not Implemented for Mock")
|
|
}
|
|
|
|
func (handler *mockOsIOHandler) ReadDir(dirname string) ([]os.FileInfo, error) {
|
|
switch dirname {
|
|
case "/sys/block/dm-1/slaves":
|
|
f1 := &fakeFileInfo{
|
|
name: "sda",
|
|
}
|
|
f2 := &fakeFileInfo{
|
|
name: "sdb",
|
|
}
|
|
return []os.FileInfo{f1, f2}, nil
|
|
case "/sys/block/":
|
|
f1 := &fakeFileInfo{
|
|
name: "sda",
|
|
}
|
|
f2 := &fakeFileInfo{
|
|
name: "dm-1",
|
|
}
|
|
return []os.FileInfo{f1, f2}, nil
|
|
case "/sys/class/iscsi_host":
|
|
f1 := &fakeFileInfo{
|
|
name: "host2",
|
|
}
|
|
f2 := &fakeFileInfo{
|
|
name: "host3",
|
|
}
|
|
f3 := &fakeFileInfo{
|
|
name: "ignore",
|
|
}
|
|
return []os.FileInfo{f1, f2, f3}, nil
|
|
case "/sys/class/iscsi_host/host2/device":
|
|
f1 := &fakeFileInfo{
|
|
name: "session1",
|
|
}
|
|
f2 := &fakeFileInfo{
|
|
name: "ignore",
|
|
}
|
|
return []os.FileInfo{f1, f2}, nil
|
|
case "/sys/class/iscsi_host/host3/device":
|
|
f1 := &fakeFileInfo{
|
|
name: "session2",
|
|
}
|
|
f2 := &fakeFileInfo{
|
|
name: "ignore",
|
|
}
|
|
return []os.FileInfo{f1, f2}, nil
|
|
case "/sys/class/iscsi_host/host2/device/session1":
|
|
f1 := &fakeFileInfo{
|
|
name: "connection1:0",
|
|
}
|
|
f2 := &fakeFileInfo{
|
|
name: "ignore",
|
|
}
|
|
return []os.FileInfo{f1, f2}, nil
|
|
case "/sys/class/iscsi_host/host3/device/session2":
|
|
f1 := &fakeFileInfo{
|
|
name: "connection2:0",
|
|
}
|
|
f2 := &fakeFileInfo{
|
|
name: "ignore",
|
|
}
|
|
return []os.FileInfo{f1, f2}, nil
|
|
case "/sys/class/iscsi_host/host2/device/session1/target2:0:0/2:0:0:1/block":
|
|
f1 := &fakeFileInfo{
|
|
name: "sda",
|
|
}
|
|
return []os.FileInfo{f1}, nil
|
|
case "/sys/class/iscsi_host/host2/device/session1/target2:0:0/2:0:0:2/block":
|
|
f1 := &fakeFileInfo{
|
|
name: "sdc",
|
|
}
|
|
return []os.FileInfo{f1}, nil
|
|
case "/sys/class/iscsi_host/host3/device/session2/target3:0:0/3:0:0:1/block":
|
|
f1 := &fakeFileInfo{
|
|
name: "sdb",
|
|
}
|
|
return []os.FileInfo{f1}, nil
|
|
case "/sys/class/iscsi_host/host3/device/session2/target3:0:0/3:0:0:2/block":
|
|
f1 := &fakeFileInfo{
|
|
name: "sdd",
|
|
}
|
|
return []os.FileInfo{f1}, nil
|
|
}
|
|
return nil, errors.New("Not Implemented for Mock")
|
|
}
|
|
|
|
func (handler *mockOsIOHandler) Lstat(name string) (os.FileInfo, error) {
|
|
links := map[string]string{
|
|
"/sys/block/dm-1/slaves/sda": "sda",
|
|
"/dev/sda": "sda",
|
|
"/sys/class/iscsi_host/host2/device/session1/target2:0:0/2:0:0:1": "2:0:0:1",
|
|
"/sys/class/iscsi_host/host2/device/session1/target2:0:0/2:0:0:2": "2:0:0:2",
|
|
"/sys/class/iscsi_host/host3/device/session2/target3:0:0/3:0:0:1": "3:0:0:1",
|
|
"/sys/class/iscsi_host/host3/device/session2/target3:0:0/3:0:0:2": "3:0:0:2",
|
|
}
|
|
if dev, ok := links[name]; ok {
|
|
return &fakeFileInfo{name: dev}, nil
|
|
}
|
|
return nil, errors.New("Not Implemented for Mock")
|
|
}
|
|
|
|
func (handler *mockOsIOHandler) EvalSymlinks(path string) (string, error) {
|
|
links := map[string]string{
|
|
"/returns/a/dev": "/dev/sde",
|
|
"/returns/non/dev": "/sys/block",
|
|
"/dev/disk/by-path/127.0.0.1:3260-eui.02004567A425678D-lun-0": "/dev/sda",
|
|
"/dev/disk/by-path/127.0.0.3:3260-eui.03004567A425678D-lun-0": "/dev/sdb",
|
|
"/dev/dm-2": "/dev/dm-2",
|
|
"/dev/dm-3": "/dev/dm-3",
|
|
"/dev/sdc": "/dev/sdc",
|
|
"/dev/sde": "/dev/sde",
|
|
}
|
|
return links[path], nil
|
|
}
|
|
|
|
func (handler *mockOsIOHandler) WriteFile(filename string, data []byte, perm os.FileMode) error {
|
|
return errors.New("Not Implemented for Mock")
|
|
}
|
|
|
|
type fakeFileInfo struct {
|
|
name string
|
|
}
|
|
|
|
func (fi *fakeFileInfo) Name() string {
|
|
return fi.name
|
|
}
|
|
|
|
func (fi *fakeFileInfo) Size() int64 {
|
|
return 0
|
|
}
|
|
|
|
func (fi *fakeFileInfo) Mode() os.FileMode {
|
|
return 777
|
|
}
|
|
|
|
func (fi *fakeFileInfo) ModTime() time.Time {
|
|
return time.Now()
|
|
}
|
|
func (fi *fakeFileInfo) IsDir() bool {
|
|
return false
|
|
}
|
|
|
|
func (fi *fakeFileInfo) Sys() interface{} {
|
|
return nil
|
|
}
|
|
|
|
func TestFindMultipathDeviceForDevice(t *testing.T) {
|
|
mockDeviceUtil := NewDeviceHandler(&mockOsIOHandler{})
|
|
dev := mockDeviceUtil.FindMultipathDeviceForDevice("/dev/disk/by-path/127.0.0.1:3260-eui.02004567A425678D-lun-0")
|
|
if dev != "/dev/dm-1" {
|
|
t.Fatalf("mpio device not found dm-1 expected got [%s]", dev)
|
|
}
|
|
dev = mockDeviceUtil.FindMultipathDeviceForDevice("/dev/disk/by-path/empty")
|
|
if dev != "" {
|
|
t.Fatalf("mpio device not found '' expected got [%s]", dev)
|
|
}
|
|
}
|
|
|
|
func TestFindDeviceForPath(t *testing.T) {
|
|
io := &mockOsIOHandler{}
|
|
|
|
disk, err := findDeviceForPath("/dev/sde", io)
|
|
if err != nil {
|
|
t.Fatalf("error finding device for path /dev/sde:%v", err)
|
|
}
|
|
if disk != "sde" {
|
|
t.Fatalf("disk [%s] didn't match expected sde", disk)
|
|
}
|
|
disk, err = findDeviceForPath("/returns/a/dev", io)
|
|
if err != nil {
|
|
t.Fatalf("error finding device for path /returns/a/dev:%v", err)
|
|
}
|
|
if disk != "sde" {
|
|
t.Fatalf("disk [%s] didn't match expected sde", disk)
|
|
}
|
|
_, err = findDeviceForPath("/returns/non/dev", io)
|
|
if err == nil {
|
|
t.Fatalf("link is to incorrect dev")
|
|
}
|
|
|
|
_, err = findDeviceForPath("/path/doesnt/exist", &osIOHandler{})
|
|
if err == nil {
|
|
t.Fatalf("path shouldn't exist but still doesn't give an error")
|
|
}
|
|
|
|
}
|
|
|
|
func TestFindSlaveDevicesOnMultipath(t *testing.T) {
|
|
mockDeviceUtil := NewDeviceHandler(&mockOsIOHandler{})
|
|
devices := mockDeviceUtil.FindSlaveDevicesOnMultipath("/dev/dm-1")
|
|
if !reflect.DeepEqual(devices, []string{"/dev/sda", "/dev/sdb"}) {
|
|
t.Fatalf("failed to find devices managed by mpio device. /dev/sda, /dev/sdb expected got [%s]", devices)
|
|
}
|
|
dev := mockDeviceUtil.FindSlaveDevicesOnMultipath("/dev/sdc")
|
|
if len(dev) != 0 {
|
|
t.Fatalf("mpio device not found '' expected got [%s]", dev)
|
|
}
|
|
}
|
|
|
|
func TestGetISCSIPortalHostMapForTarget(t *testing.T) {
|
|
mockDeviceUtil := NewDeviceHandler(&mockOsIOHandler{})
|
|
portalHostMap, err := mockDeviceUtil.GetISCSIPortalHostMapForTarget("target1")
|
|
if nil != err {
|
|
t.Fatalf("error getting scsi hosts for target: %v", err)
|
|
}
|
|
if nil == portalHostMap {
|
|
t.Fatal("no portal host map returned")
|
|
}
|
|
if 1 != len(portalHostMap) {
|
|
t.Fatalf("wrong number of map entries in portal host map: %d", len(portalHostMap))
|
|
}
|
|
if 2 != portalHostMap["10.0.0.1:3260"] {
|
|
t.Fatalf("incorrect entry in portal host map: %v", portalHostMap)
|
|
}
|
|
}
|
|
|
|
func TestFindDevicesForISCSILun(t *testing.T) {
|
|
mockDeviceUtil := NewDeviceHandler(&mockOsIOHandler{})
|
|
devices, err := mockDeviceUtil.FindDevicesForISCSILun("target1", 1)
|
|
if nil != err {
|
|
t.Fatalf("error getting devices for lun: %v", err)
|
|
}
|
|
if nil == devices {
|
|
t.Fatal("no devices returned")
|
|
}
|
|
if 1 != len(devices) {
|
|
t.Fatalf("wrong number of devices: %d", len(devices))
|
|
}
|
|
if "sda" != devices[0] {
|
|
t.Fatalf("incorrect device %v", devices)
|
|
}
|
|
}
|