test: add test case for exit code

jamesgeorge007 · Dec 13, 2020 · 6022f6a · 6022f6a
1 parent 902e8b9
commit 6022f6a
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 0 deletions.
diff --git a/test/fixtures/cliExitCode.js b/test/fixtures/cliExitCode.js
@@ -0,0 +1,15 @@
+const enquirer = require("enquirer");
+
+const choices = ["0", "1", "2", "3"];
+
+enquirer
+  .prompt({
+    type: "select",
+    name: "exitCode",
+    message: "Pick the exit code",
+    choices,
+  })
+  .then(({ exitCode }) => {
+    console.log(`You chose ${exitCode} exit code`);
+    process.exit(exitCode);
+  });
diff --git a/test/test.js b/test/test.js
@@ -7,6 +7,7 @@ const ENTER = "\x0D";
 const SPACE = "\x20";
 
 const cliPath = `${__dirname}/fixtures/cli.js`;
+const cliExitCodePath = `${__dirname}/fixtures/cliExitCode.js`;
 
 describe("cli-prompts-test", () => {
   it("picks a single option", async () => {
@@ -26,4 +27,17 @@ describe("cli-prompts-test", () => {
     expect(exitCode).toBe(0);
     expect(stdout).toContain("You chose aqua, fuchsia, green");
   });
+
+  it("pick 0 exit code options", async () => {
+    const { exitCode } = await runTest([cliExitCodePath], [ENTER]);
+    expect(exitCode).toBe(0);
+  });
+
+  it("pick 2 exit code options", async () => {
+    const { exitCode } = await runTest(
+      [cliExitCodePath],
+      [`${DOWN}${DOWN}${ENTER}`]
+    );
+    expect(exitCode).toBe(2);
+  });
 });