clistat/cgroup_internal_test.go at main · coder/clistat · GitHub

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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
package clistat

import (
	"testing"

	"github.com/stretchr/testify/require"
)

func TestCurrentProcCgroup(t *testing.T) {
	t.Parallel()

	tests := []struct {
		name        string
		procFile    string
		expectError string
		expectPath  string
	}{
		{
			name:       "RootPath",
			procFile:   `0::/`,
			expectPath: "/",
		},
		{
			name:       "SingleLevelPath",
			procFile:   `0::/init.slice`,
			expectPath: "/init.slice",
		},
		{
			name:       "MultipleLevelSlice",
			procFile:   `0::/wibble/wobble/init.slice`,
			expectPath: "/wibble/wobble/init.slice",
		},
		{
			name: "MixOfHierachy",
			procFile: `1:net_cls:/
0::/`,
			expectPath: "/",
		},
		{
			name: "MixOfHierarchyPaths",
			procFile: `1:net_cls:/init.slice
0::/`,
			expectPath: "/",
		},
		{
			name: "MixOfHierarchyPaths/Order",
			procFile: `0::/
1:net_cls:/init.slice`,
			expectPath: "/",
		},
		{
			name: "MixOfHierarchyPaths/Paths",
			procFile: `0::/init.slice
1:net_cls:/`,
			expectPath: "/init.slice",
		},
		{
			name:       "ExtraColonsInPath",
			procFile:   `0::/system.slice/kubepods-burstable.slice:cri-containerd:d24f9cc`,
			expectPath: "/system.slice/kubepods-burstable.slice:cri-containerd:d24f9cc",
		},
		{
			name: "ExtraColonsInPathWithMixedHierarchy",
			procFile: `1:net_cls:/init.slice
0::/system.slice/kubepods.slice:cri-containerd:abc123`,
			expectPath: "/system.slice/kubepods.slice:cri-containerd:abc123",
		},
		{
			name:        "MalformedEntryTooFewFields",
			procFile:    `0`,
			expectError: "parse entry",
		},
		{
			name:        "MissingHierarchy0",
			procFile:    `1:net_cls:/`,
			expectError: "no cgroup entry for hierarchy 0 found",
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			t.Parallel()

			fs := initFS(t, map[string]string{procSelfCgroup: tt.procFile})

			path, err := currentProcCgroup(fs)
			if tt.expectError != "" {
				require.ErrorContains(t, err, tt.expectError)
			} else {
				require.NoError(t, err)
				require.Equal(t, tt.expectPath, path)
			}
		})
	}
}