Add dotted counts

1 files changed, 26 insertions, 6 deletions

diff --git a/spec/default.mspec b/spec/default.mspec
index 4cdd631675..e2ad19e117 100644
--- a/spec/default.mspec
+++ b/spec/default.mspec
@@ -71,7 +71,6 @@ class MSpecScript
   prepend JobServer
 end
 
-
 require 'mspec/runner/formatters/dotted'
 
 class DottedFormatter
@@ -83,11 +82,12 @@ class DottedFormatter
       if out
         @columns = nil
       else
-        columns = ENV["COLUMNS"]
-        @columns = columns ? columns.to_i : 80
+        columns = ENV["COLUMNS"]&.to_i
+        @columns = columns&.nonzero? || 80
       end
       @dotted = 0
       @loaded = false
+      @count = 0
     end
 
     def register
@@ -97,8 +97,16 @@ class DottedFormatter
     end
 
     def after(*)
+      if @columns
+        if @dotted == 0
+          s = sprintf("%6d ", @count)
+          print(s)
+          @dotted += s.size
+        end
+        @count +=1
+      end
       super
-      if !@loaded and @columns and (@dotted += 1) >= @columns
+      if @columns and (@dotted += 1) >= @columns
         print "\n"
         @dotted = 0
       end
@@ -106,15 +114,27 @@ class DottedFormatter
 
     def load(*)
       file = MSpec.file || MSpec.files_array.first
-      print "#{file.delete_prefix(BASE)}: "
       @loaded = true
+      s = "#{file.delete_prefix(BASE)}:"
+      print s
+      if @columns
+        if (@dotted += s.size) >= @columns
+          print "\n"
+          @dotted = 0
+        else
+          print " "
+          @dotted += 1
+        end
+      end
+      @count = 0
     end
 
     def unload
       super
       if @loaded
-        print "\n"
+        print "\n" if @dotted > 0
         @dotted = 0
+        @loaded = nil
       end
     end
   }