Fixes bad align in colored level.

Signed-off-by: Eric Wang <[email protected]>

Author: skygragon

Diff for: lib/helper.js

@@ -89,11 +89,11 @@ h.prettyTime = function(n) {
 
 h.prettyLevel = function(level) {
   var chalk = require('./chalk');
-  switch (level) {
-    case 'Easy': return chalk.green('Easy');
-    case 'Medium': return chalk.yellow('Medium');
-    case 'Hard': return chalk.red('Hard');
-    default: return ' ';
+  switch (level.toLowerCase()) {
+    case 'easy':   return chalk.green(level + '  ');
+    case 'medium': return chalk.yellow(level);
+    case 'hard':   return chalk.red(level + '  ');
+    default:       return level;
   }
 };
 

Diff for: test/test_helper.js

@@ -46,6 +46,20 @@ describe('helper', function() {
     });
   }); // #prettyText
 
+  describe('#prettyLevel', function() {
+    it('should ok w/ color', function() {
+      chalk.enabled = true;
+
+      assert.equal(h.prettyLevel('Easy'), chalk.green('Easy  '));
+      assert.equal(h.prettyLevel('Medium'), chalk.yellow('Medium'));
+      assert.equal(h.prettyLevel('Hard'), chalk.red('Hard  '));
+      assert.equal(h.prettyLevel('easy'), chalk.green('easy  '));
+      assert.equal(h.prettyLevel('medium'), chalk.yellow('medium'));
+      assert.equal(h.prettyLevel('hard'), chalk.red('hard  '));
+      assert.equal(h.prettyLevel('unknown'), 'unknown');
+    });
+  }); // #prettyLevel
+
   describe('#prettySize', function() {
     it('should ok', function() {
       assert.equal(h.prettySize(0), '0.00B');