real_data_tests 0.3.1 → 0.3.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 1c5ba3c2d2b4bb4f9ec9a6c539ddd809b6840d3eb7e6cbfb33d02f524177b3ff
-  data.tar.gz: e64f18ddd2a0bab9c0f3aad9ab628fb7bd86097644807c5d99596aa3c3af55c4
+  metadata.gz: 21f2ee14eda8fe0a048506c0ac7948c0f1c3cacb603ddb9c6490f1abef0e7fda
+  data.tar.gz: '0698d429e6feb3f2dfd358014ff71b7cca4415d2393cbab950f99776191ec43c'
 SHA512:
-  metadata.gz: 14a8d12cb5fd01582b2bf22d47d7fa518f2f0be2258e15c477f42c29c1c95b9d362fc20f486fe583f655eeac5b15ceba48c888eb49e9bc47444ab3ffcf79feb6
-  data.tar.gz: 1c2a99d52fa1b740c07152bf12d6802711bba8af8af38d94924d66ddda06d2eab76e6f459b532af3c29e2a0767bfd4ee7977aa23293b3a616225a5b340010a89
+  metadata.gz: 7839506386159938f4d5764bc0871876d7bb02ebac18ed522f23fc822e137757ea5a8a8dda7c36c4c43a91e48fa44cc0b8d69dfdca6fde402a12fe743c606fc9
+  data.tar.gz: bc74ab0ab3c19d1766714ebe35b25c310360a7b2d0154b675b223b46e0ee8610dd37ada15de1620867895fb922cba7edaf63a3d252e2d86dcaeb070bb9716b5b

data/CHANGELOG.md

@@ -1,5 +1,13 @@
 ## [Unreleased]
 
+## [0.3.2] - 2025-01-14
+### Fixed
+- Enhanced association statistics tracking in RecordCollector
+  - Added separate statistics tracking method to ensure accurate counts
+  - Stats are now tracked before circular dependency checks
+  - Fixed parent-child relationship counting in recursive associations
+  - Improved initialization of statistics structures for better reliability
+
 ## [0.3.1] - 2025-01-14
 ### Fixed
 - Fixed circular dependency handling in RecordCollector to correctly limit record collection

data/lib/real_data_tests/configuration.rb

@@ -64,7 +64,7 @@ module RealDataTests
                 :prevent_reciprocal_loading, :anonymization_rules,
                 :prevented_reciprocals
 
-    attr_accessor :max_depth
+    attr_accessor :max_depth, :max_self_ref_depth
 
     def initialize
       @association_filter_mode = nil
@@ -75,6 +75,7 @@ module RealDataTests
       @anonymization_rules = {}
       @prevented_reciprocals = Set.new
       @max_depth = 10
+      @max_self_ref_depth = 2
     end
 
     def prevent_circular_dependency(klass, association_name)

data/lib/real_data_tests/record_collector.rb

@@ -10,28 +10,9 @@ module RealDataTests
       @association_path = []
       @current_depth = 0
       @visited_associations = {}
+      @processed_self_refs = Hash.new { |h, k| h[k] = Set.new }
 
-      # Initialize stats for the record's class
-      @collection_stats[record.class.name] = {
-        count: 0,
-        associations: Hash.new(0),
-        polymorphic_types: {}
-      }
-
-      record.class.reflect_on_all_associations(:belongs_to).each do |assoc|
-        if assoc.polymorphic?
-          @collection_stats[record.class.name][:polymorphic_types][assoc.name.to_s] ||= Set.new
-        end
-      end
-
-      puts "\nInitializing RecordCollector for #{record.class.name}##{record.id}"
-      record.class.reflect_on_all_associations(:belongs_to).each do |assoc|
-        if assoc.polymorphic?
-          type = record.public_send("#{assoc.name}_type")
-          id = record.public_send("#{assoc.name}_id")
-          puts "Found polymorphic belongs_to '#{assoc.name}' with type: #{type}, id: #{id}"
-        end
-      end
+      init_collection_stats(record)
     end
 
     def collect
@@ -46,6 +27,20 @@ module RealDataTests
 
     private
 
