fix(core): Added processor tests

maticzav · Jul 11, 2018 · c898380 · c898380
1 parent ec550d9
commit c898380
Show file tree

Hide file tree

Showing 2 changed files with 72 additions and 2 deletions.
diff --git a/src/index.ts b/src/index.ts
@@ -201,7 +201,7 @@ export function normaliseArguments<T>(
  * Function used to process file uploads.
  *
  */
-function processor<T>(uploadHandler: IUploadHandler<T>) {
+export function processor<T>(uploadHandler: IUploadHandler<T>) {
   return function({
     argumentName,
     upload,
@@ -213,7 +213,7 @@ function processor<T>(uploadHandler: IUploadHandler<T>) {
         argumentName: argumentName,
         upload: res,
       }))
-    } else if (upload !== undefined) {
+    } else if (upload !== undefined && upload !== null) {
       return upload.then(uploadHandler).then(res => ({
         argumentName: argumentName,
         upload: res,

diff --git a/test.js b/test.js
@@ -8,6 +8,7 @@ import {
   isGraphQLArgumentType,
   uploadTypeIdentifier,
   normaliseArguments,
+  processor as uploadPorcessor,
 } from './'
 
 // Helpers
@@ -435,3 +436,72 @@ test('Normalizes response correctly', async t => {
     arg2: { value: 'z' },
   })
 })
+
+test('Processor handles single file correctly', async t => {
+  const uploadHandler = ({ stream, filename, mimetype, encoding }) =>
+    new Promise(resolve =>
+      setTimeout(
+        () => resolve(`${stream}${filename}${mimetype}${encoding}`),
+        10,
+      ),
+    )
+
+  const res = await uploadPorcessor(uploadHandler)({
+    argumentName: 'test',
+    upload: new Promise(resolve =>
+      resolve({ stream: 's', filename: 'f', mimetype: 'm', encoding: 'e' }),
+    ),
+  })
+
+  t.deepEqual(res, {
+    argumentName: 'test',
+    upload: 'sfme',
+  })
+})
+
+test('Processor handles multiple files correctly', async t => {
+  const uploadHandler = ({ stream, filename, mimetype, encoding }) =>
+    new Promise(resolve =>
+      setTimeout(
+        () => resolve(`${stream}${filename}${mimetype}${encoding}`),
+        10,
+      ),
+    )
+
+  const file = x =>
+    new Promise(resolve =>
+      resolve({
+        stream: `s${x}`,
+        filename: `f${x}`,
+        mimetype: `m${x}`,
+        encoding: `e${x}`,
+      }),
+    )
+
+  const res = await uploadPorcessor(uploadHandler)({
+    argumentName: 'test',
+    upload: [file(1), file(2)],
+  })
+
+  t.deepEqual(res, {
+    argumentName: 'test',
+    upload: ['s1f1m1e1', 's2f2m2e2'],
+  })
+})
+
+test('Processor handles no file correctly', async t => {
+  const uploadHandler = ({ stream, filename, mimetype, encoding }) =>
+    new Promise(resolve =>
+      setTimeout(
+        () => resolve(`${stream}${filename}${mimetype}${encoding}`),
+        10,
+      ),
+    )
+
+  const res = await uploadPorcessor(uploadHandler)({
+    argumentName: 'test',
+    upload: null,
+  })
+
+  t.is(res, null)
+})