diff options
Diffstat (limited to 'railties/lib/code_statistics.rb')
-rw-r--r-- | railties/lib/code_statistics.rb | 9 |
1 files changed, 6 insertions, 3 deletions
diff --git a/railties/lib/code_statistics.rb b/railties/lib/code_statistics.rb index ea3c1ad66d..75334c4610 100644 --- a/railties/lib/code_statistics.rb +++ b/railties/lib/code_statistics.rb @@ -1,4 +1,7 @@ class CodeStatistics + + TEST_TYPES = ['Unit tests', 'Functional tests'] + def initialize(*pairs) @pairs = pairs @statistics = calculate_statistics @@ -55,13 +58,13 @@ class CodeStatistics def calculate_code code_loc = 0 - @statistics.each { |k, v| code_loc += v['codelines'] unless ['Units', 'Functionals'].include? k } + @statistics.each { |k, v| code_loc += v['codelines'] unless TEST_TYPES.include? k } code_loc end def calculate_tests test_loc = 0 - @statistics.each { |k, v| test_loc += v['codelines'] if ['Units', 'Functionals'].include? k } + @statistics.each { |k, v| test_loc += v['codelines'] if TEST_TYPES.include? k } test_loc end @@ -79,7 +82,7 @@ class CodeStatistics m_over_c = (statistics["methods"] / statistics["classes"]) rescue m_over_c = 0 loc_over_m = (statistics["codelines"] / statistics["methods"]) - 2 rescue loc_over_m = 0 - start = if ['Units', 'Functionals'].include? name + start = if TEST_TYPES.include? name "| #{name.ljust(18)} " else "| #{name.ljust(20)} " |