test unit

Author: bleonard

README.md

@@ -3,9 +3,9 @@
 See stuff about a Rails app.
 
 ```bash
-$ bundle exec rake:stats[/users/me/path/to/app/]
+$ bundle exec rake:stats[/path/to/app/]
 
-Directory: /users/me/path/to/app/
+Directory: /path/to/app/
 
 +----------------------+-------+-------+---------+---------+-----+-------+
 | Name                 | Lines |   LOC | Classes | Methods | M/C | LOC/M |
@@ -37,7 +37,7 @@ Directory: /users/me/path/to/app/
 * Library files
 * Gems that you've embedded in the project
 * Engines and their code
-* RSpec and Cucumber Tests
+* RSpec/Unit/Cucumber Tests
 
 ### TODO
 
@@ -46,5 +46,4 @@ Directory: /users/me/path/to/app/
 * Support JS for projects that have it in public (but not compiled)
 * Add CSS but don't count towards ratios
 * Output other metrics like number of tables and columns
-* Test unit support
 * Different output formatters

lib/rails_stats.rb

@@ -10,6 +10,7 @@ module RailsStats
 require 'rails_stats/code_statistics_calculator'
 require 'rails_stats/util'
 require 'rails_stats/app_statistics'
+require 'rails_stats/test_statistics'
 require 'rails_stats/spec_statistics'
 require 'rails_stats/cucumber_statistics'
 require 'rails_stats/root_statistics'

lib/rails_stats/code_statistics.rb

@@ -83,7 +83,10 @@ def calculate_spec_projects
       end
 
       def calculate_test_projects
-        [] # TODO: test unit
+        specs = Util.calculate_projects(@root_directory, "**", "test", "test_helper.rb")
+        specs.collect do |root_path|
+          TestStatistics.new(root_path, @key_concepts)
+        end
       end
 
       def calculate_root_projects

lib/rails_stats/test_statistics.rb

@@ -0,0 +1,74 @@
+module RailsStats
+  class TestStatistics
+    attr_reader :statistics, :total, :test
+
+    SPEC_FOLDERS = ['controllers',
+                    'functional',
+                    'helpers',
+                    'models',
+                    'requests',
+                    'unit',
+                    'integrations',
+                    'integration',
+                    'mailers',
+                    'lib']
+
+    def initialize(directory, key_concepts)
+      @test = true
+      @directory    = directory
+      @key_concepts = key_concepts
+      @statistics   = calculate_statistics
+      @total        = calculate_total
+    end
+
+    private
+
+    def calculate_total
+      out = CodeStatisticsCalculator.new(true)
+      @statistics.each do |key, stats|
+        out.add(stats)
+      end
+      out
+    end
+
+    def calculate_statistics
+      out = {}
+      categorize_files.each do |key, list|
+        out[key] = Util.calculate_file_statistics(list)
+      end 
+      out
+    end
+
+    def categorize_files
+      out = {}
+      Dir[File.join(@directory, "**", "*.rb")].each do |file_path|
+        if file_path =~ /.*_test.rb$/
+          key = categorize_file(file_path)
+        else
+          key = "Test Support"
+        end
+
+        out[key] ||= []
+        out[key]  << file_path
+      end
+
+      out
+    end
+
+    def categorize_file(file_path)
+      types = (@key_concepts + SPEC_FOLDERS).uniq
+      types.each do |folder|
+        if file_path =~ /\/#{folder}\//
+          folder = Inflector.humanize(folder)
+          folder = Inflector.titleize(folder)
+          folder = Inflector.singularize(folder)
+          return "#{folder} Tests"
+        end
+      end
+
+      # something else
+      return "Other Tests"
+    end
+  end
+
+end