135 lines
2.9 KiB
Go
135 lines
2.9 KiB
Go
/*
|
|
Copyright 2017 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 azure_dd
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
|
|
"k8s.io/kubernetes/pkg/util/exec"
|
|
)
|
|
|
|
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
|
|
}
|
|
|
|
var (
|
|
lun = 1
|
|
lunStr = "1"
|
|
diskPath = "4:0:0:" + lunStr
|
|
devName = "sdd"
|
|
lun1 = 2
|
|
lunStr1 = "2"
|
|
diskPath1 = "3:0:0:" + lunStr1
|
|
devName1 = "sde"
|
|
)
|
|
|
|
type fakeIOHandler struct{}
|
|
|
|
func (handler *fakeIOHandler) ReadDir(dirname string) ([]os.FileInfo, error) {
|
|
switch dirname {
|
|
case "/sys/bus/scsi/devices":
|
|
f1 := &fakeFileInfo{
|
|
name: "3:0:0:1",
|
|
}
|
|
f2 := &fakeFileInfo{
|
|
name: "4:0:0:0",
|
|
}
|
|
f3 := &fakeFileInfo{
|
|
name: diskPath,
|
|
}
|
|
f4 := &fakeFileInfo{
|
|
name: "host1",
|
|
}
|
|
f5 := &fakeFileInfo{
|
|
name: "target2:0:0",
|
|
}
|
|
return []os.FileInfo{f1, f2, f3, f4, f5}, nil
|
|
case "/sys/bus/scsi/devices/" + diskPath + "/block":
|
|
n := &fakeFileInfo{
|
|
name: devName,
|
|
}
|
|
return []os.FileInfo{n}, nil
|
|
case "/sys/bus/scsi/devices/" + diskPath1 + "/block":
|
|
n := &fakeFileInfo{
|
|
name: devName1,
|
|
}
|
|
return []os.FileInfo{n}, nil
|
|
}
|
|
return nil, fmt.Errorf("bad dir")
|
|
}
|
|
|
|
func (handler *fakeIOHandler) WriteFile(filename string, data []byte, perm os.FileMode) error {
|
|
return nil
|
|
}
|
|
|
|
func (handler *fakeIOHandler) Readlink(name string) (string, error) {
|
|
return "/dev/azure/disk/sda", nil
|
|
}
|
|
|
|
func TestIoHandler(t *testing.T) {
|
|
var fcmd exec.FakeCmd
|
|
fcmd = exec.FakeCmd{
|
|
CombinedOutputScript: []exec.FakeCombinedOutputAction{
|
|
// cat
|
|
func() ([]byte, error) {
|
|
return []byte("Msft \nVirtual Disk \n"), nil
|
|
},
|
|
// cat
|
|
func() ([]byte, error) {
|
|
return []byte("Msft \nVirtual Disk \n"), nil
|
|
},
|
|
},
|
|
}
|
|
fake := exec.FakeExec{
|
|
CommandScript: []exec.FakeCommandAction{
|
|
func(cmd string, args ...string) exec.Cmd { return exec.InitFakeCmd(&fcmd, cmd, args...) },
|
|
func(cmd string, args ...string) exec.Cmd { return exec.InitFakeCmd(&fcmd, cmd, args...) },
|
|
},
|
|
}
|
|
disk, err := findDiskByLun(lun, &fakeIOHandler{}, &fake)
|
|
// if no disk matches lun, exit
|
|
if disk != "/dev/"+devName || err != nil {
|
|
t.Errorf("no data disk found: disk %v err %v", disk, err)
|
|
}
|
|
}
|