fix: allow resetting stdmock

oclif · Jan 27, 2018 · ceabe22 · ceabe22
1 parent 84b4c18
commit ceabe22
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/src/stdmock.ts b/src/stdmock.ts
@@ -5,6 +5,7 @@ const create = <T extends 'stdout' | 'stderr'>(std: T) => (opts: {print?: boolea
     mock[std].start()
     mock[std].print = opts.print === true
     mock[std].stripColor = opts.stripColor !== false
+    if (ctx[std] as any !== undefined) return
     Object.defineProperty(ctx, std, {
       get: () => mock[std].output
     })

diff --git a/test/stdmock.test.ts b/test/stdmock.test.ts
@@ -85,4 +85,13 @@ describe('stdout', () => {
     console.log(chalk.red('foobar'))
     expect(output.stdout).to.contain(chalk.red('foobar'))
   })
+
+  fancy
+  .stdout()
+  .do(() => console.log('foo'))
+  .do(c => expect(c.stdout).to.equal('foo\n'))
+  .stdout()
+  .do(() => console.log('bar'))
+  .do(c => expect(c.stdout).to.equal('bar\n'))
+  .it('resets stdout')
 })