104 lines
2.3 KiB
Go
104 lines
2.3 KiB
Go
package disk
|
|
|
|
import (
|
|
"fmt"
|
|
"runtime"
|
|
"testing"
|
|
|
|
"github.com/influxdb/telegraf/plugins/system/ps/common"
|
|
)
|
|
|
|
func TestDisk_usage(t *testing.T) {
|
|
path := "/"
|
|
if runtime.GOOS == "windows" {
|
|
path = "C:"
|
|
}
|
|
v, err := DiskUsage(path)
|
|
if err != nil {
|
|
t.Errorf("error %v", err)
|
|
}
|
|
if v.Path != path {
|
|
t.Errorf("error %v", err)
|
|
}
|
|
}
|
|
|
|
func TestDisk_partitions(t *testing.T) {
|
|
ret, err := DiskPartitions(false)
|
|
if err != nil || len(ret) == 0 {
|
|
t.Errorf("error %v", err)
|
|
}
|
|
empty := DiskPartitionStat{}
|
|
for _, disk := range ret {
|
|
if disk == empty {
|
|
t.Errorf("Could not get device info %v", disk)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestDisk_io_counters(t *testing.T) {
|
|
ret, err := DiskIOCounters()
|
|
if err != nil {
|
|
if err == common.NotImplementedError {
|
|
t.SkipNow()
|
|
}
|
|
|
|
t.Errorf("error %v", err)
|
|
}
|
|
if len(ret) == 0 {
|
|
t.Errorf("ret is empty: %s", ret)
|
|
}
|
|
empty := DiskIOCountersStat{}
|
|
for part, io := range ret {
|
|
fmt.Println(io)
|
|
if io == empty {
|
|
t.Errorf("io_counter error %v, %v", part, io)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestDiskUsageStat_String(t *testing.T) {
|
|
v := DiskUsageStat{
|
|
Path: "/",
|
|
Total: 1000,
|
|
Free: 2000,
|
|
Used: 3000,
|
|
UsedPercent: 50.1,
|
|
InodesTotal: 4000,
|
|
InodesUsed: 5000,
|
|
InodesFree: 6000,
|
|
InodesUsedPercent: 49.1,
|
|
}
|
|
e := `{"path":"/","total":1000,"free":2000,"used":3000,"used_percent":50.1,"inodes_total":4000,"inodes_used":5000,"inodes_free":6000,"inodes_used_percent":49.1}`
|
|
if e != fmt.Sprintf("%v", v) {
|
|
t.Errorf("DiskUsageStat string is invalid: %v", v)
|
|
}
|
|
}
|
|
|
|
func TestDiskPartitionStat_String(t *testing.T) {
|
|
v := DiskPartitionStat{
|
|
Device: "sd01",
|
|
Mountpoint: "/",
|
|
Fstype: "ext4",
|
|
Opts: "ro",
|
|
}
|
|
e := `{"device":"sd01","mountpoint":"/","fstype":"ext4","opts":"ro"}`
|
|
if e != fmt.Sprintf("%v", v) {
|
|
t.Errorf("DiskUsageStat string is invalid: %v", v)
|
|
}
|
|
}
|
|
|
|
func TestDiskIOCountersStat_String(t *testing.T) {
|
|
v := DiskIOCountersStat{
|
|
Name: "sd01",
|
|
ReadCount: 100,
|
|
WriteCount: 200,
|
|
ReadBytes: 300,
|
|
WriteBytes: 400,
|
|
SerialNumber: "SERIAL",
|
|
}
|
|
e := `{"read_count":100,"write_count":200,"read_bytes":300,"write_bytes":400,"read_time":0,"write_time":0,"name":"sd01","io_time":0,"serial_number":"SERIAL"}`
|
|
if e != fmt.Sprintf("%v", v) {
|
|
t.Errorf("DiskUsageStat string is invalid: %v", v)
|
|
}
|
|
}
|