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 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70
|
//go:build linux
// +build linux
package memlimit
import (
"testing"
)
func TestFromCgroup(t *testing.T) {
if expected == 0 {
t.Skip()
}
limit, err := FromCgroup()
if cgVersion == 0 && err != ErrNoCgroup {
t.Fatalf("FromCgroup() error = %v, wantErr %v", err, ErrNoCgroup)
}
if err != nil {
t.Fatalf("FromCgroup() error = %v, wantErr %v", err, nil)
}
if limit != expected {
t.Fatalf("FromCgroup() got = %v, want %v", limit, expected)
}
}
func TestFromCgroupHybrid(t *testing.T) {
if expected == 0 {
t.Skip()
}
limit, err := FromCgroupHybrid()
if cgVersion == 0 && err != ErrNoCgroup {
t.Fatalf("FromCgroupHybrid() error = %v, wantErr %v", err, ErrNoCgroup)
}
if err != nil {
t.Fatalf("FromCgroupHybrid() error = %v, wantErr %v", err, nil)
}
if limit != expected {
t.Fatalf("FromCgroupHybrid() got = %v, want %v", limit, expected)
}
}
func TestFromCgroupV1(t *testing.T) {
if expected == 0 || cgVersion != 1 {
t.Skip()
}
limit, err := FromCgroupV1()
if err != nil {
t.Fatalf("FromCgroupV1() error = %v, wantErr %v", err, nil)
}
if limit != expected {
t.Fatalf("FromCgroupV1() got = %v, want %v", limit, expected)
}
}
func TestFromCgroupV2(t *testing.T) {
if expected == 0 || cgVersion != 2 {
t.Skip()
}
limit, err := FromCgroupV2()
if err != nil {
t.Fatalf("FromCgroupV2() error = %v, wantErr %v", err, nil)
}
if limit != expected {
t.Fatalf("FromCgroupV2() got = %v, want %v", limit, expected)
}
}
|