test-prof-autopilot 0.0.5 → 0.0.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: da180b01be948feec684ca0b80005809678834cec159dc8af26b8dcc8549d31f
-  data.tar.gz: e42323a7e60dbd7198cf4772153f6c42e3544a4b60fc55f66c1fb335055e0aad
+  metadata.gz: 9053434f1ede969d5dc60e8f0ed6bada6ba628220fa599c7700ab9fa4bfeec85
+  data.tar.gz: ba2428c6baff4d0b0c437f01b3e3f4522f1d9a3f528689bb9de317e54e94e661
 SHA512:
-  metadata.gz: 18ec8fffa21d68373ff74f32da3ff1f08c9146a0e74c22e9fd420dc469f61fec1ef7808f06f614caea0ca8c11967cf159163724248a9411fb2ace7f081671e2f
-  data.tar.gz: 5db488c8dd13b78fc623eddf05ca69465f212e1bb8f6492dd62370f9415e061cc455448d7067f9e05bfce8ef8aac048e471487a61c32c983927dd6ba0fcaaf63
+  metadata.gz: d95aeca28faec87d052dbb5ce90dd953753acc8d4026be7b8b9b270e495008405de10672cfc6dee680d3839f90db6c7564faf7655614cee6346b63c0384707c9
+  data.tar.gz: ade585ced221e23276f229db9742896e9161ecda7455a66d24f4a81f7184cbb25e4e7d061393062ad3579d9ddf0409861e54939a41ba9dc2b0cd8bbad0657b00

data/lib/test_prof/autopilot/factory_prof/printer.rb

@@ -10,31 +10,10 @@ module TestProf
         class PrinterError < StandardError; end
 
         def print_report(report)
-          result = report.raw_report
+          result = report.result
 
-          raise PrinterError, result["error"] if result["error"]
-
-          msgs = []
-
-          msgs <<
-            <<~MSG
-              Factories usage
-
-               Total: #{result["total_count"]}
-               Total top-level: #{result["total_top_level_count"]}
-               Total time: #{format("%.4f", result["total_time"])}s
-               Total uniq factories: #{result["total_uniq_factories"]}
-
-                 total   top-level     total time      time per call      top-level time               name
-            MSG
-
-          result["stats"].each do |stat|
-            time_per_call = stat["total_time"] / stat["total_count"]
-
-            msgs << format("%8d %11d %13.4fs %17.4fs %18.4fs %18s", stat["total_count"], stat["top_level_count"], stat["total_time"], time_per_call, stat["top_level_time"], stat["name"])
-          end
-
-          Logging.log msgs.join("\n")
+          # TODO: Move total_time to the report
+          TestProf::FactoryProf::Printers::Simple.dump(result, start_time: TestProf.now)
         end
 
         module_function :print_report

data/lib/test_prof/autopilot/factory_prof/report.rb

@@ -21,6 +21,36 @@ module TestProf
           @type = :factory_prof
           @raw_report = raw_report
         end
+
+        def result
+          @result ||= TestProf::FactoryProf::Result.new(
+            raw_report["stacks"],
+            raw_report["raw_stats"].transform_values! { |stats| stats.transform_keys!(&:to_sym) }
+          )
+        end
+
+        def merge(other)
+          report = raw_report.dup
+
+          report["stacks"] += other.raw_report["stacks"] if report["stacks"]
+
+          (report["raw_stats"].values + other.raw_report["raw_stats"].values).each_with_object({}) do |data, acc|
+            if acc.key?(data["name"])
+              current = acc[data["name"]]
+
+              current["total_count"] += data["total_count"]
+              current["top_level_count"] += data["top_level_count"]
+              current["total_time"] += data["total_time"]
+              current["top_level_time"] += data["top_level_time"]
+            else
+              acc[data["name"]] = data.dup
+            end
+          end.then do |stats|
+            report["raw_stats"] = stats
+          end
+
+          Report.new(report)
+        end
       end
     end
   end

data/lib/test_prof/autopilot/factory_prof/writer.rb

@@ -3,22 +3,36 @@
 module TestProf
   module Autopilot
     module FactoryProf
-      # Module is used for writing :factory_prof report in different formats
-      module Writer
+      # Class is used for writing :factory_prof report in different formats
+      class Writer < ReportWriter
         Registry.register(:factory_prof_writer, self)
 
         ARTIFACT_FILE = "factory_prof_report"
 
