factory_trace 2.0.0 → 3.0.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8eb259ef9d3572a820e5fddc7a65d484f29a8d25bfdfa5947a8306cf3c22435f
-  data.tar.gz: 0b9eeaea9bf6355772994e4ca4d1dd7b62b82d43a00000c4453be067cad21376
+  metadata.gz: 9150973821d1c8196bf9dc7ba0ccbaa9c5e81cc397fed0613f2b0e0086104ec7
+  data.tar.gz: cbbf862a0a4aa1d3a436329dd4e2d2f94c2c700a1f5e0a07da5b08395f4fa89e
 SHA512:
-  metadata.gz: 21c794f7215824581bd33ad9f90235c7c929e71803ace8bb3dfd8d0fe27ab912c052633141c21d23180b5244228b4ba38ef180284c1327da41678e618887d6f5
-  data.tar.gz: 3ef9526dfba81da0a0316aa4c45caa3b7a404c87e2573a320e719d03cc5f72d0bd041356d00d81a4dadb6fdaf8cacceb0c384a3c053ad096d6bfe3004f512807
+  metadata.gz: c12b5e6ee4de364e37c99a01c363931c35f6d5f32af1a9525e9ed990c11209c38473f1a9809eb2bc980237cb069cbffb017f3c44c50bb33f5b41b90645f46037
+  data.tar.gz: 245cc1ad516ff4a706f6f32fa6c96f2bf8d55b4ec032e9ebe10ca835f02bd06b501f94a5b5b9c742c2962d56c6c3e3eddfe55fa42e3e6d95376b545e2693ed6f

data/exe/factory_trace

@@ -7,9 +7,16 @@ require "factory_trace"
 fail "You should pass at least one file with traced information.\nYou can generate it using only_trace mode." if ARGV.empty?
 
 config = FactoryTrace.configuration
-hash = FactoryTrace::Readers::TraceReader.read_from_files(*ARGV)
-reports = FactoryTrace::Processors::FindUnused.call(hash[:defined], hash[:used])
-code = (reports.any? { |report| report[:code] == :unused && !report.key?(:value) }) ? 1 : 0
+writer = FactoryTrace::Writers::ReportWriter.new(config.out, config)
+code = 0
+
+%i[factory_bot fixtures].each do |kind|
+  result = FactoryTrace::Readers::TraceReader.read_from_files(kind, *ARGV)
+  reports = FactoryTrace::Processors::FindUnused.call(result[:defined], result[:used])
+  if reports.any? { |report| report[:code] == :unused && !report.key?(:value) }
+    code = 1
+  end
+  writer.write(reports, kind: kind)
+end
 
-FactoryTrace::Writers::ReportWriter.new(config.out, config).write(reports)
 exit(code)

data/lib/factory_trace/configuration.rb

@@ -2,7 +2,7 @@
 
 module FactoryTrace
   class Configuration
-    attr_accessor :path, :enabled, :color, :mode, :trace_definition
+    attr_accessor :path, :enabled, :color, :mode, :trace_definition, :fixture_path
 
     def initialize
       @enabled = ENV.key?("FB_TRACE") || ENV.key?("FB_TRACE_FILE")
@@ -10,6 +10,7 @@ module FactoryTrace
       @color = path.nil?
       @mode = extract_mode(ENV["FB_TRACE"]) || :full
       @trace_definition = true
+      @fixture_path = "test/fixtures"
     end
 
     def trace_definition?

data/lib/factory_trace/fixture_tracker.rb

@@ -0,0 +1,25 @@
+# frozen_string_literal: true
+
+module FactoryTrace
+  class FixtureTracker
+    attr_reader :storage
+
+    def initialize
+      @storage = {}
+    end
+
+    def track!
+      tracker_storage = @storage
+
+      ::ActiveRecord::FixtureSet.prepend(Module.new do
+        define_method(:[]) do |fixture_name|
+          if fixture_name
+            tracker_storage[name] ||= Set.new
+            tracker_storage[name].add(fixture_name.to_s)
+          end
+          super(fixture_name)
+        end
+      end)
+    end
+  end
+end

data/lib/factory_trace/monkey_patches/dsl.rb

@@ -10,7 +10,7 @@ module FactoryTrace
           proxy = FactoryBot::DefinitionProxy.new(factory.definition)
           proxy.instance_eval(&block) if block
 
