Montassar Ghanmy
85 lines
2.4 KiB
JavaScript
85 lines
2.4 KiB
JavaScript
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();
|
|
}
|