coverage.rb in teaspoon-0.8.0

- old
+ new
@@ -1,36 +1,63 @@
 module Teaspoon
   class Coverage
-    include Teaspoon::Utility
 
-    def initialize(data, suite_name)
-      @data = data
+    def initialize(suite_name, config_name, data)
       @suite_name = suite_name
+      @data = data
+      @executable = Teaspoon::Instrumentation.executable
+      @config = coverage_configuration(config_name.to_s)
     end
 
-    def reports
-      Dir.mktmpdir do |path|
-        input = File.join(path, 'coverage.json')
-        File.open(input, 'w') { |file| file.write(@data.to_json) }
+    def generate_reports(&block)
+      input_path do |input|
         results = []
-        for format in Teaspoon.configuration.coverage_reports
+        for format in @config.reports
           result = generate_report(input, format)
           results << result if ["text", "text-summary"].include?(format.to_s)
         end
-        Teaspoon::CheckCoverage.new(input).check_coverage
-        "\n#{results.join("\n\n")}\n"
+        block.call(results.join("\n\n")) unless results.blank?
       end
     end
 
+    def check_thresholds(&block)
+      args = threshold_args
+      return if args.blank?
+      input_path do |input|
+        result = %x{#{@executable} check-coverage #{args.join(" ")} #{input.shellescape} 2>&1}
+        return if $?.exitstatus == 0
+        result = result.scan(/ERROR: .*$/).join("\n").gsub("ERROR: ", "")
+        block.call(result) unless result.blank?
+      end
+    end
+
     private
 
+    def coverage_configuration(name)
+      config = Teaspoon.configuration.coverage_configs[name]
+      raise Teaspoon::UnknownCoverage, "Unknown coverage configuration \"#{name}\"" unless config.present?
+      config[:instance] ||= Teaspoon::Configuration::Coverage.new(&config[:block])
+    end
+
+    def input_path(&block)
+      Dir.mktmpdir do |temp_path|
+        input_path = File.join(temp_path, "coverage.json")
+        File.open(input_path, "w") { |f| f.write(@data.to_json) }
+        block.call(input_path)
+      end
+    end
+
     def generate_report(input, format)
-      result = %x{#{executable} report #{format} #{input.shellescape} --dir #{File.join(Teaspoon.configuration.coverage_output_dir, @suite_name)}}
-      raise "Could not generate coverage report for #{format}" unless $?.exitstatus == 0
-      result.gsub("Done", "").gsub("Using reporter [#{format}]", "").strip
+      output_path = File.join(@config.output_path, @suite_name)
+      result = %x{#{@executable} report #{format} #{input.shellescape} --dir #{output_path} 2>&1}
+      return result.gsub("Done", "").gsub("Using reporter [#{format}]", "").strip if $?.exitstatus == 0
+      raise Teaspoon::DependencyFailure, "Could not generate coverage report for #{format}"
     end
 
-    def executable
-      @executable ||= istanbul()
+    def threshold_args
+      %w{statements functions branches lines}.map do |assert|
+        threshold = @config.send(:"#{assert}")
+        "--#{assert}=#{threshold}" if threshold
+      end.compact
     end
   end
 end