119 lines
3.2 KiB
Go
119 lines
3.2 KiB
Go
package collector
|
|
|
|
import (
|
|
"os"
|
|
"os/exec"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestMergeStorageDevicePrefersNonEmptyFields(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
got := mergeStorageDevice(
|
|
lsblkDevice{Name: "nvme0n1", Type: "disk", Tran: "nvme"},
|
|
lsblkDevice{Name: "nvme0n1", Type: "disk", Size: "1024", Serial: "SN123", Model: "Kioxia"},
|
|
)
|
|
|
|
if got.Serial != "SN123" {
|
|
t.Fatalf("serial=%q want SN123", got.Serial)
|
|
}
|
|
if got.Model != "Kioxia" {
|
|
t.Fatalf("model=%q want Kioxia", got.Model)
|
|
}
|
|
if got.Size != "1024" {
|
|
t.Fatalf("size=%q want 1024", got.Size)
|
|
}
|
|
}
|
|
|
|
func TestParseStorageBytes(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
if got := parseStorageBytes(" 2048 "); got != 2048 {
|
|
t.Fatalf("parseStorageBytes=%d want 2048", got)
|
|
}
|
|
if got := parseStorageBytes("1.92 TB"); got != 0 {
|
|
t.Fatalf("parseStorageBytes invalid=%d want 0", got)
|
|
}
|
|
}
|
|
|
|
func TestBestEffortRescanHotplugStorage(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
tmp := t.TempDir()
|
|
rescanPath := filepath.Join(tmp, "pci-rescan")
|
|
scanDir := filepath.Join(tmp, "scsi_host")
|
|
host0Path := filepath.Join(scanDir, "host0", "scan")
|
|
host1Path := filepath.Join(scanDir, "host1", "scan")
|
|
argsPath := filepath.Join(tmp, "udevadm-args")
|
|
toolPath := filepath.Join(tmp, "udevadm")
|
|
if err := os.MkdirAll(filepath.Dir(host0Path), 0755); err != nil {
|
|
t.Fatalf("mkdir host0: %v", err)
|
|
}
|
|
if err := os.MkdirAll(filepath.Dir(host1Path), 0755); err != nil {
|
|
t.Fatalf("mkdir host1: %v", err)
|
|
}
|
|
if err := os.WriteFile(host0Path, nil, 0644); err != nil {
|
|
t.Fatalf("touch host0 scan: %v", err)
|
|
}
|
|
if err := os.WriteFile(host1Path, nil, 0644); err != nil {
|
|
t.Fatalf("touch host1 scan: %v", err)
|
|
}
|
|
script := "#!/bin/sh\nprintf '%s' \"$*\" > \"" + argsPath + "\"\n"
|
|
if err := os.WriteFile(toolPath, []byte(script), 0755); err != nil {
|
|
t.Fatalf("write udevadm stub: %v", err)
|
|
}
|
|
|
|
oldPath := os.Getenv("PATH")
|
|
if err := os.Setenv("PATH", tmp+string(os.PathListSeparator)+oldPath); err != nil {
|
|
t.Fatalf("set PATH: %v", err)
|
|
}
|
|
defer func() { _ = os.Setenv("PATH", oldPath) }()
|
|
|
|
oldRescanPath := pciRescanPath
|
|
oldSCSIGlob := scsiHostScanGlob
|
|
oldWriteFile := hotplugWriteFile
|
|
oldExecCommand := hotplugExecCommand
|
|
oldGlob := hotplugGlob
|
|
pciRescanPath = rescanPath
|
|
scsiHostScanGlob = filepath.Join(scanDir, "host*", "scan")
|
|
hotplugWriteFile = os.WriteFile
|
|
hotplugExecCommand = exec.Command
|
|
hotplugGlob = filepath.Glob
|
|
defer func() {
|
|
pciRescanPath = oldRescanPath
|
|
scsiHostScanGlob = oldSCSIGlob
|
|
hotplugWriteFile = oldWriteFile
|
|
hotplugExecCommand = oldExecCommand
|
|
hotplugGlob = oldGlob
|
|
}()
|
|
|
|
bestEffortRescanHotplugStorage()
|
|
|
|
raw, err := os.ReadFile(rescanPath)
|
|
if err != nil {
|
|
t.Fatalf("read rescan file: %v", err)
|
|
}
|
|
if string(raw) != "1\n" {
|
|
t.Fatalf("rescan payload=%q want %q", string(raw), "1\n")
|
|
}
|
|
for _, path := range []string{host0Path, host1Path} {
|
|
raw, err := os.ReadFile(path)
|
|
if err != nil {
|
|
t.Fatalf("read scsi scan file %s: %v", path, err)
|
|
}
|
|
if string(raw) != "- - -\n" {
|
|
t.Fatalf("scsi scan payload at %s =%q want %q", path, string(raw), "- - -\n")
|
|
}
|
|
}
|
|
|
|
args, err := os.ReadFile(argsPath)
|
|
if err != nil {
|
|
t.Fatalf("read udevadm args: %v", err)
|
|
}
|
|
if got := strings.TrimSpace(string(args)); got != "settle --timeout=10" {
|
|
t.Fatalf("udevadm args=%q want %q", got, "settle --timeout=10")
|
|
}
|
|
}
|