Add simple() - fixes porsager#472

porsager · porsager · commit e1a21adf99ee · 2022-08-31T07:47:23.000+02:00
diff --git a/src/query.js b/src/query.js
@@ -54,16 +54,20 @@ export class Query extends Promise {
     return this.canceller && (this.canceller(this), this.canceller = null)
   }
 
-  async readable() {
+  simple() {
     this.options.simple = true
     this.options.prepare = false
+    return this
+  }
+
+  async readable() {
+    this.simple()
     this.streaming = true
     return this
   }
 
   async writable() {
-    this.options.simple = true
-    this.options.prepare = false
+    this.simple()
     this.streaming = true
     return this
   }
diff --git a/tests/index.js b/tests/index.js
@@ -628,6 +628,20 @@ t('unsafe describe', async() => {
   ]
 })
 
+t('simple query using unsafe with multiple statements', async() => {
+  return [
+    '1,2',
+    (await sql.unsafe('select 1 as x;select 2 as x')).map(x => x[0].x).join()
+  ]
+})
+
+t('simple query using simple() with multiple statements', async() => {
+  return [
+    '1,2',
+    (await sql`select 1 as x;select 2 as x`.simple()).map(x => x[0].x).join()
+  ]
+})
+
 t('listen and notify', async() => {
   const sql = postgres(options)
   const channel = 'hello'
