chore: fix testing when stdout does not support color-codes. (#62)

yargs · Apr 10, 2019 · e1ad447 · e1ad447
1 parent c9dab69
commit e1ad447
Showing 1 changed file with 31 additions and 0 deletions.
diff --git a/test/cliui.js b/test/cliui.js
@@ -430,6 +430,11 @@ describe('cliui', function () {
     })
 
     it('ignores ansi escape codes when measuring padding', function () {
+      // Forcefully enable color-codes for this test
+      const { enabled, level } = chalk
+      chalk.enabled = true
+      chalk.level = 1
+
       var ui = cliui({
         width: 25
       })
@@ -448,6 +453,32 @@ describe('cliui', function () {
         '                         '
       ]
 
+      ui.toString().split('\n').map(function (l) {
+        return stripAnsi(l)
+      }).should.eql(expected)
+      chalk.enabled = enabled
+      chalk.level = level
+    })
+
+    it('correctly handles lack of ansi escape codes when measuring padding', function () {
+      var ui = cliui({
+        width: 25
+      })
+
+      // using figlet font 'Shadow' rendering of text 'true' here
+      ui.div(
+        '  |                      \n  __|   __|  |   |   _ \\ \n  |    |     |   |   __/ \n \\__| _|    \\__,_| \\___| \n                         '
+      )
+
+      // The difference
+      var expected = [
+        '  |',
+        '  __|   __|  |   |   _ \\',
+        '  |    |     |   |   __/',
+        ' \\__| _|    \\__,_| \\___|',
+        ''
+      ]
+
       ui.toString().split('\n').map(function (l) {
         return stripAnsi(l)
       }).should.eql(expected)