feat: Linux Control Group version 2 API support cgroup v2 (aws-greeng…

…rass#1262)

src/test/java/com/aws/greengrass/util/platforms/unix/linux/LinuxSystemResourceControllerV2Test.java

@@ -0,0 +1,104 @@
+package com.aws.greengrass.util.platforms.unix.linux;
+
+import com.aws.greengrass.lifecyclemanager.GreengrassService;
+import com.aws.greengrass.testcommons.testutilities.GGExtension;
+import com.aws.greengrass.util.Utils;
+import com.aws.greengrass.util.platforms.SystemResourceController;
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.condition.DisabledOnOs;
+import org.junit.jupiter.api.condition.OS;
+import org.junit.jupiter.api.extension.ExtendWith;
+import org.mockito.Mock;
+import org.mockito.MockedStatic;
+import org.mockito.junit.jupiter.MockitoExtension;
+
+import java.io.File;
+import java.io.IOException;
+import java.nio.charset.StandardCharsets;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.nio.file.Paths;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.mockito.Mockito.doReturn;
+import static org.mockito.Mockito.mockStatic;
+
+@ExtendWith({MockitoExtension.class, GGExtension.class})
+@DisabledOnOs(OS.WINDOWS)
+class LinuxSystemResourceControllerV2Test {
+    private SystemResourceController systemResourceController = new LinuxSystemResourceControllerV2(new LinuxPlatform());
+    @Mock
+    GreengrassService component;
+
+    private static final String FILE_PATH = "/cgroupv2test";
+    private static final String CGROUP_MEMORY_LIMIT_FILE_NAME = "memory.txt";
+    private static final String CGROUP_CPU_LIMIT_FILE_NAME = "cpu.txt";
+    private static final long MEMORY_IN_KB = 2048000;
+    private static final double CPU_TIME = 0.5;
+    private static final String COMPONENT_NAME = "testComponentName";
+
+    @Test
+    void GIVEN_cgroupv2_WHEN_memory_limit_updated_THEN_memory_limit_file_updated() throws IOException {
+        Map<String, Object> resourceLimit = new HashMap<>();
+        resourceLimit.put("memory", String.valueOf(MEMORY_IN_KB));
+        doReturn("testComponentName").when(component).getServiceName();
+
+        Path path = Paths.get(FILE_PATH + "/" + CGROUP_MEMORY_LIMIT_FILE_NAME);
+
+        Path componentNameFolderPath = Paths.get(FILE_PATH);
+        Utils.createPaths(componentNameFolderPath);
+        File file = new File(FILE_PATH + "/" + CGROUP_MEMORY_LIMIT_FILE_NAME);
+        if (!Files.exists(path)) {
+            file.createNewFile();
+        }
+
+        try (MockedStatic<CgroupV2> utilities = mockStatic(CgroupV2.class)) {
+            utilities.when(() -> CgroupV2.getComponentMemoryMaxPath(COMPONENT_NAME))
+                    .thenReturn(path);
+            utilities.when(() -> CgroupV2.getSubsystemComponentPath(COMPONENT_NAME))
+                    .thenReturn(componentNameFolderPath);
+            systemResourceController.updateResourceLimits(component, resourceLimit);
+        }
+
+        List<String> mounts = Files.readAllLines(path);
+        assertEquals(String.valueOf(MEMORY_IN_KB * 1024), mounts.get(0));
+
+        Files.deleteIfExists(path);
+        Files.deleteIfExists(componentNameFolderPath);
+    }
+
+    @Test
+    void GIVEN_cgroupv2_WHEN_cpu_limit_updated_THEN_cpu_limit_file_updated() throws IOException {
+        Map<String, Object> resourceLimit = new HashMap<>();
+        resourceLimit.put("cpus", String.valueOf(CPU_TIME));
+        doReturn("testComponentName").when(component).getServiceName();
+
+        Path path = Paths.get(FILE_PATH + "/" + CGROUP_CPU_LIMIT_FILE_NAME);
+
+        Path componentNameFolderPath = Paths.get(FILE_PATH);
+        Utils.createPaths(componentNameFolderPath);
+        File file = new File(FILE_PATH + "/" + CGROUP_CPU_LIMIT_FILE_NAME);
+        if (!Files.exists(path)) {
+            file.createNewFile();
+        }
+
+        Files.write(path, "max 100000".getBytes(StandardCharsets.UTF_8));
+
+        try (MockedStatic<CgroupV2> utilities = mockStatic(CgroupV2.class)) {
+            utilities.when(() -> CgroupV2.getComponentCpuMaxPath(COMPONENT_NAME))
+                    .thenReturn(path);
+            utilities.when(() -> CgroupV2.getSubsystemComponentPath(COMPONENT_NAME))
+                    .thenReturn(componentNameFolderPath);
+            systemResourceController.updateResourceLimits(component, resourceLimit);
+        }
+
+        List<String> mounts = Files.readAllLines(path);
+        assertEquals((int) (CPU_TIME * 100000) + " 100000", mounts.get(0));
+
+        Files.deleteIfExists(path);
+        Files.deleteIfExists(componentNameFolderPath);
+    }
+}