-          REGISTER.register_factory(factory)
+          FactoryTrace.factory_bot_register.register_factory(factory)
 
           proxy.child_factories.each do |(child_name, child_caller_location, child_options, child_block)|
             parent_factory = child_options.delete(:parent) || name
@@ -19,7 +19,7 @@ module FactoryTrace
         end
 
         def trait(name, &block)
-          REGISTER.register_trait(FactoryBot::Trait.new(name, Helpers::Caller.location, &block))
+          FactoryTrace.factory_bot_register.register_trait(FactoryBot::Trait.new(name, Helpers::Caller.location, &block))
         end
       end
     end

data/lib/factory_trace/monkey_patches/monkey_patches.rb

@@ -2,6 +2,5 @@
 
 module FactoryTrace
   module MonkeyPatches
-    REGISTER = (FactoryBot::VERSION >= "5.1.0") ? FactoryBot::Internal : FactoryBot
   end
 end

data/lib/factory_trace/preprocessors/extract_defined.rb

@@ -7,7 +7,7 @@ module FactoryTrace
       def self.call
         collection = FactoryTrace::Structures::Collection.new
 
-        MonkeyPatches::REGISTER.traits.each do |trait|
+        FactoryTrace.factory_bot_register.traits.each do |trait|
           collection.add(FactoryTrace::Helpers::Converter.trait(trait))
         end
 

data/lib/factory_trace/preprocessors/extract_defined_fixtures.rb

@@ -0,0 +1,58 @@
+# frozen_string_literal: true
+
+module FactoryTrace
+  module Preprocessors
+    class ExtractDefinedFixtures
+      # Extracts all defined fixture sets and their entries from fixture YAML files
+      #
+      # @param [String, Array<String>] fixture_path path(s) to directories containing fixture YAML files
+      #
+      # @return [FactoryTrace::Structures::Collection]
+      def self.call(fixture_path)
+        collection = FactoryTrace::Structures::Collection.new
+
+        Array(fixture_path).each do |path|
+          Dir.glob(File.join(path, "**", "*.yml")).sort.each do |file|
+            fixture_set_name = File.basename(file, ".yml")
+            entries = parse_fixture_file(file)
+
+            traits = entries.map do |entry_name, definition_path|
+              FactoryTrace::Structures::Trait.new(entry_name, definition_path: definition_path)
+            end
+
+            factory = FactoryTrace::Structures::Factory.new(
+              [fixture_set_name],
+              traits,
+              definition_path: "#{file}:1"
+            )
+
+            collection.add(factory)
+          end
+        end
+
+        collection
+      end
+
+      # Parses a fixture YAML file and returns a hash of entry names to definition paths
+      #
+      # @param [String] file_path
+      #
+      # @return [Hash<String, String>]
+      def self.parse_fixture_file(file_path)
+        entries = {}
+        lineno = 1
+
+        File.foreach(file_path) do |line|
+          if (match = line.match(/\A(\w+):(?:\s|$)/))
+            entries[match[1]] = "#{file_path}:#{lineno}"
+          end
+          lineno += 1
+        end
+
+        entries
+      end
+
+      private_class_method :parse_fixture_file
+    end
+  end
+end

data/lib/factory_trace/readers/trace_reader.rb

@@ -3,20 +3,20 @@
 module FactoryTrace
   module Readers
     class TraceReader
-      attr_reader :io, :configuration
+      attr_reader :io, :configuration, :kind
 
       # Read the data from files and merge it
       #
       # @return [Hash<Symbol, FactoryTrace::Structures::Collection>]
-      def self.read_from_files(*file_names, configuration: Configuration.new)
+      def self.read_from_files(kind, *file_names, configuration: Configuration.new)
         result = {defined: FactoryTrace::Structures::Collection.new, used: FactoryTrace::Structures::Collection.new}
 
         file_names.each do |file_name|
           File.open(file_name, "r") do |file|
-            data = new(file, configuration: configuration).read
+            data = new(kind, file, configuration: configuration).read
 
             [:defined, :used].each do |key|
-              result[key].merge!(data[key])
+              result[key].merge!(data[key]) unless data.empty?
             end
           end
         end
@@ -24,7 +24,8 @@ module FactoryTrace
         result
       end
 
-      def initialize(io, configuration: Configuration.new)
+      def initialize(kind, io, configuration: Configuration.new)
+        @kind = kind
         @io = io
         @configuration = configuration
       end
