50x template calls #19

gionatamettifogo · gionatamettifogo · commit 4bd36825d27f · 2023-11-13T08:43:27.000+01:00
diff --git a/test/database.test.ts b/test/database.test.ts
@@ -4,12 +4,11 @@
 
 import { SQLiteCloudError, SQLiteCloudConnection } from '../src/index'
 import { Database, ErrorCallback } from '../src/database'
-import { CHINOOK_DATABASE_URL, TESTING_DATABASE_URL } from './protocol.test'
+import { CHINOOK_DATABASE_URL, TESTING_DATABASE_URL, LONG_TIMEOUT } from './protocol.test'
+
 import * as dotenv from 'dotenv'
 dotenv.config()
 
-const LONG_TIMEOUT = 30 * 1000
-
 /* eslint-disable @typescript-eslint/no-unsafe-member-access */
 
 import { join } from 'path'
@@ -266,12 +265,27 @@ describe('Database', () => {
       const results = await database.sql`SELECT * FROM people WHERE name = 'Eva' OR name = ${name} OR age < 30`
       expect(results).toHaveLength(11)
     })
+  })
 
-    it('template string with multiple queries', async () => {
-      for (let i = 0; i < 2; i++) {
-        const results = await database.sql`SELECT * FROM people ORDER BY RANDOM() LIMIT 12`
-        expect(results).toHaveLength(12)
-      }
-    })
+  describe('stress testing', () => {
+    it(
+      '50x sql async with random selects',
+      async () => {
+        const numQueries = 50
+        const startTime = Date.now()
+
+        const table = 'people'
+        for (let i = 0; i < numQueries; i++) {
+          const results = await database.sql`SELECT * FROM ${table} ORDER BY RANDOM() LIMIT 12`
+          expect(results).toHaveLength(12)
+          expect(Object.keys(results[0])).toEqual(['id', 'name', 'age', 'hobby'])
+        }
+
+        const queryMs = (Date.now() - startTime) / numQueries
+        console.log(`${numQueries}x template selects, ${queryMs.toFixed(0)}ms per query`)
+        expect(queryMs).toBeLessThan(500)
+      },
+      LONG_TIMEOUT
+    )
   })
 })
diff --git a/test/protocol.test.ts b/test/protocol.test.ts
@@ -14,7 +14,7 @@ export const TESTING_DATABASE_URL = process.env.TESTING_DATABASE_URL as string
 console.assert(CHINOOK_DATABASE_URL, 'CHINOOK_DATABASE_URL is not defined')
 console.assert(TESTING_DATABASE_URL, 'TESTING_DATABASE_URL is not defined')
 
-const LONG_TIMEOUT = 100 * 1000 // 100 seconds
+export const LONG_TIMEOUT = 100 * 1000 // 100 seconds
 
 export function getChinoookConfig(): SQLiteCloudConfig {
   return parseConnectionString(CHINOOK_DATABASE_URL)
@@ -270,9 +270,9 @@ describe('protocol', () => {
 
   describe('connection stress testing', () => {
     it(
-      'run individual selects',
+      '50x individual selects',
       async () => {
-        const numQueries = 150
+        const numQueries = 50
         const startTime = Date.now()
         for (let i = 0; i < numQueries; i++) {
           let rowset = await connection.sendCommands('SELECT * FROM albums ORDER BY RANDOM() LIMIT 4;')
@@ -287,9 +287,9 @@ describe('protocol', () => {
     )
 
     it(
-      'run batched selects',
+      '50x batched selects',
       async () => {
-        const numQueries = 150
+        const numQueries = 50
         const startTime = Date.now()
         for (let i = 0; i < numQueries; i++) {
           let rowset = await connection.sendCommands(
