Changes to support tddium parallelization.

noahd1 · noahd1 · commit b6e0039d089b · 2013-09-03T16:32:30.000-04:00
diff --git a/lib/codeclimate-test-reporter.rb b/lib/codeclimate-test-reporter.rb
@@ -2,6 +2,8 @@
 require "digest/sha1"
 require "net/https"
 require "uri"
+require "tmpdir"
+require "securerandom"
 
 module CodeClimate
   class TestReporter
@@ -43,8 +45,31 @@ def self.post_results(result)
 
     class Formatter
       def format(result)
-        totals = Hash.new(0)
+        print "Coverage = #{result.covered_percent.round(2)}%."
+
+        payload = to_payload(result)
+        if tddium? || ENV["TO_FILE"]
+          file_path = File.join(Dir.tmpdir, "coverage-worker-#{SecureRandom.uuid}.json")
+          print "Coverage results saved to #{file_path}..."
+          File.open(file_path, "w") { |file| file.write(payload.to_json) }
+        else
+          print "Sending report to #{API.host}... "
+          API.post_results(payload)
+        end
+
+        puts "done."
+        true
+      rescue => ex
+        puts "\nCode Climate encountered an exception: #{ex.class}"
+        puts ex.message
+        ex.backtrace.each do |line|
+          puts line
+        end
+        false
+      end
 
+      def to_payload(result)
+        totals = Hash.new(0)
         source_files = result.files.map do |file|
           totals[:total]      += file.lines.count
           totals[:covered]    += file.covered_lines.count
@@ -64,9 +89,7 @@ def format(result)
           }
         end
 
-        print "Coverage = #{result.covered_percent.round(2)}%. Sending report to #{API.host}... "
-
-        API.post_results({
+        {
           repo_token:       ENV["CODECLIMATE_REPO_TOKEN"],
           source_files:     source_files,
           run_at:           result.created_at,
@@ -86,17 +109,7 @@ def format(result)
             gem_version:    VERSION
           },
           ci_service: ci_service_data
-        })
-
-        puts "done."
-        true
-      rescue => ex
-        puts "\nCode Climate encountered an exception: #{ex.class}"
-        puts ex.message
-        ex.backtrace.each do |line|
-          puts line
-        end
-        false
+        }
       end
 
       def ci_service_data
@@ -161,6 +174,10 @@ def git_branch
         branch = [branch].flatten.first
         branch ? branch.gsub("* ","") : nil
       end
+
+      def tddium?
+        ci_service_data && ci_service_data[:name] == "tddium"
+      end
     end
 
     def self.start
