fix slow test properly fot dot reporter

Signed-off-by: Outsider <outsideris@gmail.com>
mochajs · Sep 28, 2018 · 114eaf1 · 114eaf1
1 parent 04469bf
commit 114eaf1
Showing 1 changed file with 25 additions and 11 deletions.
diff --git a/test/reporters/dot.spec.js b/test/reporters/dot.spec.js
@@ -13,6 +13,10 @@ describe('Dot reporter', function() {
   var useColors;
   var windowWidth;
 
+  function color(type, str) {
+    return '\u001b[' + Base.colors[type] + 'm' + str + '\u001b[0m';
+  }
+
   beforeEach(function() {
     stdout = [];
     stdoutWrite = process.stdout.write;
@@ -22,7 +26,7 @@ describe('Dot reporter', function() {
     };
     useColors = Base.useColors;
     windowWidth = Base.window.width;
-    Base.useColors = false;
+    Base.useColors = true;
     Base.window.width = 0;
   });
 
@@ -50,7 +54,7 @@ describe('Dot reporter', function() {
         runner = createMockRunner('pending', 'pending');
         Dot.call({epilogue: function() {}}, runner);
         process.stdout.write = stdoutWrite;
-        var expectedArray = ['\n  ', Base.symbols.comma];
+        var expectedArray = ['\n  ', color('pending', Base.symbols.comma)];
         expect(stdout, 'to equal', expectedArray);
       });
     });
@@ -59,7 +63,7 @@ describe('Dot reporter', function() {
         runner = createMockRunner('pending', 'pending');
         Dot.call({epilogue: function() {}}, runner);
         process.stdout.write = stdoutWrite;
-        var expectedArray = [Base.symbols.comma];
+        var expectedArray = [color('pending', Base.symbols.comma)];
         expect(stdout, 'to equal', expectedArray);
       });
     });
@@ -80,28 +84,38 @@ describe('Dot reporter', function() {
           runner = createMockRunner('pass', 'pass', null, null, test);
           Dot.call({epilogue: function() {}}, runner);
           process.stdout.write = stdoutWrite;
-          var expectedArray = ['\n  ', Base.symbols.dot];
+          var expectedArray = ['\n  ', color('fast', Base.symbols.dot)];
           expect(stdout, 'to equal', expectedArray);
         });
       });
     });
     describe('if window width is equal to or less than 1', function() {
       describe('if test speed is fast', function() {
-        it('should return a dot', function() {
+        it('should return a grey dot', function() {
           runner = createMockRunner('pass', 'pass', null, null, test);
           Dot.call({epilogue: function() {}}, runner);
           process.stdout.write = stdoutWrite;
-          var expectedArray = [Base.symbols.dot];
+          var expectedArray = [color('fast', Base.symbols.dot)];
           expect(stdout, 'to equal', expectedArray);
         });
       });
-      describe('if test speed is slow', function() {
-        it('should return a dot', function() {
+      describe('if test speed is medium', function() {
+        it('should return a yellow dot', function() {
           test.duration = 2;
           runner = createMockRunner('pass', 'pass', null, null, test);
           Dot.call({epilogue: function() {}}, runner);
           process.stdout.write = stdoutWrite;
-          var expectedArray = [Base.symbols.dot];
+          var expectedArray = [color('medium', Base.symbols.dot)];
+          expect(stdout, 'to equal', expectedArray);
+        });
+      });
+      describe('if test speed is slow', function() {
+        it('should return a bright yellow dot', function() {
+          test.duration = 3;
+          runner = createMockRunner('pass', 'pass', null, null, test);
+          Dot.call({epilogue: function() {}}, runner);
+          process.stdout.write = stdoutWrite;
+          var expectedArray = [color('bright yellow', Base.symbols.dot)];
           expect(stdout, 'to equal', expectedArray);
         });
       });
@@ -121,7 +135,7 @@ describe('Dot reporter', function() {
         runner = createMockRunner('fail', 'fail', null, null, test);
         Dot.call({epilogue: function() {}}, runner);
         process.stdout.write = stdoutWrite;
-        var expectedArray = ['\n  ', Base.symbols.bang];
+        var expectedArray = ['\n  ', color('fail', Base.symbols.bang)];
         expect(stdout, 'to equal', expectedArray);
       });
     });
@@ -130,7 +144,7 @@ describe('Dot reporter', function() {
         runner = createMockRunner('fail', 'fail', null, null, test);
         Dot.call({epilogue: function() {}}, runner);
         process.stdout.write = stdoutWrite;
-        var expectedArray = [Base.symbols.bang];
+        var expectedArray = [color('fail', Base.symbols.bang)];
         expect(stdout, 'to equal', expectedArray);
       });
     });