Merge remote-tracking branch 'origin/main' into only-output-on-failure

src/util.test.ts

@@ -1,5 +1,6 @@
 import test from 'ava';
 import * as fs from 'fs';
+import * as os from "os";
 
 import {silenceDebugOutput} from './testing-utils';
 import * as util from './util';
@@ -11,3 +12,55 @@ test('getToolNames', t => {
   const toolNames = util.getToolNames(input);
   t.deepEqual(toolNames, ["CodeQL command-line toolchain", "ESLint"]);
 });
+
+test('getMemoryFlag() should return the correct --ram flag', t => {
+
+  const totalMem = Math.floor(os.totalmem() / (1024 * 1024));
+
+  const tests = {
+    "": `--ram=${totalMem - 256}`,
+    "512": "--ram=512",
+  };
+
+  for (const [input, expectedFlag] of Object.entries(tests)) {
+
+    process.env['INPUT_RAM'] = input;
+
+    const flag = util.getMemoryFlag();
+    t.deepEqual(flag, expectedFlag);
+  }
+});
+
+test('getMemoryFlag() throws if the ram input is < 0 or NaN', t => {
+  for (const input of ["-1", "hello!"]) {
+    process.env['INPUT_RAM'] = input;
+    t.throws(util.getMemoryFlag);
+  }
+});
+
+test('getThreadsFlag() should return the correct --threads flag', t => {
+
+  const numCpus = os.cpus().length;
+
+  const tests = {
+    "0": "--threads=0",
+    "1": "--threads=1",
+    [`${numCpus + 1}`]: `--threads=${numCpus}`,
+    [`${-numCpus - 1}`]: `--threads=${-numCpus}`
+  };
+
+  for (const [input, expectedFlag] of Object.entries(tests)) {
+
+    process.env['INPUT_THREADS'] = input;
+
+    const flag = util.getThreadsFlag();
+    t.deepEqual(flag, expectedFlag);
+  }
+});
+
+test('getThreadsFlag() throws if the ram input is < 0 or NaN', t => {
+  for (const input of ["hello!"]) {
+    process.env['INPUT_THREADS'] = input;
+    t.throws(util.getThreadsFlag);
+  }
+});