From 8af94f9002b655a348cdf03501ea05fe0f6f2a56 Mon Sep 17 00:00:00 2001
From: Radon Rosborough <radon.neon@gmail.com>
Date: Sat, 22 Aug 2020 19:54:01 -0600
Subject: [PATCH] Reduce concurrency only when building prod image

---
 Dockerfile.prod            | 2 +-
 backend/src/test-runner.ts | 9 +++++++--
 2 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/Dockerfile.prod b/Dockerfile.prod
index 8de085f..a2fb906 100644
--- a/Dockerfile.prod
+++ b/Dockerfile.prod
@@ -68,4 +68,4 @@ RUN sudo cp -a /tmp/riju/* /home/docker/src/ && rm -rf /tmp/riju
 
 WORKDIR /home/docker/src
 RUN sudo deluser docker sudo
-RUN RIJU_PRIVILEGED=1 yarn test
+RUN RIJU_PRIVILEGED=1 TIMEOUT_SECS=20 yarn test
diff --git a/backend/src/test-runner.ts b/backend/src/test-runner.ts
index 36ab925..3010692 100644
--- a/backend/src/test-runner.ts
+++ b/backend/src/test-runner.ts
@@ -10,8 +10,13 @@ import { v4 as getUUID } from "uuid";
 import * as api from "./api";
 import { LangConfig, langs } from "./langs";
 
-const TIMEOUT_SECS = 5;
-const CONCURRENCY = 1;
+function parseIntOr(thing: any, def: number) {
+  const num = parseInt(thing);
+  return Number.isNaN(num) ? def : num;
+}
+
+const TIMEOUT_SECS = parseIntOr(process.env.TIMEOUT, 5);
+const CONCURRENCY = parseIntOr(process.env.CONCURRENCY, 2);
 
 function findPosition(str: string, idx: number) {
   const lines = str.substring(0, idx).split("\n");