test: fix cgroup mock test

fix cgroup mock test because of containerd/cgroup vendor update Signed-off-by: Ace-Tang <aceapril@126.com>
kata-containers · Aug 19, 2019 · dc38ba7 · dc38ba7
1 parent 6534357
commit dc38ba7
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/virtcontainers/cgroups_test.go b/virtcontainers/cgroups_test.go
@@ -72,11 +72,15 @@ func (m *mockCgroup) Subsystems() []cgroups.Subsystem {
 	return nil
 }
 
-func mockCgroupNew(hierarchy cgroups.Hierarchy, path cgroups.Path, resources *specs.LinuxResources) (cgroups.Cgroup, error) {
+func (m *mockCgroup) Tasks(cgroups.Name, bool) ([]cgroups.Task, error) {
+	return nil, nil
+}
+
+func mockCgroupNew(hierarchy cgroups.Hierarchy, path cgroups.Path, resources *specs.LinuxResources, opts ...cgroups.InitOpts) (cgroups.Cgroup, error) {
 	return &mockCgroup{}, nil
 }
 
-func mockCgroupLoad(hierarchy cgroups.Hierarchy, path cgroups.Path) (cgroups.Cgroup, error) {
+func mockCgroupLoad(hierarchy cgroups.Hierarchy, path cgroups.Path, opts ...cgroups.InitOpts) (cgroups.Cgroup, error) {
 	return &mockCgroup{}, nil
 }