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,64 @@
+const args = process.argv.slice(2);
+const user = args[0];
+const port = args[1];
+const version = args[2];
+
+export const dbName = "postgres";
+
+export function getArgs() {
+  return { user, port };
+}
+
+export function getDoltgresVersion() {
+  return version;
+}
+
+export function getConfig() {
+  const { user, port } = getArgs();
+  return {
+    host: "localhost",
+    port: port,
+    database: dbName,
+    user: user,
+    password: "password",
+  };
+}
+
+export function assertEqualRows(test, data) {
+  const expected = test.res;
+  const resultStr = JSON.stringify(data);
+  const result = JSON.parse(resultStr);
+  if (!assertQueryResult(test.q, expected, data, test.matcher)) {
+    console.log("Query:", test.q);
+    console.log("Results:", result);
+    console.log("Expected:", expected);
+    throw new Error("Query failed");
+  } else {
+    console.log("Query succeeded:", test.q);
+  }
+}
+
+export function assertQueryResult(q, expected, data, matcher) {
+  if (matcher) {
+    return matcher(data, expected);
+  }
+  if (q.toLowerCase().includes("dolt_commit")) {
+    if (data.rows.length !== 1) return false;
+    const hash = data.rows[0].dolt_commit[0];
+    if (hash.length !== 32) {
+      console.log("Invalid hash for dolt_commit:", hash);
+      return false;
+    }
+    expected.rows[0].dolt_commit = data.rows[0].dolt_commit;
+  }
+
+  // Does partial matching of actual and expected results.
+  const partialRes = {
+    command: data.command,
+    rowCount: data.rowCount,
+    oid: data.oid,
+    rows: data.rows,
+    fields: data.fields,
+  };
+  return JSON.stringify(expected) === JSON.stringify(partialRes);
+}