+    def init_collection_stats(record)
+      @collection_stats[record.class.name] = {
+        count: 0,
+        associations: Hash.new(0),
+        polymorphic_types: {}
+      }
+
+      record.class.reflect_on_all_associations(:belongs_to).each do |assoc|
+        if assoc.polymorphic?
+          @collection_stats[record.class.name][:polymorphic_types][assoc.name.to_s] ||= Set.new
+        end
+      end
+    end
+
     def collect_record(record, depth)
       return if @collected_records.include?(record)
       return unless record # Guard against nil records
@@ -119,12 +114,18 @@ module RealDataTests
     def should_process_association?(record, association, depth = 0)
       return false if depth >= RealDataTests.configuration.current_preset.max_depth
 
+      # Handle self-referential associations
+      if self_referential_association?(record.class, association)
+        track_key = "#{record.class.name}:#{association.name}"
+        return false if @processed_self_refs[track_key].include?(record.id)
+        @processed_self_refs[track_key].add(record.id)
+      end
+
       association_key = "#{record.class.name}##{record.id}:#{association.name}"
       return false if @processed_associations.include?(association_key)
 
       # Check if the association is allowed by configuration
       should_process = RealDataTests.configuration.current_preset.should_process_association?(record, association.name)
-      puts "  Configuration says: #{should_process}"
 
       if should_process
         @processed_associations.add(association_key)
@@ -138,17 +139,26 @@ module RealDataTests
       @association_path.push(association.name)
 
       begin
+        related_records = fetch_related_records(record, association)
+        count = related_records.length
+
+        # Track statistics even if we're going to skip processing
+        track_association_stats(record.class.name, association.name, count)
+
+        # Check for circular dependency after getting the related records
         if detect_circular_dependency?(record, association)
           puts "    Skipping circular dependency for #{association.name} on #{record.class.name}##{record.id}"
           return
         end
 
-        related_records = fetch_related_records(record, association)
-        count = related_records.length
-        puts "    Found #{count} related #{association.name} records"
-
-        @collection_stats[record.class.name][:associations][association.name.to_s] ||= 0
-        @collection_stats[record.class.name][:associations][association.name.to_s] += count
+        # For self-referential associations, check depth
+        if self_referential_association?(record.class, association)
+          max_self_ref_depth = 2  # Default max depth for self-referential associations
+          if depth >= max_self_ref_depth
+            puts "    Reached max self-referential depth for #{association.name}"
+            return
+          end
+        end
 
         related_records.each { |related_record| collect_record(related_record, depth + 1) }
       rescue => e
@@ -158,10 +168,30 @@ module RealDataTests
       end
     end
 
+    def track_association_stats(class_name, association_name, count)
+      # Initialize stats for this class if not already done
+      @collection_stats[class_name] ||= {
+        count: 0,
+        associations: Hash.new(0),
+        polymorphic_types: {}
+      }
+
+      # Update the association count
+      @collection_stats[class_name][:associations][association_name.to_s] ||= 0
+      @collection_stats[class_name][:associations][association_name.to_s] += count
+    end
+
     def self_referential_association?(klass, association)
       return false unless association.options[:class_name]
       return false if association.polymorphic?
-      association.options[:class_name] == klass.name
+
+      target_class_name = if association.options[:class_name].is_a?(String)
+        association.options[:class_name]
+      else
+        association.options[:class_name].name
+      end
+
+      klass.name == target_class_name
     end
 
     def detect_circular_dependency?(record, association)
@@ -171,6 +201,11 @@ module RealDataTests
       target_class = association.klass
       return false unless target_class
 
+      if self_referential_association?(record.class, association)
+        track_key = "#{target_class.name}:#{association.name}"
+        return @processed_self_refs[track_key].include?(record.id)
+      end
+
       path_key = "#{target_class.name}:#{association.name}"
       visited_count = @association_path.count { |assoc| "#{target_class.name}:#{assoc}" == path_key }
 

data/lib/real_data_tests/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module RealDataTests
-  VERSION = "0.3.1"
+  VERSION = "0.3.2"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: real_data_tests
 version: !ruby/object:Gem::Version
-  version: 0.3.1
+  version: 0.3.2
 platform: ruby
 authors:
 - Kevin Dias