Connection with callback in testing code

Author: gionatamettifogo

test/connection-tls.test.ts

@@ -21,8 +21,11 @@ import {
 describe('connection-tls', () => {
   let chinook: SQLiteCloudConnection
 
-  beforeEach(() => {
-    chinook = getChinookTlsConnection()
+  beforeEach(done => {
+    chinook = getChinookTlsConnection(error => {
+      expect(error).toBeNull()
+      done(error)
+    })
   })
 
   afterEach(() => {
@@ -54,20 +57,24 @@ describe('connection-tls', () => {
       expect(remoteConfig.tlsOptions).toBeFalsy()
     })
 
-    it('should connect with config object string', done => {
-      const configObj = getChinookConfig()
-      const conn = new SQLiteCloudTlsConnection(configObj, error => {
-        expect(error).toBeNull()
-        expect(conn.connected).toBe(true)
+    it(
+      'should connect with config object string',
+      done => {
+        const configObj = getChinookConfig()
+        const conn = new SQLiteCloudTlsConnection(configObj, error => {
+          expect(error).toBeNull()
+          expect(conn.connected).toBe(true)
 
-        chinook.sendCommands('TEST STRING', (error, results) => {
-          conn.close()
-          expect(conn.connected).toBe(false)
-          done()
+          chinook.sendCommands('TEST STRING', (error, results) => {
+            conn.close()
+            expect(conn.connected).toBe(false)
+            done()
+          })
         })
-      })
-      expect(conn).toBeDefined()
-    })
+        expect(conn).toBeDefined()
+      },
+      LONG_TIMEOUT
+    )
 
     it('should connect with connection string', done => {
       if (CHINOOK_DATABASE_URL.indexOf('localhost') > 0) {

test/connection-ws.test.ts

@@ -223,7 +223,7 @@ describe('connection-ws', () => {
       it('should test rowset', done => {
         chinook.sendCommands('TEST ROWSET', (error, results) => {
           expect(error).toBeNull()
-          expect(results.numberOfRows).toBe(41)
+          expect(results.numberOfRows).toBeGreaterThan(20)
           expect(results.numberOfColumns).toBe(2)
           expect(results.version == 1 || results.version == 2).toBeTruthy()
           expect(results.columnsNames).toEqual(['key', 'value'])