@@ -35,9 +36,14 @@ module FactoryTrace
       def read
         hash = JSON.parse(io.read)
 
+        defined = hash["defined"][kind.to_s]
+        used = hash["used"][kind.to_s]
+
+        return {} if defined.nil? || used.nil?
+
         {
-          defined: parse_collection(hash["defined"]),
-          used: parse_collection(hash["used"])
+          defined: parse_collection(defined),
+          used: parse_collection(used)
         }
       end
 

data/lib/factory_trace/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module FactoryTrace
-  VERSION = "2.0.0"
+  VERSION = "3.0.1"
 end

data/lib/factory_trace/writers/report_writer.rb

@@ -16,11 +16,14 @@ module FactoryTrace
       }.freeze
 
       # @param [Array<Hash>] results
-      def write(results)
+      # @param [Symbol] kind - :factory (default) or :fixture
+      def write(results, kind:)
+        io.puts("") unless file?
+
         total_color = (results.any? { |result| result[:code] == :unused && !result.key?(:value) }) ? :red : :green
 
         results.each do |result|
-          io.puts(convert(result, total_color: total_color))
+          io.puts(convert(result, total_color: total_color, kind: kind))
         end
       end
 
@@ -28,13 +31,23 @@ module FactoryTrace
 
       # @param [Hash<Symbol, Object>] result
       # @param [Symbol] total_color
-      def convert(result, total_color:)
+      # @param [Symbol] kind - :factory_bot or :fixtures
+      def convert(result, total_color:, kind:)
         if result[:value]
-          colorize(total_color, "total number of unique #{humanize_code(result[:code])} factories & traits: #{result[:value]}")
+          label = (kind == :fixtures) ? "fixture sets & entries" : "factories & traits"
+          colorize(total_color, "total number of unique #{humanize_code(result[:code])} #{label}: #{result[:value]}")
         elsif result[:factory_names] && result[:trait_name]
-          append_definition_path(result) { "#{humanize_code(result[:code])} trait #{colorize(:blue, result[:trait_name])} of factory #{list(result[:factory_names])}" }
+          if kind == :fixtures
+            append_definition_path(result) { "#{humanize_code(result[:code])} fixture entry #{colorize(:blue, result[:trait_name])} of fixture set #{list(result[:factory_names])}" }
+          else
+            append_definition_path(result) { "#{humanize_code(result[:code])} trait #{colorize(:blue, result[:trait_name])} of factory #{list(result[:factory_names])}" }
+          end
         elsif result[:factory_names]
-          append_definition_path(result) { "#{humanize_code(result[:code])} factory #{list(result[:factory_names])}" }
+          if kind == :fixtures
+            append_definition_path(result) { "#{humanize_code(result[:code])} fixture set #{list(result[:factory_names])}" }
+          else
+            append_definition_path(result) { "#{humanize_code(result[:code])} factory #{list(result[:factory_names])}" }
+          end
         else
           append_definition_path(result) { "#{humanize_code(result[:code])} global trait #{colorize(:blue, result[:trait_name])}" }
         end

data/lib/factory_trace/writers/trace_writer.rb

@@ -3,8 +3,8 @@
 module FactoryTrace
   module Writers
     class TraceWriter < Writer
-      # @param [FactoryTrace::Structures::Collection] defined
-      # @param [FactoryTrace::Structures::Collection] used
+      # @param [Hash] defined
+      # @param [Hash] used
       def write(defined, used)
         io.puts(JSON.pretty_generate(defined: defined.to_h, used: used.to_h))
       end

data/lib/factory_trace/writers/writer.rb

@@ -9,6 +9,10 @@ module FactoryTrace
         @io = io
         @configuration = configuration
       end
+
+      def file?
+        io.is_a?(File)
+      end
     end
   end
 end

data/lib/factory_trace.rb

@@ -2,7 +2,6 @@
 
 # External dependencies
 require "active_support"
-require "factory_bot"
 require "json"
 require "set"
 require "pathname"
@@ -13,6 +12,7 @@ require "factory_trace/helpers/converter"
 require "factory_trace/helpers/statusable"
 require "factory_trace/helpers/caller"
 require "factory_trace/tracker"
+require "factory_trace/fixture_tracker"
 
 require "factory_trace/structures/factory"
 require "factory_trace/structures/trait"
