Add unit tests for getCgroupCpuCountFromCpus

github · Feb 16, 2024 · 8cb81db · 8cb81db
1 parent ef0a773
commit 8cb81db
Show file tree

Hide file tree

Showing 6 changed files with 59 additions and 4 deletions.
diff --git a/lib/util.js b/lib/util.js
diff --git a/lib/util.js.map b/lib/util.js.map
diff --git a/lib/util.test.js b/lib/util.test.js
diff --git a/lib/util.test.js.map b/lib/util.test.js.map
diff --git a/src/util.test.ts b/src/util.test.ts
@@ -449,3 +449,36 @@ for (const [
     versionStub.restore();
   });
 }
+
+test("getCgroupCpuCountFromCpus calculates the number of CPUs correctly", async (t) => {
+  await util.withTmpDir(async (tmpDir: string) => {
+    const testCpuFile = `${tmpDir}/cpus-file`;
+    fs.writeFileSync(testCpuFile, "1, 9-10\n", "utf-8");
+    t.deepEqual(
+      util.getCgroupCpuCountFromCpus(testCpuFile, getRunnerLogger(true)),
+      3,
+    );
+  });
+});
+
+test("getCgroupCpuCountFromCpus returns undefined if the CPU file doesn't exist", async (t) => {
+  await util.withTmpDir(async (tmpDir: string) => {
+    const testCpuFile = `${tmpDir}/cpus-file`;
+    t.false(fs.existsSync(testCpuFile));
+    t.deepEqual(
+      util.getCgroupCpuCountFromCpus(testCpuFile, getRunnerLogger(true)),
+      undefined,
+    );
+  });
+});
+
+test("getCgroupCpuCountFromCpus returns undefined if the CPU file exists but is empty", async (t) => {
+  await util.withTmpDir(async (tmpDir: string) => {
+    const testCpuFile = `${tmpDir}/cpus-file`;
+    fs.writeFileSync(testCpuFile, "\n", "utf-8");
+    t.deepEqual(
+      util.getCgroupCpuCountFromCpus(testCpuFile, getRunnerLogger(true)),
+      undefined,
+    );
+  });
+});
diff --git a/src/util.ts b/src/util.ts
@@ -440,7 +440,7 @@ function getCgroupCpuCountFromCpuMax(
 /**
  * Gets the number of available cores listed in the cgroup cpuset.cpus file at the given path.
  */
-function getCgroupCpuCountFromCpus(
+export function getCgroupCpuCountFromCpus(
   cpusFile: string,
   logger: Logger,
 ): number | undefined {