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,208 @@
+import { Database } from "./database.js";
+import { assertEqualRows, getConfig } from "./helpers.js";
+import {
+  doltAddFields,
+  doltCheckoutFields,
+  doltCommitFields,
+  countFields,
+} from "./fields.js";
+import { mergeMatcher } from "./workbenchTests/matchers.js";
+
+const tests = [
+  {
+    q: "create table test (pk int, value int, primary key(pk))",
+    res: {
+      command: "CREATE",
+      rowCount: null,
+      oid: null,
+      rows: [],
+      fields: [],
+    },
+  },
+  {
+    q: "select * from test",
+    res: {
+      command: "SELECT",
+      rowCount: 0,
+      oid: null,
+      rows: [],
+      fields: [
+        {
+          name: "pk",
+          tableID: 0, // TODO: need to be filled? Got 16859 from Postgres
+          columnID: 1,
+          dataTypeID: 23,
+          dataTypeSize: 4,
+          dataTypeModifier: -1,
+          format: "text",
+        },
+        {
+          name: "value",
+          tableID: 0, // TODO: need to be filled? Got 16859 from Postgres
+          columnID: 2,
+          dataTypeID: 23,
+          dataTypeSize: 4,
+          dataTypeModifier: -1,
+          format: "text",
+        },
+      ],
+    },
+  },
+  {
+    q: "insert into test (pk, value) values (0,0)",
+    res: {
+      command: "INSERT",
+      rowCount: 1,
+      oid: 0,
+      rows: [],
+      fields: [],
+    },
+  },
+  {
+    q: "select * from test",
+    res: {
+      command: "SELECT",
+      rowCount: 1,
+      oid: null,
+      rows: [{ pk: 0, value: 0 }],
+      fields: [
+        {
+          name: "pk",
+          tableID: 0,
+          columnID: 1,
+          dataTypeID: 23,
+          dataTypeSize: 4,
+          dataTypeModifier: -1,
+          format: "text",
+        },
+        {
+          name: "value",
+          tableID: 0,
+          columnID: 2,
+          dataTypeID: 23,
+          dataTypeSize: 4,
+          dataTypeModifier: -1,
+          format: "text",
+        },
+      ],
+    },
+  },
+  {
+    q: "select dolt_add('-A');",
+    res: {
+      command: "SELECT",
+      rowCount: 1,
+      oid: null,
+      rows: [{ dolt_add: ["0"] }],
+      fields: doltAddFields,
+    },
+  },
+  {
+    q: "select dolt_commit('-m', 'my commit')",
+    res: {
+      command: "SELECT",
+      rowCount: 1,
+      oid: null,
+      rows: [{ dolt_commit: [""] }],
+      fields: doltCommitFields,
+    },
+  },
+  {
+    q: "select COUNT(*) FROM dolt.log",
+    res: {
+      command: "SELECT",
+      rowCount: 1,
+      oid: null,
+      rows: [{ count: "3" }],
+      fields: countFields,
+    },
+  },
+  {
+    q: "select dolt_checkout('-b', 'mybranch')",
+    res: {
+      command: "SELECT",
+      rowCount: 1,
+      oid: null,
+      rows: [{ dolt_checkout: ["0", "Switched to branch 'mybranch'"] }],
+      fields: doltCheckoutFields,
+    },
+  },
+  {
+    q: "insert into test (pk, value) values (1,1),(2,3)",
+    res: {
+      command: "INSERT",
+      rowCount: 2,
+      oid: 0,
+      rows: [],
+      fields: [],
+    },
+  },
+  {
+    q: "select dolt_commit('-a', '-m', 'my commit2')",
+    res: {
+      command: "SELECT",
+      rowCount: 1,
+      oid: null,
+      rows: [{ dolt_commit: [""] }],
+      fields: doltCommitFields,
+    },
+  },
+  {
+    q: "select dolt_checkout('main')",
+    res: {
+      command: "SELECT",
+      rowCount: 1,
+      oid: null,
+      rows: [{ dolt_checkout: ["0", "Switched to branch 'main'"] }],
+      fields: doltCheckoutFields,
+    },
+  },
+  {
+    q: "select dolt_merge('mybranch')",
+    res: {
+      command: "SELECT",
+      rowCount: 1,
+      oid: null,
+      rows: [
+        {
+          dolt_merge: ["", "1", "0", "merge successful"],
+        },
+      ],
+      fields: [],
+    },
+    matcher: mergeMatcher,
+  },
+  {
+    q: "select COUNT(*) FROM dolt.log",
+    res: {
+      command: "SELECT",
+      rowCount: 1,
+      oid: null,
+      rows: [{ count: "4" }],
+      fields: countFields,
+    },
+  },
+];
+
+async function main() {
+  const database = new Database(getConfig());
+
+  await Promise.all(
+    tests.map((test) => {
+      return database
+        .query(test.q)
+        .then((data) => {
+          assertEqualRows(test, data);
+        })
+        .catch((err) => {
+          console.error(err);
+          process.exit(1);
+        });
+    })
+  );
+
+  database.close();
+  process.exit(0);
+}
+
+main();