Extract HEADINGS constant and #to_table method

1 files changed, 10 insertions, 5 deletions

diff --git a/lib/spandx/core/table_printer.rb b/lib/spandx/core/table_printer.rb
index 596450a..f77f4ac 100644
--- a/lib/spandx/core/table_printer.rb
+++ b/lib/spandx/core/table_printer.rb
@@ -3,6 +3,8 @@
 module Spandx
   module Core
     class TablePrinter < Printer
+      HEADINGS = ['Name', 'Version', 'Licenses', 'Location'].freeze
+
       def match?(format)
         format.to_sym == :table
       end
@@ -16,12 +18,15 @@ module Spandx
       end
 
       def print_footer(io)
-        table = Terminal::Table.new(headings: ['Name', 'Version', 'Licenses', 'Location'], output: io) do |t|
-          @dependencies.each do |d|
-            t.add_row d.to_a
-          end
+        io.puts(to_table(@dependencies.map(&:to_a)))
+      end
+
+      private
+
+      def to_table(rows)
+        Terminal::Table.new(headings: HEADINGS) do |table|
+          rows.each { |row| table.add_row(row) }
         end
-        io.puts(table)
       end
     end
   end