From 7849f017a38d44e0673389982464d9db84e0f42d Mon Sep 17 00:00:00 2001
From: Lee Byron <lee@leebyron.com>
Date: Thu, 14 Nov 2019 13:47:34 -0800
Subject: [PATCH] [FIX] Fix case where priming a cache with an Error results in
 UnhandledPromiseRejection

---
 src/__tests__/dataloader.test.js | 23 +++++++++++++++++++++++
 src/index.js                     | 13 +++++++++----
 2 files changed, 32 insertions(+), 4 deletions(-)

diff --git a/src/__tests__/dataloader.test.js b/src/__tests__/dataloader.test.js
index aebf5f7..2ad12d9 100644
--- a/src/__tests__/dataloader.test.js
+++ b/src/__tests__/dataloader.test.js
@@ -341,6 +341,29 @@ describe('Represents Errors', () => {
 
     identityLoader.prime(1, new Error('Error: 1'));
 
+    // Wait a bit.
+    await new Promise(setImmediate);
+
+    let caughtErrorA;
+    try {
+      await identityLoader.load(1);
+    } catch (error) {
+      caughtErrorA = error;
+    }
+    expect(caughtErrorA).toBeInstanceOf(Error);
+    expect((caughtErrorA: any).message).toBe('Error: 1');
+
+    expect(loadCalls).toEqual([]);
+  });
+
+  it('Handles priming the cache with an error', async () => {
+    const [ identityLoader, loadCalls ] = idLoader<number>();
+
+    identityLoader.prime(1, new Error('Error: 1'));
+
+    // Wait a bit.
+    await new Promise(setImmediate);
+
     let caughtErrorA;
     try {
       await identityLoader.load(1);
diff --git a/src/index.js b/src/index.js
index f109d48..3068ac4 100644
--- a/src/index.js
+++ b/src/index.js
@@ -184,10 +184,15 @@ class DataLoader<K, V, C = K> {
       if (cache.get(cacheKey) === undefined) {
         // Cache a rejected promise if the value is an Error, in order to match
         // the behavior of load(key).
-        var promise = value instanceof Error ?
-          Promise.reject(value) :
-          Promise.resolve(value);
-
+        var promise;
+        if (value instanceof Error) {
+          promise = Promise.reject(value);
+          // Since this is a case where an Error is intentionally being primed
+          // for a given key, we want to disable unhandled promise rejection.
+          promise.catch(() => {});
+        } else {
+          promise = Promise.resolve(value);
+        }
         cache.set(cacheKey, promise);
       }
     }