unit tests

https-sam · https-sam · commit 18a3b36b7819 · 2023-03-19T17:31:33.000-06:00
diff --git a/__tests__/code-excution/python.ts b/__tests__/code-excution/python.ts
@@ -7,11 +7,11 @@ describe("Python transform into executable", () => {
   const worker = new JobWorker();
    it('should instantiate an object and execute the function', () => {
     const functionName = "solution"
-    const code = worker.transformCodeIntoExecutable("python3", {
+    const codeContext = worker.transformCodeIntoExecutable("python3", {
       code: "",
       functionName: functionName
     })
-    expect(code).toBe(`\nprint(${mainClassName}().${functionName}())`)
+    expect(codeContext).toBe(`\nprint(${mainClassName}().${functionName}())`)
    })
 })
 
diff --git a/__tests__/worker.ts b/__tests__/worker.ts
@@ -1,5 +1,4 @@
-import { mainClassName } from "../src/config";
-import { CodeContext, ContainerInitialization } from "../src/types/worker";
+import { ContainerInitialization } from "../src/types/worker";
 import JobWorker from "../src/worker";
 
 
diff --git a/dockerfiles/python3.Dockerfile b/dockerfiles/python3.Dockerfile
@@ -1,4 +1,3 @@
 FROM python:3.8-slim-buster
 WORKDIR /src
-# COPY ./test.py /src
 CMD ["python3", "target.py"]
diff --git a/src/config.ts b/src/config.ts
@@ -1 +1,3 @@
 export const mainClassName = "Execute"
+export const containerMemLimit = "--memory=100M"
+export const containerCPULimit = "--cpus=0.1"
diff --git a/src/worker.ts b/src/worker.ts
@@ -12,7 +12,7 @@ import {
 	ExecuteContainer,
 	Stdout,
 } from "./types/worker";
-import { mainClassName } from "./config";
+import { containerCPULimit, containerMemLimit, mainClassName } from "./config";
 
 class JobWorker {
 	constructor() {}
@@ -28,7 +28,7 @@ class JobWorker {
 	 */
 	createContainer(language: language): Promise<ContainerInitialization> {
 		return new Promise((resolve, reject) => {
-			const initCommand = `docker create ${language}`;
+			const initCommand = `docker create ${containerMemLimit} ${containerCPULimit} ${language}`;
 			if (!(language in this._fileFormats)) {
 				return reject({
 					error: true,
@@ -103,7 +103,7 @@ class JobWorker {
 	/**
 	 * Removes a container with the provided containerID
 	 */
-	removeContainer(containerID: string): Promise<Stdout> {
+	async removeContainer(containerID: string): Promise<Stdout> {
 		return new Promise((resolve, reject) => {
 			const removeContainer = `docker rm --force ${containerID}`;
 			child.exec(removeContainer, (error, stdout, stderr) => {

Original file line number	Diff line number	Diff line change
`@@ -1 +1,3 @@`
`1`	`1`	`export const mainClassName = "Execute"`
	`2`	`+export const containerMemLimit = "--memory=100M"`
	`3`	`+export const containerCPULimit = "--cpus=0.1"`