1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
|
package rlimit
import (
"testing"
"github.com/cilium/ebpf/internal"
"github.com/cilium/ebpf/internal/unix"
qt "github.com/frankban/quicktest"
)
func TestRemoveMemlock(t *testing.T) {
var before unix.Rlimit
qt.Assert(t, unix.Prlimit(0, unix.RLIMIT_MEMLOCK, nil, &before), qt.IsNil)
err := RemoveMemlock()
qt.Assert(t, err, qt.IsNil)
var after unix.Rlimit
qt.Assert(t, unix.Prlimit(0, unix.RLIMIT_MEMLOCK, nil, &after), qt.IsNil)
// We can't use testutils here due to an import cycle.
version, err := internal.KernelVersion()
qt.Assert(t, err, qt.IsNil)
if version.Less(unsupportedMemcgAccounting.MinimumVersion) {
qt.Assert(t, after.Cur, qt.Equals, uint64(unix.RLIM_INFINITY), qt.Commentf("cur should be INFINITY"))
qt.Assert(t, after.Max, qt.Equals, uint64(unix.RLIM_INFINITY), qt.Commentf("max should be INFINITY"))
} else {
qt.Assert(t, after.Cur, qt.Equals, before.Cur, qt.Commentf("cur should be unchanged"))
qt.Assert(t, after.Max, qt.Equals, before.Max, qt.Commentf("max should be unchanged"))
}
}
|
