Last active
August 28, 2022 15:57
-
-
Save yyx990803/f61f347b6892078c40a9e8e77b9bd984 to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
// place this file in __mocks__ | |
let pendingAssertions | |
exports.prompt = prompts => { | |
if (!pendingAssertions) { | |
throw new Error(`inquirer was mocked and used without pending assertions: ${prompts}`) | |
} | |
const answers = {} | |
let skipped = 0 | |
prompts.forEach((prompt, i) => { | |
if (prompt.when && !prompt.when(answers)) { | |
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) { | |
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) => { | |
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') | |
setValue(prompt.choices[a.choose].value) | |
} | |
if (a.check != null) { | |
expect(prompt.type).toBe('checkbox') | |
setValue(a.check.map(i => prompt.choices[i].value)) | |
} | |
if (a.confirm != null) { | |
expect(prompt.type).toBe('confirm') | |
setValue(a.confirm) | |
} | |
if (a.useDefault) { | |
expect('default' in prompt).toBe(true) | |
setValue( | |
typeof prompt.default === 'function' | |
? prompt.default(answers) | |
: prompt.default | |
) | |
} | |
}) | |
expect(prompts.length).toBe(pendingAssertions.length + skipped) | |
pendingAssertions = null | |
return Promise.resolve(answers) | |
} | |
exports.expectPrompts = assertions => { | |
pendingAssertions = assertions | |
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
jest.mock('inquirer') | |
const { prompt, expectPrompts } = require('inquirer') | |
it('should pass', async () => { | |
// specify expected prompts and corresponding actions | |
// before making the prompt | |
expectPrompts([ | |
{ | |
message: 'Hello', | |
choices: ['Foo', 'Bar'], | |
check: [0, 1] | |
}, | |
{ | |
message: 'World', | |
choices: ['Baz', 'Qux'], | |
choose: 0 | |
}, | |
{ | |
message: 'Really ok?', | |
confirm: true | |
} | |
]) | |
// then just use inquirer as normal. | |
// if a imported mdoule uses inquirer, it's also using the mocked version. | |
const answers = await prompt([ | |
{ | |
name: 'hello', | |
message: 'Hello with some extra text', | |
type: 'checkbox', | |
choices: [ | |
{ name: 'Foo', value: 'foo' }, | |
{ name: 'Bar', value: 'bar' } | |
] | |
}, | |
{ | |
name: 'world', | |
message: 'World', | |
type: 'list', | |
choices: [ | |
{ name: 'Baz', value: 'baz' }, | |
{ name: 'Qux', value: 'qux' } | |
] | |
}, | |
{ | |
name: 'ok', | |
message: 'Really ok?', | |
type: 'confirm' | |
} | |
]) | |
expect(answers).toEqual({ | |
hello: ['foo', 'bar'], | |
world: 'baz', | |
ok: true | |
}) | |
}) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
This is fantastic, thank you!