Report

Adding CSV support

Signed-off-by: Rodrigo Nardi <rnardi@netdef.org>

Author: RodrigoMNardi

reports/re_run_report.rb

@@ -9,6 +9,7 @@
 #  frozen_string_literal: true
 
 require 'json'
+require 'csv'
 require_relative '../database_loader'
 
 module Reports
@@ -57,6 +58,8 @@ def save_rerun_info(result, output, filename)
       case output
       when 'json'
         File.write(filename, result.to_json)
+      when 'csv'
+        create_csv(filename)
       when 'file'
         File.open(filename, 'a') do |f|
           raw_output(result, file_descriptor: f)
@@ -66,6 +69,19 @@ def save_rerun_info(result, output, filename)
       end
     end
 
+    def create_csv(filename)
+      CSV.open(filename, 'wb') do |csv_input|
+        csv_input << %w[PullRequest CheckSuiteId BambooJob GithubUsername RequestedAt Type TestsOrBuilds]
+        @result.each_pair do |pull_requst, info|
+          info[:check_suites].each do |cs|
+            csv_input << [pull_requst,
+                          cs[:check_suite_id], cs[:bamboo_job], cs[:requested_at],
+                          cs[:github_username], cs[:type], cs[:tests_or_builds].join(',')]
+          end
+        end
+      end
+    end
+
     def raw_output(result, file_descriptor: nil)
       result.each_pair do |pull_request, info|
         print("\nPull Request: ##{pull_request} - Reruns: #{info[:total]}", file_descriptor)

reports/re_run_report_by_user.rb

@@ -59,6 +59,8 @@ def save_rerun_info(result, output, filename)
       case output
       when 'json'
         File.write(filename, result.to_json)
+      when 'csv'
+        create_csv(filename)
       when 'file'
         File.open(filename, 'a') do |f|
           raw_output(result, file_descriptor: f)
@@ -68,6 +70,19 @@ def save_rerun_info(result, output, filename)
       end
     end
 
+    def create_csv(filename)
+      CSV.open(filename, 'wb') do |csv_input|
+        csv_input << %w[PullRequest CheckSuiteId BambooJob GithubUsername RequestedAt Type TestsOrBuilds]
+        @result.each_pair do |pull_requst, info|
+          info[:check_suites].each do |cs|
+            csv_input << [pull_requst,
+                          cs[:check_suite_id], cs[:bamboo_job], cs[:requested_at],
+                          cs[:github_username], cs[:type], cs[:tests_or_builds].join(',')]
+          end
+        end
+      end
+    end
+
     def raw_output(result, file_descriptor: nil)
       result.each_pair do |pull_request, info|
         print("\nPull Request: ##{pull_request} - Reruns: #{info[:total]}", file_descriptor)