Test python runner

container/tests/unit/use-cases/run-python.spec.js

@@ -0,0 +1,70 @@
+const { LANGUAGE } = require("../../../src/common/constants");
+const { main } = require("../../../src/use-cases/run-python");
+
+const mockGetTemplateByLanguage = jest.fn();
+jest.mock("../../../src/common/file", () => {
+  return {
+    getTemplateByLanguage: (params) => mockGetTemplateByLanguage(params),
+  };
+});
+
+const mockWriteFile = jest.fn();
+jest.mock("node:fs/promises", () => {
+  return {
+    writeFile: (filename, content) => mockWriteFile(filename, content),
+  };
+});
+
+const mockRandomUUID = jest.fn();
+
+const uuid = "6c095a0d-b1da-40cc-aa0b-96580ef29b01";
+mockRandomUUID.mockReturnValueOnce(uuid);
+
+jest.mock("node:crypto", () => {
+  return {
+    randomUUID: () => mockRandomUUID(),
+  };
+});
+
+const mockCommand = jest.fn();
+jest.mock(`../../../src/common/command`, () => {
+  return {
+    command: (program, args) => mockCommand(program, args),
+  };
+});
+
+describe("Run Python Use Case", () => {
+  it("should be able to run Python code", async () => {
+    const template = "# @code-here";
+    mockGetTemplateByLanguage.mockResolvedValueOnce(template);
+
+    const inputs = [
+      {
+        value: 5,
+      },
+    ];
+
+    const code = "def run(value):\n  return pow(value, 2)\n";
+
+    const value = inputs[0].value;
+    const expected = [{ input: value, output: value ** 2 }];
+    mockCommand.mockResolvedValueOnce(JSON.stringify(expected));
+
+    const response = await main({ code, inputs });
+
+    expect(mockGetTemplateByLanguage).toHaveBeenCalledWith({
+      language: LANGUAGE.PYTHON,
+    });
+
+    const fileName = `/tmp/${uuid}.py`;
+    expect(mockWriteFile).toHaveBeenCalledWith(
+      expect.stringMatching(fileName),
+      code
+    );
+    expect(mockCommand).toHaveBeenCalledWith("python", [
+      fileName,
+      JSON.stringify(inputs),
+    ]);
+    expect(response).toStrictEqual(expected);
+  });
+});