Add tests for bun:sqlite statement iterate

oven-sh · Oct 5, 2024 · e894668 · e894668
1 parent b5efed5
commit e894668
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/test/js/bun/sqlite/sqlite.test.js b/test/js/bun/sqlite/sqlite.test.js
@@ -587,6 +587,24 @@ it("db.query()", () => {
     }
   })(domjit);
 
+  // iterate (no args)
+  let i;
+  i = 0;
+  for (const row of db.query("SELECT * FROM test").iterate()) {
+    i === 0 && expect(JSON.stringify(row)).toBe(JSON.stringify({ id: 1, name: "Hello" }));
+    i === 1 && expect(JSON.stringify(row)).toBe(JSON.stringify({ id: 2, name: "World" }));
+    i++;
+  }
+  expect(i).toBe(2);
+
+  // iterate (args)
+  i = 0;
+  for (const row of db.query("SELECT * FROM test WHERE name = $name").iterate({ $name: "World" })) {
+    i === 0 && expect(JSON.stringify(row)).toBe(JSON.stringify({ id: 2, name: "World" }));
+    i++;
+  }
+  expect(i).toBe(1);
+
   db.close();
 
   // Check that a closed database doesn't crash