Skip to content

Commit 337ce0c

Browse files
committed
[common-go] Add test for reading psi values
1 parent 638f6b3 commit 337ce0c

File tree

1 file changed

+53
-0
lines changed

1 file changed

+53
-0
lines changed

Diff for: components/common-go/cgroups/cgroups_test.go

+53
Original file line numberDiff line numberDiff line change
@@ -134,3 +134,56 @@ func TestReadSingleValue(t *testing.T) {
134134
})
135135
}
136136
}
137+
138+
func TestReadPSI(t *testing.T) {
139+
scenarios := []struct {
140+
name string
141+
content string
142+
expected PSI
143+
}{
144+
{
145+
name: "psi some",
146+
content: "some avg10=61.00 avg60=64.28 avg300=29.94 total=149969752",
147+
expected: PSI{
148+
Some: 149969752,
149+
Full: 0,
150+
},
151+
},
152+
{
153+
name: "psi full",
154+
content: "full avg10=36.27 avg60=37.15 avg300=17.59 total=93027571",
155+
expected: PSI{
156+
Some: 0,
157+
Full: 93027571,
158+
},
159+
},
160+
{
161+
name: "psi some and full",
162+
content: "some avg10=61.00 avg60=64.28 avg300=29.94 total=149969752\nfull avg10=36.27 avg60=37.15 avg300=17.59 total=93027571",
163+
expected: PSI{
164+
Some: 149969752,
165+
Full: 93027571,
166+
},
167+
},
168+
}
169+
170+
for _, s := range scenarios {
171+
t.Run(s.name, func(t *testing.T) {
172+
f, err := os.CreateTemp("", "cgroup_test*")
173+
if err != nil {
174+
t.Fatal(err)
175+
}
176+
177+
if _, err := f.Write([]byte(s.content)); err != nil {
178+
t.Fatal(err)
179+
}
180+
181+
v, err := ReadPSIValue(f.Name())
182+
if err != nil {
183+
t.Fatal(err)
184+
}
185+
186+
assert.Equal(t, s.expected, v)
187+
})
188+
}
189+
}

0 commit comments

Comments
 (0)