Add test for cgroups load when missing hierarchy in one subsystem

MichaelKatsoulis · MichaelKatsoulis · commit 7d825b29aecc · 2018-11-02T11:18:26.000+02:00
Signed-off-by: MichaelKatsoulis &lt;michaelkatsoulis88@gmail.com&gt;
diff --git a/cgroup_test.go b/cgroup_test.go
@@ -197,6 +197,20 @@ func checkTaskid(mock *mockCgroup, path string, expected int) error {
 	return nil
 }
 
+func mockNewNotInRdma(subsystems []Subsystem, path Path, resources *specs.LinuxResources) (Cgroup, error) {
+	for _, s := range subsystems {
+		if s.Name() != Rdma {
+			if err := initializeSubsystem(s, path, resources); err != nil {
+				return nil, err
+			}
+		}
+	}
+	return &cgroup{
+		path:       path,
+		subsystems: subsystems,
+	}, nil
+}
+
 func TestLoad(t *testing.T) {
 	mock, err := newMock()
 	if err != nil {
@@ -218,6 +232,36 @@ func TestLoad(t *testing.T) {
 	}
 }
 
+func TestLoadWithMissingSubsystems(t *testing.T) {
+	mock, err := newMock()
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer mock.delete()
+	subsystems, err := mock.hierarchy()
+	if err != nil {
+		t.Error(err)
+		return
+	}
+	control, err := mockNewNotInRdma(subsystems, StaticPath("test"), &specs.LinuxResources{})
+	if err != nil {
+		t.Error(err)
+		return
+	}
+	if control, err = Load(mock.hierarchy, StaticPath("test")); err != nil {
+		t.Error(err)
+		return
+	}
+	if control == nil {
+		t.Error("control is nil")
+		return
+	}
+	if len(control.Subsystems()) != len(subsystems) - 1 {
+		t.Error("wrong number of active subsystems")
+		return
+	}
+}
+
 func TestDelete(t *testing.T) {
 	mock, err := newMock()
 	if err != nil {
