mirror of https://github.com/k3s-io/k3s
117 lines
2.4 KiB
Go
117 lines
2.4 KiB
Go
|
/*
|
||
|
|
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 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 = "sda"
|
||
|
|
)
|
||
|
|
|
||
|
|
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
|
||
|
|
}
|
||
|
|
return nil, fmt.Errorf("bad dir")
|
||
|
|
}
|
||
|
|
|
||
|
|
func (handler *fakeIOHandler) WriteFile(filename string, data []byte, perm os.FileMode) error {
|
||
|
|
return 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
|
||
|
|
},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
fake := exec.FakeExec{
|
||
|
|
CommandScript: []exec.FakeCommandAction{
|
||
|
|
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 disk found: disk %v err %v", disk, err)
|
||
|
|
}
|
||
|
|
}
|