improve inquirer mock

xdlover · Jan 3, 2018 · 91d1fc2 · 91d1fc2
1 parent 3da4e9d
commit 91d1fc2
Showing 1 changed file with 44 additions and 6 deletions.
diff --git a/__mocks__/inquirer.js b/__mocks__/inquirer.js
@@ -12,27 +12,65 @@ exports.prompt = prompts => {
       skipped++
       return
     }
+
+    const setValue = val => {
+      if (prompt.validate) {
+        const res = prompt.validate(val)
+        if (res !== true) {
+          throw new Error(`validation failed for prompt: ${prompt}`)
+        }
+      }
+      answers[prompt.name] = prompt.filter
+        ? prompt.filter(val)
+        : val
+    }
+
     const a = pendingAssertions[i - skipped]
+
     if (a.message) {
-      expect(prompt.message).toContain(a.message)
+      const message = typeof prompt.message === 'function'
+        ? prompt.message(answers)
+        : prompt.message
+      expect(message).toContain(a.message)
     }
+
     if (a.choices) {
       expect(prompt.choices.length).toBe(a.choices.length)
       a.choices.forEach((c, i) => {
-        expect(prompt.choices[i].name).toContain(a.choices[i])
+        const expected = a.choices[i]
+        if (expected) {
+          expect(prompt.choices[i].name).toContain(expected)
+        }
       })
     }
+
+    if (a.input != null) {
+      expect(prompt.type).toBe('input')
+      setValue(a.input)
+    }
+
     if (a.choose != null) {
-      expect(prompt.type).toBe('list')
-      answers[prompt.name] = prompt.choices[a.choose].value
+      expect(prompt.type === 'list' || prompt.type === 'rawList').toBe(true)
+      setValue(prompt.choices[a.choose].value)
     }
+
     if (a.check != null) {
       expect(prompt.type).toBe('checkbox')
-      answers[prompt.name] = a.check.map(i => prompt.choices[i].value)
+      setValue(a.check.map(i => prompt.choices[i].value))
     }
+
     if (a.confirm != null) {
       expect(prompt.type).toBe('confirm')
-      answers[prompt.name] = a.confirm
+      setValue(a.confirm)
+    }
+
+    if (a.useDefault) {
+      expect('default' in prompt).toBe(true)
+      setValue(
+        typeof prompt.default === 'function'
+          ? prompt.default(answers)
+          : prompt.default
+      )
     }
   })