Refactor around the fileRead in test files & add unit test cases

https-sam · https-sam · commit 5c2e9c181c63 · 2023-03-20T22:36:32.000-06:00
diff --git a/__tests__/code-excution/python.ts b/__tests__/code-excution/python.ts
@@ -14,21 +14,35 @@ describe("Python transform into executable", () => {
 
 
 describe("Python code excution test", () => {
-  let code: string
+  let fileJobs: Promise<any>[] = []
+  let codes: string[]
+  fileJobs.push(fs.promises.readFile(`${__dirname}/../test-code/python/counter.py`))
+  fileJobs.push(fs.promises.readFile(`${__dirname}/../test-code/python/helloWorld.py`))
   beforeAll((done) => {
-    fs.readFile(`${__dirname}/../test-code/python/counter.py`, 'utf8', (err, data) => {
-      code = data
+    Promise.all(fileJobs)
+    .then(outputs => {
+      codes = outputs
       done()
     })
   })
   it('should calculate a frequency in string', async () => {
     const codeContext: CodeContext = {
-      code: code,
+      code: codes[0],
       functionName: "calculate"
     }
     const worker = new JobWorker("python3", codeContext);
     const response = await worker.startContainer()
     expect(response.codeOutput).toBe("Counter({'a': 2, 's': 2, 'w': 2, 'e': 2, 'l': 1, 'd': 1, 'k': 1})")
     await worker.cleanupJob();
   })
+  it('should return string "Hello World"', async () => {
+    const codeContext: CodeContext = {
+      code: codes[1],
+      functionName: "run"
+    }
+    const worker = new JobWorker("python3", codeContext);
+    const response = await worker.startContainer()
+    expect(response.codeOutput).toBe("Hello World!")
+    await worker.cleanupJob();
+  });
 })
diff --git a/__tests__/test-code/python/helloWorld.py b/__tests__/test-code/python/helloWorld.py
@@ -0,0 +1,4 @@
+class Execute:
+  def run(self):
+    return "Hello World!"
+  
diff --git a/dockerfiles/python3.Dockerfile b/dockerfiles/python3.Dockerfile
@@ -1,3 +1,5 @@
 FROM python:3.8-slim-buster
+
 WORKDIR /src
+
 CMD ["python3", "target.py"]
diff --git a/src/index.ts b/src/index.ts
@@ -0,0 +1,17 @@
+import { CodeContext } from "./types/worker";
+import JobWorker from "./worker";
+import fs from "fs"
+
+fs.readFile(`${__dirname}/../__tests__/test-code/python/counter.py`, 'utf8', (err, data) => {
+  const codeContext: CodeContext = {
+    code: data,
+    functionName: "calculate"
+  }
+  const worker = new JobWorker("python3", codeContext);
+  worker.startContainer()
+  .then((output) => {
+    console.log(output) 
+  }).catch(e => {
+    console.log(e)
+  })
+})
