Add tests for 'table_output', 'block_output' methods

test/test_simplecov-console.rb

@@ -2,7 +2,17 @@
 
 class TestSimplecovConsole < MiniTest::Test
 
-  Source = Struct.new(:line_number)
+  # mock for SimpleCov::SourceFile::Line
+  Line = Struct.new(:line_number)
+
+  # mock for SimpleCov::SourceFile
+  SourceFile = Struct.new(
+    :filename,
+    :lines_of_code,
+    :covered_lines,
+    :missed_lines,
+    :covered_percent
+  )
 
   def setup
     @console = SimpleCov::Formatter::Console.new
@@ -14,9 +24,29 @@ def test_defined
   end
 
   def test_missed
-    missed_lines = [Source.new(1), Source.new(2),
-                    Source.new(3), Source.new(5)]
+    missed_lines = [Line.new(1), Line.new(2),
+                    Line.new(3), Line.new(5)]
     expected_result = ["1-3", "5"]
     assert_equal @console.missed(missed_lines), expected_result
   end
+
+  def test_table_output
+    SimpleCov::Formatter::Console.output_style = 'table'
+    files = [
+      SourceFile.new('foo.rb',5,[2,3],[Line.new(1), Line.new(4), Line.new(5)],40.0)
+    ]
+    actual = @console.table_output(files,'/')
+    assert actual.is_a? Terminal::Table
+    assert_equal 1, actual.rows.count 
+  end
+
+  def test_block_output
+    SimpleCov::Formatter::Console.use_colors = false
+    SimpleCov::Formatter::Console.output_style = 'block'
+    files = [
+      SourceFile.new('foo.rb',5,[2,3],[Line.new(1), Line.new(4), Line.new(5)],40.0)
+    ]
+    expected = "\n    file: foo.rb\ncoverage: 40.00% (2/5 lines)\n  missed: 1, 4-5"
+    assert_equal expected, @console.block_output(files,'/')
+  end
 end