chore: fix tests

prisma · Jul 28, 2020 · 3c3d777 · 3c3d777
1 parent 512ceaa
commit 3c3d777
Showing 1 changed file with 12 additions and 4 deletions.
diff --git a/src/packages/engine-core/src/NodeEngine.ts b/src/packages/engine-core/src/NodeEngine.ts
@@ -858,14 +858,15 @@ ${this.lastErrorLog.fields.file}:${this.lastErrorLog.fields.line}:${this.lastErr
       })
 
       .catch(async (e) => {
-        const isError = await this.handleRequestError(e)
+        const isError = await this.handleRequestError(e, numTry < 3)
         if (!isError) {
           // retry
           if (numTry < 3) {
             await new Promise((r) => setTimeout(r, Math.random() * 1000))
             return this.request(query, headers, numTry + 1)
           }
         }
+        throw isError
       })
   }
 
@@ -911,18 +912,23 @@ ${this.lastErrorLog.fields.file}:${this.lastErrorLog.fields.line}:${this.lastErr
         }
       })
       .catch(async (e) => {
-        const isError = await this.handleRequestError(e)
+        const isError = await this.handleRequestError(e, numTry < 3)
         if (!isError) {
           // retry
           if (numTry < 3) {
             await new Promise((r) => setTimeout(r, Math.random() * 1000))
             return this.requestBatch(queries, transaction, numTry + 1)
           }
         }
+
+        throw isError
       })
   }
 
-  private handleRequestError = async (error: Error & { code?: string }) => {
+  private handleRequestError = async (
+    error: Error & { code?: string },
+    graceful: boolean,
+  ) => {
     debug({ error })
     let err
     if (this.currentRequestPromise.isCanceled && this.lastError) {
@@ -1045,7 +1051,9 @@ Please look into the logs or turn on the env var DEBUG=* to debug the constantly
           }),
         )
         debug(err.message)
-        return false
+        if (graceful) {
+          return false
+        }
       }
     }