chore: import upstream snapshot with attribution

2026-07-13 12:32:25 +08:00
commit e014feafe1
2285 changed files with 1131979 additions and 0 deletions
@@ -0,0 +1,75 @@
+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);
+        });
+    })
+  );
+}