@@ -20,6 +20,7 @@ require "factory_trace/structures/collection"
 
 require "factory_trace/preprocessors/extract_defined"
 require "factory_trace/preprocessors/extract_used"
+require "factory_trace/preprocessors/extract_defined_fixtures"
 
 require "factory_trace/processors/find_unused"
 
@@ -44,19 +45,30 @@ module FactoryTrace
       return unless configuration.enabled
       trace_definitions! if configuration.trace_definition?
 
-      tracker.track!
+      tracker.track! if factory_bot?
+      fixture_tracker.track! if fixtures?
     end
 
     def stop
       return unless configuration.enabled
 
       # This is required to exclude parent traits from +defined_traits+
-      FactoryBot.reload
+      FactoryBot.reload if factory_bot?
 
       if configuration.mode?(:full)
-        Writers::ReportWriter.new(configuration.out, configuration).write(Processors::FindUnused.call(defined, used))
+        writer = Writers::ReportWriter.new(configuration.out, configuration)
+        writer.write(Processors::FindUnused.call(defined, used), kind: :factory_bot) if factory_bot?
+        writer.write(Processors::FindUnused.call(defined_fixtures, used_fixtures), kind: :fixtures) if fixtures?
       elsif configuration.mode?(:trace_only)
-        Writers::TraceWriter.new(configuration.out, configuration).write(defined, used)
+        writer = Writers::TraceWriter.new(configuration.out, configuration)
+        writer.write(mix(defined_fixtures, defined), mix(used_fixtures, used))
+      end
+    end
+
+    def mix(fixtures, factory_bot)
+      {}.tap do |result|
+        result[:fixtures] = fixtures.to_h if fixtures?
+        result[:factory_bot] = factory_bot.to_h if factory_bot?
       end
     end
 
@@ -68,6 +80,10 @@ module FactoryTrace
       @configuration ||= Configuration.new
     end
 
+    def factory_bot_register
+      @factory_bot_register ||= (FactoryBot::VERSION >= "5.1.0") ? FactoryBot::Internal : FactoryBot
+    end
+
     private
 
     def used
@@ -82,7 +98,29 @@ module FactoryTrace
       @tracker ||= Tracker.new
     end
 
+    def fixture_tracker
+      @fixture_tracker ||= FixtureTracker.new
+    end
+
+    def defined_fixtures
+      @defined_fixtures ||= Preprocessors::ExtractDefinedFixtures.call(configuration.fixture_path)
+    end
+
+    def used_fixtures
+      @used_fixtures ||= Preprocessors::ExtractUsed.call(fixture_tracker.storage)
+    end
+
+    def factory_bot?
+      defined?(FactoryBot)
+    end
+
+    def fixtures?
+      defined?(ActiveRecord::FixtureSet)
+    end
+
     def trace_definitions!
+      return unless factory_bot?
+
       FactoryBot::Factory.prepend(FactoryTrace::MonkeyPatches::Factory)
       FactoryBot::Trait.prepend(FactoryTrace::MonkeyPatches::Trait)
       FactoryBot::Syntax::Default::DSL.prepend(FactoryTrace::MonkeyPatches::Default::DSL)

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: factory_trace
 version: !ruby/object:Gem::Version
-  version: 2.0.0
+  version: 3.0.1
 platform: ruby
 authors:
 - djezzzl
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2026-03-10 00:00:00.000000000 Z
+date: 2026-03-20 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: factory_bot
@@ -17,7 +17,21 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '4.0'
-  type: :runtime
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '4.0'
+- !ruby/object:Gem::Dependency
+  name: activerecord
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '4.0'
+  type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
@@ -91,6 +105,7 @@ files:
 - exe/factory_trace
 - lib/factory_trace.rb
 - lib/factory_trace/configuration.rb
+- lib/factory_trace/fixture_tracker.rb
 - lib/factory_trace/helpers/caller.rb
 - lib/factory_trace/helpers/converter.rb
 - lib/factory_trace/helpers/statusable.rb
@@ -100,6 +115,7 @@ files:
 - lib/factory_trace/monkey_patches/monkey_patches.rb
 - lib/factory_trace/monkey_patches/trait.rb
 - lib/factory_trace/preprocessors/extract_defined.rb
+- lib/factory_trace/preprocessors/extract_defined_fixtures.rb
 - lib/factory_trace/preprocessors/extract_used.rb
 - lib/factory_trace/processors/find_unused.rb
 - lib/factory_trace/readers/trace_reader.rb