🏆 Performance tests with K6 (#303)

tdrive/backend/node/test/performance/index.js

@@ -0,0 +1,84 @@
+const { execSync } = require("child_process");
+const fs = require("fs");
+const path = require("path");
+const http = require("http"); // or 'https' if your URL is https
+
+const COMPANY_ID = "COMPANY_ID";
+const JWT = "JWT";
+const BASE_URL = "https://tdrive.qa.lin-saas.com";
+const shouldDownload = process.argv.includes("--download");
+
+const directory = path.join(__dirname, "./");
+const tests = fs.readdirSync(directory).filter(filePath => fs.statSync(filePath).isDirectory());
+
+// Ensure the assets folder exists
+const assetsDirectory = path.join(__dirname, "assets");
+if (!fs.existsSync(assetsDirectory)) {
+  fs.mkdirSync(assetsDirectory);
+}
+
+// Array of item IDs with corresponding file formats
+const items = [
+  { id: "751416c3-e486-4e2e-8ad6-f6155e4c7c4b", format: "pdf" },
+  { id: "another-item-id", format: "png" },
+  { id: "yet-another-item-id", format: "jpg" },
+  // Add more items and formats as needed
+];
+
+// Function to download the file
+const downloadFile = (url, dest) => {
+  return new Promise((resolve, reject) => {
+    const file = fs.createWriteStream(dest);
+    http
+      .get(url, response => {
+        response.pipe(file);
+        file.on("finish", () => {
+          file.close(resolve);
+        });
+      })
+      .on("error", err => {
+        fs.unlink(dest);
+        reject(err.message);
+      });
+  });
+};
+
+// Function to run the tests
+const runTests = () => {
+  tests.forEach(test => {
+    const testDirectory = path.join(directory, test);
+    const files = fs.readdirSync(testDirectory).filter(file => file.endsWith(".js"));
+
+    files.forEach(file => {
+      console.log(`Running ${test}/${file}...`);
+      execSync(
+        `k6 run -e BACKEND=${BASE_URL} -e COMPANY_ID=${COMPANY_ID} -e JWT=${JWT} ./${test}/${file}`,
+        {
+          stdio: "inherit",
+        },
+      );
+    });
+  });
+};
+
+// Function to download all files
+const downloadAllFiles = async () => {
+  for (const item of items) {
+    const fileUrl = `${BASE_URL}/internal/services/documents/v1/companies/${COMPANY_ID}/item/${item.id}/download`;
+    const localFilePath = path.join(assetsDirectory, `sample.${item.format}`);
+
+    try {
+      await downloadFile(fileUrl, localFilePath);
+      console.log(`File sample.${item.format} downloaded successfully.`);
+    } catch (err) {
+      console.error(`Error downloading file sample.${item.format}:`, err);
+    }
+  }
+};
+
+// Main logic
+if (shouldDownload) {
+  downloadAllFiles().then(runTests);
+} else {
+  runTests();
+}