mirror of
https://github.com/genuinetools/reg.git
synced 2024-08-21 06:10:23 -04:00
118 lines
3.7 KiB
Go
118 lines
3.7 KiB
Go
// +build linux
|
|
|
|
package fs
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/Sirupsen/logrus"
|
|
"github.com/opencontainers/runc/libcontainer/cgroups"
|
|
)
|
|
|
|
func blkioStatEntryEquals(expected, actual []cgroups.BlkioStatEntry) error {
|
|
if len(expected) != len(actual) {
|
|
return fmt.Errorf("blkioStatEntries length do not match")
|
|
}
|
|
for i, expValue := range expected {
|
|
actValue := actual[i]
|
|
if expValue != actValue {
|
|
return fmt.Errorf("Expected blkio stat entry %v but found %v", expValue, actValue)
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func expectBlkioStatsEquals(t *testing.T, expected, actual cgroups.BlkioStats) {
|
|
if err := blkioStatEntryEquals(expected.IoServiceBytesRecursive, actual.IoServiceBytesRecursive); err != nil {
|
|
logrus.Printf("blkio IoServiceBytesRecursive do not match - %s\n", err)
|
|
t.Fail()
|
|
}
|
|
|
|
if err := blkioStatEntryEquals(expected.IoServicedRecursive, actual.IoServicedRecursive); err != nil {
|
|
logrus.Printf("blkio IoServicedRecursive do not match - %s\n", err)
|
|
t.Fail()
|
|
}
|
|
|
|
if err := blkioStatEntryEquals(expected.IoQueuedRecursive, actual.IoQueuedRecursive); err != nil {
|
|
logrus.Printf("blkio IoQueuedRecursive do not match - %s\n", err)
|
|
t.Fail()
|
|
}
|
|
|
|
if err := blkioStatEntryEquals(expected.SectorsRecursive, actual.SectorsRecursive); err != nil {
|
|
logrus.Printf("blkio SectorsRecursive do not match - %s\n", err)
|
|
t.Fail()
|
|
}
|
|
|
|
if err := blkioStatEntryEquals(expected.IoServiceTimeRecursive, actual.IoServiceTimeRecursive); err != nil {
|
|
logrus.Printf("blkio IoServiceTimeRecursive do not match - %s\n", err)
|
|
t.Fail()
|
|
}
|
|
|
|
if err := blkioStatEntryEquals(expected.IoWaitTimeRecursive, actual.IoWaitTimeRecursive); err != nil {
|
|
logrus.Printf("blkio IoWaitTimeRecursive do not match - %s\n", err)
|
|
t.Fail()
|
|
}
|
|
|
|
if err := blkioStatEntryEquals(expected.IoMergedRecursive, actual.IoMergedRecursive); err != nil {
|
|
logrus.Printf("blkio IoMergedRecursive do not match - %v vs %v\n", expected.IoMergedRecursive, actual.IoMergedRecursive)
|
|
t.Fail()
|
|
}
|
|
|
|
if err := blkioStatEntryEquals(expected.IoTimeRecursive, actual.IoTimeRecursive); err != nil {
|
|
logrus.Printf("blkio IoTimeRecursive do not match - %s\n", err)
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func expectThrottlingDataEquals(t *testing.T, expected, actual cgroups.ThrottlingData) {
|
|
if expected != actual {
|
|
logrus.Printf("Expected throttling data %v but found %v\n", expected, actual)
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func expectHugetlbStatEquals(t *testing.T, expected, actual cgroups.HugetlbStats) {
|
|
if expected != actual {
|
|
logrus.Printf("Expected hugetlb stats %v but found %v\n", expected, actual)
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func expectMemoryStatEquals(t *testing.T, expected, actual cgroups.MemoryStats) {
|
|
expectMemoryDataEquals(t, expected.Usage, actual.Usage)
|
|
expectMemoryDataEquals(t, expected.SwapUsage, actual.SwapUsage)
|
|
expectMemoryDataEquals(t, expected.KernelUsage, actual.KernelUsage)
|
|
|
|
for key, expValue := range expected.Stats {
|
|
actValue, ok := actual.Stats[key]
|
|
if !ok {
|
|
logrus.Printf("Expected memory stat key %s not found\n", key)
|
|
t.Fail()
|
|
}
|
|
if expValue != actValue {
|
|
logrus.Printf("Expected memory stat value %d but found %d\n", expValue, actValue)
|
|
t.Fail()
|
|
}
|
|
}
|
|
}
|
|
|
|
func expectMemoryDataEquals(t *testing.T, expected, actual cgroups.MemoryData) {
|
|
if expected.Usage != actual.Usage {
|
|
logrus.Printf("Expected memory usage %d but found %d\n", expected.Usage, actual.Usage)
|
|
t.Fail()
|
|
}
|
|
if expected.MaxUsage != actual.MaxUsage {
|
|
logrus.Printf("Expected memory max usage %d but found %d\n", expected.MaxUsage, actual.MaxUsage)
|
|
t.Fail()
|
|
}
|
|
if expected.Failcnt != actual.Failcnt {
|
|
logrus.Printf("Expected memory failcnt %d but found %d\n", expected.Failcnt, actual.Failcnt)
|
|
t.Fail()
|
|
}
|
|
if expected.Limit != actual.Limit {
|
|
logrus.Printf("Expected memory limit %d but found %d\n", expected.Limit, actual.Limit)
|
|
t.Fail()
|
|
}
|
|
}
|