-        def write_report(report, file_name: ARTIFACT_FILE)
-          dir_path = FileUtils.mkdir_p(Autopilot.config.artifacts_dir)[0]
-          file_path = File.join(dir_path, file_name + ".json")
+        def generate_json
+          report.result.to_json
+        end
 
-          File.write(file_path, JSON.generate(report.raw_report))
+        def generate_html
+          result = report.result
 
-          Logging.log "FactoryProf report saved: #{file_path}"
-        end
+          report_data = {
+            total_stacks: result.stacks.size,
+            total: result.total_count
+          }
+
+          report_data[:roots] = TestProf::FactoryProf::Printers::Flamegraph.convert_stacks(result)
 
-        module_function :write_report
+          path = TestProf::Utils::HTMLBuilder.generate(
+            data: report_data,
+            template: TestProf::FactoryProf::Printers::Flamegraph::TEMPLATE,
+            output: TestProf::FactoryProf::Printers::Flamegraph::OUTPUT_NAME
+          )
+
+          File.read(path).tap do
+            FileUtils.rm(path)
+          end
+        end
       end
     end
   end

data/lib/test_prof/autopilot/patches/factory_prof_patch.rb

@@ -1,6 +1,17 @@
 # frozen_string_literal: true
 
 module TestProf
+  using(Module.new do
+    refine FactoryProf::Result do
+      def to_json
+        {
+          stacks: stacks,
+          raw_stats: raw_stats
+        }.to_json
+      end
+    end
+  end)
+
   module Autopilot
     module Patches
       # Monkey-patch for 'TestProf::FactoryProf::Printers::Simple'.
@@ -9,52 +20,18 @@ module TestProf
       module FactoryProfPatch
         ARTIFACT_FILE = "factory_prof_report.json"
 
-        def patch
-          TestProf::FactoryProf::Printers::Simple.module_eval do
-            def self.dump(result, **)
-              profiler_hash =
-                if result.raw_stats == {}
-                  {
-                    error: "No factories detected"
-                  }
-                else
-                  {
-                    total_count: result.stats.sum { |stat| stat[:total_count] },
-                    total_top_level_count: result.stats.sum { |stat| stat[:top_level_count] },
-                    total_time: result.stats.sum { |stat| stat[:top_level_time] },
-                    total_uniq_factories: result.stats.uniq { |stat| stat[:name] }.count,
-                    stats: result.stats
-                  }
-                end
-
-              dir_path = FileUtils.mkdir_p(Autopilot.config.tmp_dir)[0]
-              file_path = File.join(dir_path, ARTIFACT_FILE)
+        module PrinterExt
+          def dump(result, **)
+            dir_path = FileUtils.mkdir_p(Autopilot.config.tmp_dir)[0]
+            file_path = File.join(dir_path, ARTIFACT_FILE)
 
-              File.write(file_path, JSON.generate(profiler_hash))
-            end
+            File.write(file_path, result.to_json)
           end
+        end
 
-          TestProf::FactoryProf::Printers::Flamegraph.module_eval do
-            def self.dump(result, **)
-              profiler_hash =
-                if result.raw_stats == {}
-                  {
-                    error: "No factories detected"
-                  }
-                else
-                  {
-                    total_stacks: result.stacks.size,
-                    total: result.total_count,
-                    roots: convert_stacks(result)
-                  }
-                end
-
-              dir_path = FileUtils.mkdir_p(Autopilot.config.tmp_dir)[0]
-              file_path = File.join(dir_path, ARTIFACT_FILE)
-
-              File.write(file_path, JSON.generate(profiler_hash))
-            end
-          end
+        def patch
+          TestProf::FactoryProf::Printers::Simple.singleton_class.prepend(PrinterExt)
+          TestProf::FactoryProf::Printers::Flamegraph.singleton_class.prepend(PrinterExt)
         end
 
         module_function :patch

data/lib/test_prof/autopilot/version.rb

@@ -2,6 +2,6 @@
 
 module TestProf
   module Autopilot # :nodoc:
-    VERSION = "0.0.5"
+    VERSION = "0.0.6"
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: test-prof-autopilot
 version: !ruby/object:Gem::Version
-  version: 0.0.5
+  version: 0.0.6
 platform: ruby
 authors:
 - Ruslan Shakirov