package system import ( "os" "testing" "github.com/influxdata/telegraf/testutil" "github.com/shirou/gopsutil/disk" "github.com/stretchr/testify/assert" "github.com/stretchr/testify/mock" "github.com/stretchr/testify/require" ) type MockFileInfo struct { os.FileInfo } func TestDiskUsage(t *testing.T) { mck := &mock.Mock{} mps := MockPSDisk{&systemPS{&mockDiskUsage{mck}}, mck} defer mps.AssertExpectations(t) var acc testutil.Accumulator var err error psAll := []disk.PartitionStat{ { Device: "/dev/sda", Mountpoint: "/", Fstype: "ext4", Opts: "ro,noatime,nodiratime", }, { Device: "/dev/sdb", Mountpoint: "/home", Fstype: "ext4", Opts: "rw,noatime,nodiratime,errors=remount-ro", }, } duAll := []disk.UsageStat{ { Path: "/", Fstype: "ext4", Total: 128, Free: 23, Used: 100, InodesTotal: 1234, InodesFree: 234, InodesUsed: 1000, }, { Path: "/home", Fstype: "ext4", Total: 256, Free: 46, Used: 200, InodesTotal: 2468, InodesFree: 468, InodesUsed: 2000, }, } mps.On("Partitions", true).Return(psAll, nil) mps.On("OSGetenv", "HOST_MOUNT_PREFIX").Return("") mps.On("PSDiskUsage", "/").Return(&duAll[0], nil) mps.On("PSDiskUsage", "/home").Return(&duAll[1], nil) err = (&DiskStats{ps: mps}).Gather(&acc) require.NoError(t, err) numDiskMetrics := acc.NFields() expectedAllDiskMetrics := 14 assert.Equal(t, expectedAllDiskMetrics, numDiskMetrics) tags1 := map[string]string{ "path": "/", "fstype": "ext4", "device": "sda", "mode": "ro", } tags2 := map[string]string{ "path": "/home", "fstype": "ext4", "device": "sdb", "mode": "rw", } fields1 := map[string]interface{}{ "total": uint64(128), "used": uint64(100), "free": uint64(23), "inodes_total": uint64(1234), "inodes_free": uint64(234), "inodes_used": uint64(1000), "used_percent": float64(81.30081300813008), } fields2 := map[string]interface{}{ "total": uint64(256), "used": uint64(200), "free": uint64(46), "inodes_total": uint64(2468), "inodes_free": uint64(468), "inodes_used": uint64(2000), "used_percent": float64(81.30081300813008), } acc.AssertContainsTaggedFields(t, "disk", fields1, tags1) acc.AssertContainsTaggedFields(t, "disk", fields2, tags2) // We expect 6 more DiskMetrics to show up with an explicit match on "/" // and /home not matching the /dev in MountPoints err = (&DiskStats{ps: &mps, MountPoints: []string{"/", "/dev"}}).Gather(&acc) assert.Equal(t, expectedAllDiskMetrics+7, acc.NFields()) // We should see all the diskpoints as MountPoints includes both // / and /home err = (&DiskStats{ps: &mps, MountPoints: []string{"/", "/home"}}).Gather(&acc) assert.Equal(t, 2*expectedAllDiskMetrics+7, acc.NFields()) } func TestDiskUsageHostMountPrefix(t *testing.T) { tests := []struct { name string partitionStats []disk.PartitionStat usageStats []*disk.UsageStat hostMountPrefix string expectedTags map[string]string expectedFields map[string]interface{} }{ { name: "no host mount prefix", partitionStats: []disk.PartitionStat{ { Device: "/dev/sda", Mountpoint: "/", Fstype: "ext4", Opts: "ro", }, }, usageStats: []*disk.UsageStat{ &disk.UsageStat{ Path: "/", Total: 42, }, }, expectedTags: map[string]string{ "path": "/", "device": "sda", "fstype": "ext4", "mode": "ro", }, expectedFields: map[string]interface{}{ "total": uint64(42), "used": uint64(0), "free": uint64(0), "inodes_total": uint64(0), "inodes_free": uint64(0), "inodes_used": uint64(0), "used_percent": float64(0), }, }, { name: "host mount prefix", partitionStats: []disk.PartitionStat{ { Device: "/dev/sda", Mountpoint: "/hostfs/var", Fstype: "ext4", Opts: "ro", }, }, usageStats: []*disk.UsageStat{ &disk.UsageStat{ Path: "/hostfs/var", Total: 42, }, }, hostMountPrefix: "/hostfs", expectedTags: map[string]string{ "path": "/var", "device": "sda", "fstype": "ext4", "mode": "ro", }, expectedFields: map[string]interface{}{ "total": uint64(42), "used": uint64(0), "free": uint64(0), "inodes_total": uint64(0), "inodes_free": uint64(0), "inodes_used": uint64(0), "used_percent": float64(0), }, }, { name: "host mount prefix exact match", partitionStats: []disk.PartitionStat{ { Device: "/dev/sda", Mountpoint: "/hostfs", Fstype: "ext4", Opts: "ro", }, }, usageStats: []*disk.UsageStat{ &disk.UsageStat{ Path: "/hostfs", Total: 42, }, }, hostMountPrefix: "/hostfs", expectedTags: map[string]string{ "path": "/", "device": "sda", "fstype": "ext4", "mode": "ro", }, expectedFields: map[string]interface{}{ "total": uint64(42), "used": uint64(0), "free": uint64(0), "inodes_total": uint64(0), "inodes_free": uint64(0), "inodes_used": uint64(0), "used_percent": float64(0), }, }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { mck := &mock.Mock{} mps := MockPSDisk{&systemPS{&mockDiskUsage{mck}}, mck} defer mps.AssertExpectations(t) var acc testutil.Accumulator var err error mps.On("Partitions", true).Return(tt.partitionStats, nil) for _, v := range tt.usageStats { mps.On("PSDiskUsage", v.Path).Return(v, nil) } mps.On("OSGetenv", "HOST_MOUNT_PREFIX").Return(tt.hostMountPrefix) err = (&DiskStats{ps: mps}).Gather(&acc) require.NoError(t, err) acc.AssertContainsTaggedFields(t, "disk", tt.expectedFields, tt.expectedTags) }) } } func TestDiskStats(t *testing.T) { var mps MockPS defer mps.AssertExpectations(t) var acc testutil.Accumulator var err error duAll := []*disk.UsageStat{ { Path: "/", Fstype: "ext4", Total: 128, Free: 23, Used: 100, InodesTotal: 1234, InodesFree: 234, InodesUsed: 1000, }, { Path: "/home", Fstype: "ext4", Total: 256, Free: 46, Used: 200, InodesTotal: 2468, InodesFree: 468, InodesUsed: 2000, }, } duFiltered := []*disk.UsageStat{ { Path: "/", Fstype: "ext4", Total: 128, Free: 23, Used: 100, InodesTotal: 1234, InodesFree: 234, InodesUsed: 1000, }, } psAll := []*disk.PartitionStat{ { Device: "/dev/sda", Mountpoint: "/", Fstype: "ext4", Opts: "ro,noatime,nodiratime", }, { Device: "/dev/sdb", Mountpoint: "/home", Fstype: "ext4", Opts: "rw,noatime,nodiratime,errors=remount-ro", }, } psFiltered := []*disk.PartitionStat{ { Device: "/dev/sda", Mountpoint: "/", Fstype: "ext4", Opts: "ro,noatime,nodiratime", }, } mps.On("DiskUsage", []string(nil), []string(nil)).Return(duAll, psAll, nil) mps.On("DiskUsage", []string{"/", "/dev"}, []string(nil)).Return(duFiltered, psFiltered, nil) mps.On("DiskUsage", []string{"/", "/home"}, []string(nil)).Return(duAll, psAll, nil) err = (&DiskStats{ps: &mps}).Gather(&acc) require.NoError(t, err) numDiskMetrics := acc.NFields() expectedAllDiskMetrics := 14 assert.Equal(t, expectedAllDiskMetrics, numDiskMetrics) tags1 := map[string]string{ "path": "/", "fstype": "ext4", "device": "sda", "mode": "ro", } tags2 := map[string]string{ "path": "/home", "fstype": "ext4", "device": "sdb", "mode": "rw", } fields1 := map[string]interface{}{ "total": uint64(128), "used": uint64(100), "free": uint64(23), "inodes_total": uint64(1234), "inodes_free": uint64(234), "inodes_used": uint64(1000), "used_percent": float64(81.30081300813008), } fields2 := map[string]interface{}{ "total": uint64(256), "used": uint64(200), "free": uint64(46), "inodes_total": uint64(2468), "inodes_free": uint64(468), "inodes_used": uint64(2000), "used_percent": float64(81.30081300813008), } acc.AssertContainsTaggedFields(t, "disk", fields1, tags1) acc.AssertContainsTaggedFields(t, "disk", fields2, tags2) // We expect 6 more DiskMetrics to show up with an explicit match on "/" // and /home not matching the /dev in MountPoints err = (&DiskStats{ps: &mps, MountPoints: []string{"/", "/dev"}}).Gather(&acc) assert.Equal(t, expectedAllDiskMetrics+7, acc.NFields()) // We should see all the diskpoints as MountPoints includes both // / and /home err = (&DiskStats{ps: &mps, MountPoints: []string{"/", "/home"}}).Gather(&acc) assert.Equal(t, 2*expectedAllDiskMetrics+7, acc.NFields()) }