wehub-resource-sync
76 lines
2.4 KiB
JavaScript
76 lines
2.4 KiB
JavaScript
import fs from "fs";
|
|
import { pipeline } from "stream/promises";
|
|
import { from as copyFrom } from "pg-copy-streams";
|
|
import path from "path";
|
|
import { branchTests } from "./branches.js";
|
|
import { databaseTests } from "./databases.js";
|
|
import { docsTests } from "./docs.js";
|
|
import { logTests } from "./logs.js";
|
|
import { assertEqualRows } from "../helpers.js";
|
|
import { mergeTests } from "./merge.js";
|
|
import { tableTests } from "./table.js";
|
|
import { schemaTests } from "./schemas.js";
|
|
import { tagsTests } from "./tags.js";
|
|
import { viewsTests } from "./views.js";
|
|
import { diffTests } from "./diffs.js";
|
|
|
|
const args = process.argv.slice(2);
|
|
const testDataPath = args[3];
|
|
|
|
export default async function runWorkbenchTests(database) {
|
|
await runTests(database, databaseTests, "database");
|
|
await runTests(database, branchTests, "branches");
|
|
await runTests(database, schemaTests, "schemas");
|
|
await runTests(database, logTests, "logs");
|
|
await runTests(database, mergeTests, "merge");
|
|
await runTests(database, tableTests, "tables");
|
|
await runTests(database, docsTests, "docs");
|
|
await runTests(database, tagsTests, "tags");
|
|
await runTests(database, viewsTests, "views");
|
|
await runTests(database, diffTests, "diffs");
|
|
}
|
|
|
|
async function runTests(database, tests, name) {
|
|
console.log("Running tests for", name);
|
|
await Promise.all(
|
|
tests.map(async (test) => {
|
|
if (test.skip) return;
|
|
|
|
if (test.file) {
|
|
const filePath = path.resolve(testDataPath, test.file);
|
|
try {
|
|
// TODO: Is it possible to test the COPY FROM output?
|
|
const ingestStream = database.client.query(copyFrom(test.q));
|
|
const sourceStream = fs.createReadStream(filePath);
|
|
await pipeline([sourceStream, ingestStream]);
|
|
} catch (err) {
|
|
console.log("Query errored:", test.q);
|
|
console.error(err);
|
|
process.exit(1);
|
|
}
|
|
|
|
return;
|
|
}
|
|
|
|
return database
|
|
.query(test.q, test.p)
|
|
.then((data) => {
|
|
assertEqualRows(test, data);
|
|
})
|
|
.catch((err) => {
|
|
if (test.expectedErr) {
|
|
if (err.message.includes(test.expectedErr)) {
|
|
return;
|
|
} else {
|
|
console.log("Query error did not match expected:", test.q);
|
|
}
|
|
} else {
|
|
console.log("Query errored:", test.q);
|
|
}
|
|
console.error(err);
|
|
process.exit(1);
|
|
});
|
|
})
|
|
);
|
|
}
|