77 "testing"
88
99 "github.com/opencontainers/runc/libcontainer/cgroups"
10-
11- "github.com/sirupsen/logrus"
1210)
1311
1412func blkioStatEntryEquals (expected , actual []cgroups.BlkioStatEntry ) error {
@@ -26,57 +24,47 @@ func blkioStatEntryEquals(expected, actual []cgroups.BlkioStatEntry) error {
2624
2725func expectBlkioStatsEquals (t * testing.T , expected , actual cgroups.BlkioStats ) {
2826 if err := blkioStatEntryEquals (expected .IoServiceBytesRecursive , actual .IoServiceBytesRecursive ); err != nil {
29- logrus .Printf ("blkio IoServiceBytesRecursive do not match - %s\n " , err )
30- t .Fail ()
27+ t .Errorf ("blkio IoServiceBytesRecursive do not match - %s\n " , err )
3128 }
3229
3330 if err := blkioStatEntryEquals (expected .IoServicedRecursive , actual .IoServicedRecursive ); err != nil {
34- logrus .Printf ("blkio IoServicedRecursive do not match - %s\n " , err )
35- t .Fail ()
31+ t .Errorf ("blkio IoServicedRecursive do not match - %s\n " , err )
3632 }
3733
3834 if err := blkioStatEntryEquals (expected .IoQueuedRecursive , actual .IoQueuedRecursive ); err != nil {
39- logrus .Printf ("blkio IoQueuedRecursive do not match - %s\n " , err )
40- t .Fail ()
35+ t .Errorf ("blkio IoQueuedRecursive do not match - %s\n " , err )
4136 }
4237
4338 if err := blkioStatEntryEquals (expected .SectorsRecursive , actual .SectorsRecursive ); err != nil {
44- logrus .Printf ("blkio SectorsRecursive do not match - %s\n " , err )
45- t .Fail ()
39+ t .Errorf ("blkio SectorsRecursive do not match - %s\n " , err )
4640 }
4741
4842 if err := blkioStatEntryEquals (expected .IoServiceTimeRecursive , actual .IoServiceTimeRecursive ); err != nil {
49- logrus .Printf ("blkio IoServiceTimeRecursive do not match - %s\n " , err )
50- t .Fail ()
43+ t .Errorf ("blkio IoServiceTimeRecursive do not match - %s\n " , err )
5144 }
5245
5346 if err := blkioStatEntryEquals (expected .IoWaitTimeRecursive , actual .IoWaitTimeRecursive ); err != nil {
54- logrus .Printf ("blkio IoWaitTimeRecursive do not match - %s\n " , err )
55- t .Fail ()
47+ t .Errorf ("blkio IoWaitTimeRecursive do not match - %s\n " , err )
5648 }
5749
5850 if err := blkioStatEntryEquals (expected .IoMergedRecursive , actual .IoMergedRecursive ); err != nil {
59- logrus .Printf ("blkio IoMergedRecursive do not match - %v vs %v\n " , expected .IoMergedRecursive , actual .IoMergedRecursive )
60- t .Fail ()
51+ t .Errorf ("blkio IoMergedRecursive do not match - %v vs %v\n " , expected .IoMergedRecursive , actual .IoMergedRecursive )
6152 }
6253
6354 if err := blkioStatEntryEquals (expected .IoTimeRecursive , actual .IoTimeRecursive ); err != nil {
64- logrus .Printf ("blkio IoTimeRecursive do not match - %s\n " , err )
65- t .Fail ()
55+ t .Errorf ("blkio IoTimeRecursive do not match - %s\n " , err )
6656 }
6757}
6858
6959func expectThrottlingDataEquals (t * testing.T , expected , actual cgroups.ThrottlingData ) {
7060 if expected != actual {
71- logrus .Printf ("Expected throttling data %v but found %v\n " , expected , actual )
72- t .Fail ()
61+ t .Errorf ("Expected throttling data %v but found %v\n " , expected , actual )
7362 }
7463}
7564
7665func expectHugetlbStatEquals (t * testing.T , expected , actual cgroups.HugetlbStats ) {
7766 if expected != actual {
78- logrus .Printf ("Expected hugetlb stats %v but found %v\n " , expected , actual )
79- t .Fail ()
67+ t .Errorf ("Expected hugetlb stats %v but found %v\n " , expected , actual )
8068 }
8169}
8270
@@ -86,38 +74,69 @@ func expectMemoryStatEquals(t *testing.T, expected, actual cgroups.MemoryStats)
8674 expectMemoryDataEquals (t , expected .KernelUsage , actual .KernelUsage )
8775
8876 if expected .UseHierarchy != actual .UseHierarchy {
89- logrus .Printf ("Expected memory use hierarchy %v, but found %v\n " , expected .UseHierarchy , actual .UseHierarchy )
90- t .Fail ()
77+ t .Errorf ("Expected memory use hierarchy %v, but found %v\n " , expected .UseHierarchy , actual .UseHierarchy )
9178 }
9279
9380 for key , expValue := range expected .Stats {
9481 actValue , ok := actual .Stats [key ]
9582 if ! ok {
96- logrus .Printf ("Expected memory stat key %s not found\n " , key )
97- t .Fail ()
83+ t .Errorf ("Expected memory stat key %s not found\n " , key )
9884 }
9985 if expValue != actValue {
100- logrus .Printf ("Expected memory stat value %d but found %d\n " , expValue , actValue )
101- t .Fail ()
86+ t .Errorf ("Expected memory stat value %d but found %d\n " , expValue , actValue )
10287 }
10388 }
10489}
10590
10691func expectMemoryDataEquals (t * testing.T , expected , actual cgroups.MemoryData ) {
10792 if expected .Usage != actual .Usage {
108- logrus .Printf ("Expected memory usage %d but found %d\n " , expected .Usage , actual .Usage )
109- t .Fail ()
93+ t .Errorf ("Expected memory usage %d but found %d\n " , expected .Usage , actual .Usage )
11094 }
11195 if expected .MaxUsage != actual .MaxUsage {
112- logrus .Printf ("Expected memory max usage %d but found %d\n " , expected .MaxUsage , actual .MaxUsage )
113- t .Fail ()
96+ t .Errorf ("Expected memory max usage %d but found %d\n " , expected .MaxUsage , actual .MaxUsage )
11497 }
11598 if expected .Failcnt != actual .Failcnt {
116- logrus .Printf ("Expected memory failcnt %d but found %d\n " , expected .Failcnt , actual .Failcnt )
117- t .Fail ()
99+ t .Errorf ("Expected memory failcnt %d but found %d\n " , expected .Failcnt , actual .Failcnt )
118100 }
119101 if expected .Limit != actual .Limit {
120- logrus .Printf ("Expected memory limit %d but found %d\n " , expected .Limit , actual .Limit )
121- t .Fail ()
102+ t .Errorf ("Expected memory limit %d but found %d\n " , expected .Limit , actual .Limit )
103+ }
104+ }
105+
106+ func expectCPUUsageEqual (t * testing.T , expected , actual cgroups.CpuUsage ) {
107+ if expected .TotalUsage != actual .TotalUsage {
108+ t .Errorf ("Expected CPU total usage %d but found %d\n " ,
109+ expected .TotalUsage , actual .TotalUsage )
110+ }
111+
112+ for cpuID , actualCPUUsage := range actual .PercpuUsage {
113+ if expected .PercpuUsage [cpuID ] != actualCPUUsage {
114+ t .Errorf ("Expected CPU%d usage %d but found %d\n " ,
115+ cpuID , expected .PercpuUsage [cpuID ], actualCPUUsage )
116+ }
117+ }
118+
119+ for cpuID , actualKernelModeUsage := range actual .PercpuUsageInKernelmode {
120+ if expected .PercpuUsageInKernelmode [cpuID ] != actualKernelModeUsage {
121+ t .Errorf ("Expected CPU%d usage in kernel mode %d but found %d\n " ,
122+ cpuID , expected .PercpuUsageInKernelmode [cpuID ], actualKernelModeUsage )
123+ }
124+ }
125+
126+ for cpuID , actualUserModeUsage := range actual .PercpuUsageInUsermode {
127+ if expected .PercpuUsageInUsermode [cpuID ] != actualUserModeUsage {
128+ t .Errorf ("Expected CPU%d usage in user mode %d but found %d\n " ,
129+ cpuID , expected .PercpuUsageInUsermode [cpuID ], actualUserModeUsage )
130+ }
131+ }
132+
133+ if expected .UsageInKernelmode != actual .UsageInKernelmode {
134+ t .Errorf ("Expected CPU usage in kernel mode %d but found %d\n " ,
135+ expected .UsageInKernelmode , actual .UsageInKernelmode )
136+ }
137+
138+ if expected .UsageInUsermode != actual .UsageInUsermode {
139+ t .Errorf ("Expected CPU usage in user mode %d but found %d\n " ,
140+ expected .UsageInUsermode , actual .UsageInUsermode )
122141 }
123142}
0 commit comments