real_data_tests 0.1.0 → 0.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 40264604e537da8c1f365cd1afce57fef95988d3e51d36e1c7c68c425f0e3e8b
-  data.tar.gz: 57d486a6a25bc26522051aa2ad2a105698e28911679b1bccf6c25e1c4dd11880
+  metadata.gz: a8d2dd0beed227113467c6660eaa2101c603ef7c127ad3bd7fb19c835a641b1b
+  data.tar.gz: ac0e5fdb541110b7a523a6fa8a894d49d1c10ab9eeab86003f9ae13ebbf1968a
 SHA512:
-  metadata.gz: d7784e89db22fd3c522a9c4b60a8cca7c4c8e15bdbc86f3d20b9aa8cce4a58dbe1f1724c0355defbec3ab00f2d26e831a4351e2b5dc6fd15c97a724ed5079c95
-  data.tar.gz: 60ea36eb4824cb15554026431e0fd06d52ca1194417d025be3db9b9f93e093ccce8b50db48baecf4f5388e8458c76028ae687b6513f00bb1cb3c1acfb2b1637b
+  metadata.gz: e048bb72149153484dc179e7628ea3fc41e083906e7cbbd8c1a970b147aa1a232c5d32a6d26553de22c8c3b21a281913515bf1c51b43700c80ba0ac3dc35ffe6
+  data.tar.gz: 3859be802e7aea039691af38cc309bb56c81832bb5740d510e9f6438d10873a66d1efa8aae167a8cdc2a7e3f24b8b4c26f92d2481303d948797bbbd6a5d08d4d

data/CHANGELOG.md

@@ -1,5 +1,22 @@
 ## [Unreleased]
 
-## [0.1.0] - 2025-01-11
+## [0.2.1] - 2025-01-13
+### Fixed
+- Fixed JSONB field handling to output '{}' instead of empty string for blank values
+- Added test coverage for JSONB field handling in PgDumpGenerator
+
+## [0.2.0] - 2025-01-13
+### Added
+- New preset system for managing different test data configurations
+- Added `preset`, `use_preset`, and `with_preset` methods for configuration
+- Support for multiple named configuration presets
+- Added documentation for using presets
+- New PresetConfig class to handle preset-specific configurations
 
-- Initial release
+### Changed
+- Refactored Configuration class to use preset-based approach
+- Moved configuration methods into PresetConfig class
+- Updated documentation with preset usage examples and best practices
+
+## [0.1.0] - 2025-01-11
+- Initial release

data/Gemfile.lock

@@ -0,0 +1,220 @@
+PATH
+  remote: .
+  specs:
+    real_data_tests (0.2.0)
+      activerecord (>= 5.0)
+      faker (~> 3.0)
+      pg (>= 1.1)
+      rails (>= 5.0)
+      thor (~> 1.0)
+
+GEM
+  remote: https://rubygems.org/
+  specs:
+    actioncable (7.2.2.1)
+      actionpack (= 7.2.2.1)
+      activesupport (= 7.2.2.1)
+      nio4r (~> 2.0)
+      websocket-driver (>= 0.6.1)
+      zeitwerk (~> 2.6)
+    actionmailbox (7.2.2.1)
+      actionpack (= 7.2.2.1)
+      activejob (= 7.2.2.1)
+      activerecord (= 7.2.2.1)
+      activestorage (= 7.2.2.1)
+      activesupport (= 7.2.2.1)
+      mail (>= 2.8.0)
+    actionmailer (7.2.2.1)
+      actionpack (= 7.2.2.1)
+      actionview (= 7.2.2.1)
+      activejob (= 7.2.2.1)
+      activesupport (= 7.2.2.1)
+      mail (>= 2.8.0)
+      rails-dom-testing (~> 2.2)
+    actionpack (7.2.2.1)
+      actionview (= 7.2.2.1)
+      activesupport (= 7.2.2.1)
+      nokogiri (>= 1.8.5)
+      racc
+      rack (>= 2.2.4, < 3.2)
+      rack-session (>= 1.0.1)
+      rack-test (>= 0.6.3)
+      rails-dom-testing (~> 2.2)
+      rails-html-sanitizer (~> 1.6)
+      useragent (~> 0.16)
+    actiontext (7.2.2.1)
+      actionpack (= 7.2.2.1)
+      activerecord (= 7.2.2.1)
+      activestorage (= 7.2.2.1)
+      activesupport (= 7.2.2.1)
+      globalid (>= 0.6.0)
+      nokogiri (>= 1.8.5)
+    actionview (7.2.2.1)
+      activesupport (= 7.2.2.1)
+      builder (~> 3.1)
+      erubi (~> 1.11)
+      rails-dom-testing (~> 2.2)
+      rails-html-sanitizer (~> 1.6)
+    activejob (7.2.2.1)
+      activesupport (= 7.2.2.1)
+      globalid (>= 0.3.6)
+    activemodel (7.2.2.1)
+      activesupport (= 7.2.2.1)
+    activerecord (7.2.2.1)
+      activemodel (= 7.2.2.1)
+      activesupport (= 7.2.2.1)
+      timeout (>= 0.4.0)
+    activestorage (7.2.2.1)
+      actionpack (= 7.2.2.1)
+      activejob (= 7.2.2.1)
+      activerecord (= 7.2.2.1)
+      activesupport (= 7.2.2.1)
+      marcel (~> 1.0)
+    activesupport (7.2.2.1)
+      base64
+      benchmark (>= 0.3)
+      bigdecimal
+      concurrent-ruby (~> 1.0, >= 1.3.1)
+      connection_pool (>= 2.2.5)
+      drb
+      i18n (>= 1.6, < 2)
+      logger (>= 1.4.2)
+      minitest (>= 5.1)
+      securerandom (>= 0.3)
+      tzinfo (~> 2.0, >= 2.0.5)
+    base64 (0.2.0)
+    benchmark (0.4.0)
+    bigdecimal (3.1.9)
+    builder (3.3.0)
+    concurrent-ruby (1.3.4)
+    connection_pool (2.5.0)
+    crass (1.0.6)
+    database_cleaner (2.1.0)
+      database_cleaner-active_record (>= 2, < 3)
+    database_cleaner-active_record (2.2.0)
+      activerecord (>= 5.a)
+      database_cleaner-core (~> 2.0.0)
+    database_cleaner-core (2.0.1)
+    date (3.4.1)
+    diff-lcs (1.5.1)
+    drb (2.2.1)
+    erubi (1.13.1)
+    faker (3.5.1)
+      i18n (>= 1.8.11, < 2)
+    globalid (1.2.1)
+      activesupport (>= 6.1)
+    i18n (1.14.6)
+      concurrent-ruby (~> 1.0)
+    io-console (0.8.0)
+    irb (1.14.3)
+      rdoc (>= 4.0.0)
+      reline (>= 0.4.2)
+    logger (1.6.5)
+    loofah (2.24.0)
+      crass (~> 1.0.2)
+      nokogiri (>= 1.12.0)
+    mail (2.8.1)
+      mini_mime (>= 0.1.1)
+      net-imap
+      net-pop
+      net-smtp
+    marcel (1.0.4)
+    mini_mime (1.1.5)
+    minitest (5.25.4)
+    net-imap (0.5.5)
+      date
+      net-protocol
+    net-pop (0.1.2)
+      net-protocol
+    net-protocol (0.2.2)
+      timeout
+    net-smtp (0.5.0)
+      net-protocol
+    nio4r (2.7.4)
+    nokogiri (1.18.1-x86_64-darwin)
+      racc (~> 1.4)
+    pg (1.5.9)
+    psych (5.2.2)
+      date
+      stringio
+    racc (1.8.1)
+    rack (3.1.8)
+    rack-session (2.1.0)
+      base64 (>= 0.1.0)
+      rack (>= 3.0.0)
+    rack-test (2.2.0)
+      rack (>= 1.3)
+    rackup (2.2.1)
+      rack (>= 3)
+    rails (7.2.2.1)
+      actioncable (= 7.2.2.1)
+      actionmailbox (= 7.2.2.1)
+      actionmailer (= 7.2.2.1)
+      actionpack (= 7.2.2.1)
+      actiontext (= 7.2.2.1)
+      actionview (= 7.2.2.1)
+      activejob (= 7.2.2.1)
+      activemodel (= 7.2.2.1)
+      activerecord (= 7.2.2.1)
+      activestorage (= 7.2.2.1)
+      activesupport (= 7.2.2.1)
+      bundler (>= 1.15.0)
+      railties (= 7.2.2.1)
+    rails-dom-testing (2.2.0)
+      activesupport (>= 5.0.0)
+      minitest
+      nokogiri (>= 1.6)
+    rails-html-sanitizer (1.6.2)
+      loofah (~> 2.21)
+      nokogiri (>= 1.15.7, != 1.16.7, != 1.16.6, != 1.16.5, != 1.16.4, != 1.16.3, != 1.16.2, != 1.16.1, != 1.16.0.rc1, != 1.16.0)
+    railties (7.2.2.1)
+      actionpack (= 7.2.2.1)
+      activesupport (= 7.2.2.1)
+      irb (~> 1.13)
+      rackup (>= 1.0.0)
+      rake (>= 12.2)
+      thor (~> 1.0, >= 1.2.2)
+      zeitwerk (~> 2.6)
+    rake (13.2.1)
+    rdoc (6.10.0)
+      psych (>= 4.0.0)
+    reline (0.6.0)
+      io-console (~> 0.5)
+    rspec (3.13.0)
+      rspec-core (~> 3.13.0)
+      rspec-expectations (~> 3.13.0)
+      rspec-mocks (~> 3.13.0)
+    rspec-core (3.13.2)
+      rspec-support (~> 3.13.0)
+    rspec-expectations (3.13.3)
+      diff-lcs (>= 1.2.0, < 2.0)
+      rspec-support (~> 3.13.0)
+    rspec-mocks (3.13.2)
+      diff-lcs (>= 1.2.0, < 2.0)
+      rspec-support (~> 3.13.0)
+    rspec-support (3.13.2)
+    securerandom (0.4.1)
+    stringio (3.1.2)
+    thor (1.3.2)
+    timeout (0.4.3)
+    tzinfo (2.0.6)
+      concurrent-ruby (~> 1.0)
+    useragent (0.16.11)
+    websocket-driver (0.7.7)
+      base64
+      websocket-extensions (>= 0.1.0)
+    websocket-extensions (0.1.5)
+    zeitwerk (2.6.18)
+
+PLATFORMS
+  x86_64-darwin-21
+
+DEPENDENCIES
+  database_cleaner (~> 2.0)
+  database_cleaner-active_record
+  rake (~> 13.0)
+  real_data_tests!
+  rspec (~> 3.0)
+
+BUNDLED WITH
+   2.3.27

data/README.md

@@ -50,31 +50,115 @@ Rails.application.config.after_initialize do
     # Directory where SQL dumps will be stored
     config.dump_path = 'spec/fixtures/real_data_dumps'
 
-    # Global associations (apply to all models)
-    config.include_associations(
-      :user,
-      :organization,
-      :profile
-    )
-
-    # Model-specific associations
-    config.include_associations_for 'Patient', :visit_notes, :treatment_reports
-    config.include_associations_for 'Discipline', :organization, :user
-
-    # Control association loading behavior
-    config.limit_association 'Patient.visit_notes', 10
-    config.prevent_reciprocal 'VisitNoteType.visit_notes'
-
-    # Configure data anonymization
-    config.anonymize 'User', {
-      first_name: -> (_) { Faker::Name.first_name },
-      last_name:  -> (_) { Faker::Name.last_name },
-      email:      -> (user) { Faker::Internet.email(name: "user#{user.id}") }
+    # Define a preset for collecting patient visit data
+    config.preset :patient_visits do |p|
+      p.include_associations(
+        :visit_note_type,
+        :patient_status
+      )
+
+      p.include_associations_for 'Patient',
+        :visit_notes,
+        :treatment_reports
+
+      p.prevent_reciprocal 'VisitNoteType.visit_notes'
+
+      p.anonymize 'Patient', {
+        first_name: -> (_) { Faker::Name.first_name },
+        last_name:  -> (_) { Faker::Name.last_name }
+      }
+    end
+
+    # Define a preset for organization structure
+    config.preset :org_structure do |p|
+      p.include_associations(
+        :organization,
+        :user
+      )
+
+      p.include_associations_for 'Department',
+        :employees,
+        :managers
+
+      p.limit_association 'Department.employees', 100
+
+      p.anonymize 'User', {
+        email: -> (user) { Faker::Internet.email(name: "user#{user.id}") }
+      }
+    end
+  end
+end
+```
+
+## Using Presets
+
+Real Data Tests allows you to define multiple configuration presets for different data extraction needs. This is particularly useful when you need different association rules and anonymization settings for different testing scenarios.
+
+### Defining Presets
+
+You can define presets in your configuration:
+
+```ruby
+RealDataTests.configure do |config|
+  # Define a preset for patient data
+  config.preset :patient_data do |p|
+    p.include_associations(:patient_status, :visit_note_type)
+    p.include_associations_for 'Patient', :visit_notes
+    p.limit_association 'Patient.visit_notes', 10
+  end
+
+  # Define another preset for billing data
+  config.preset :billing_data do |p|
+    p.include_associations(:payment_method, :insurance_provider)
+    p.include_associations_for 'Invoice', :line_items, :payments
+    p.anonymize 'PaymentMethod', {
+      account_number: -> (_) { Faker::Finance.credit_card }
     }
   end
 end
 ```
 
+### Using Presets in Your Code
+
+You can use presets in several ways:
+
+```ruby
+# Create dump file using a specific preset
+RealDataTests.with_preset(:patient_data) do
+  RealDataTests.create_dump_file(patient, name: "patient_with_visits")
+end
+
+# Switch to a different preset
+RealDataTests.use_preset(:billing_data)
+RealDataTests.create_dump_file(invoice, name: "invoice_with_payments")
+
+# Use in tests
+RSpec.describe "Patient Visits" do
+  it "loads visit data correctly" do
+    RealDataTests.with_preset(:patient_data) do
+      load_real_test_data("patient_with_visits")
+      # Your test code here
+    end
+  end
+end
+```
+
+### Benefits of Using Presets
+
+- **Organized Configuration**: Keep related association rules and anonymization settings together
+- **Reusability**: Define configurations once and reuse them across different tests
+- **Clarity**: Make it clear what data is being extracted for each testing scenario
+- **Flexibility**: Easily switch between different data extraction rules
+- **Maintainability**: Update all related settings in one place
+
+### Best Practices for Presets
+
+1. **Descriptive Names**: Use clear, purpose-indicating names for your presets
+2. **Single Responsibility**: Each preset should focus on a specific testing scenario
+3. **Documentation**: Comment your presets to explain their purpose and usage
+4. **Composition**: Group related models and their associations in the same preset
+5. **Version Control**: Keep preset definitions with your test code for easy reference
+
 ## Usage
 
 ### 1. Preparing Test Data

data/lib/real_data_tests/configuration.rb

@@ -1,53 +1,75 @@
-# lib/real_data_tests/configuration.rb
 module RealDataTests
   class Configuration
-    attr_accessor :dump_path, :cleanup_models
-    attr_reader :association_filter_mode, :association_filter_list, :anonymization_rules,
-                :model_specific_associations, :association_limits, :prevent_reciprocal_loading
+    attr_accessor :dump_path, :current_preset
+    attr_reader :presets
 
     def initialize
       @dump_path = 'spec/fixtures/real_data_dumps'
-      @anonymization_rules = {}
-      @association_filter_mode = nil
-      @association_filter_list = []
-      @model_specific_associations = {}
-      @cleanup_models = []
-      @association_limits = {}
-      @prevent_reciprocal_loading = {}
+      @presets = {}
+      @current_preset = nil
+      create_preset(:default) # Always have a default preset
     end
 
-    def anonymize(model_name, mappings = {})
-      if defined?(::Rails::Engine)
-        unless ::Rails::Engine.subclasses.map(&:name).include?('RealDataTests::Engine')
-          @delayed_anonymizations << [model_name, mappings]
-          return
-        end
-      end
+    private def create_preset(name)
+      @presets[name] = PresetConfig.new
+      @current_preset = @presets[name]
+    end
 
-      begin
-        model = model_name.to_s.constantize
-        @anonymization_rules[model_name.to_s] = mappings
-      rescue => e
-        warn "Note: Anonymization for #{model_name} will be configured when Rails is fully initialized."
-      end
+    def get_association_limit(record_class, association_name)
+      current_preset&.get_association_limit(record_class, association_name)
     end
 
-    def process_delayed_anonymizations
-      return unless @delayed_anonymizations
+    def prevent_reciprocal?(record_class, association_name)
+      current_preset&.prevent_reciprocal?(record_class, association_name)
+    end
 
-      @delayed_anonymizations.each do |model_name, mappings|
-        anonymize(model_name, mappings)
-      end
-      @delayed_anonymizations = []
+    def preset(name, &block)
+      name = name.to_sym
+      @presets[name] = PresetConfig.new
+      @current_preset = @presets[name]
+      yield(@current_preset) if block_given?
+      @current_preset = @presets[:default]
     end
 
-    def exclude_associations(*associations)
-      if @association_filter_mode == :whitelist
-        raise Error, "Cannot set excluded_associations when included_associations is already set"
+    def use_preset(name)
+      name = name.to_sym
+      raise Error, "Preset '#{name}' not found" unless @presets.key?(name)
+      @current_preset = @presets[name]
+    end
+
+    def with_preset(name)
+      previous_preset = @current_preset
+      use_preset(name)
+      yield if block_given?
+    ensure
+      @current_preset = previous_preset
+    end
+
+    def method_missing(method_name, *args, &block)
+      if @current_preset.respond_to?(method_name)
+        @current_preset.public_send(method_name, *args, &block)
+      else
+        super
       end
+    end
 
-      @association_filter_mode = :blacklist
-      @association_filter_list = associations.flatten
+    def respond_to_missing?(method_name, include_private = false)
+      @current_preset.respond_to?(method_name) || super
+    end
+  end
+
+  class PresetConfig
+    attr_reader :association_filter_mode, :association_filter_list,
+                :model_specific_associations, :association_limits,
+                :prevent_reciprocal_loading, :anonymization_rules
+
+    def initialize
+      @association_filter_mode = nil
+      @association_filter_list = []
+      @model_specific_associations = {}
+      @association_limits = {}
+      @prevent_reciprocal_loading = {}
+      @anonymization_rules = {}
     end
 
     def include_associations(*associations)
@@ -58,21 +80,48 @@ module RealDataTests
       @association_filter_list = associations.flatten
     end
 
-    # New method for model-specific association rules
+    def exclude_associations(*associations)
+      if @association_filter_mode == :whitelist
+        raise Error, "Cannot set excluded_associations when included_associations is already set"
+      end
+      @association_filter_mode = :blacklist
+      @association_filter_list = associations.flatten
+    end
+
     def include_associations_for(model, *associations)
       model_name = model.is_a?(String) ? model : model.name
       @model_specific_associations[model_name] = associations.flatten
     end
 
+    def limit_association(path, limit)
+      @association_limits[path.to_s] = limit
+    end
+
+    def get_association_limit(record_class, association_name)
+      path = "#{record_class.name}.#{association_name}"
+      @association_limits[path]
+    end
+
+    def prevent_reciprocal?(record_class, association_name)
+      path = "#{record_class.name}.#{association_name}"
+      @prevent_reciprocal_loading[path]
+    end
+
+    def prevent_reciprocal(path)
+      @prevent_reciprocal_loading[path.to_s] = true
+    end
+
+    def anonymize(model_name, mappings = {})
+      @anonymization_rules[model_name.to_s] = mappings
+    end
+
     def should_process_association?(model, association_name)
       model_name = model.is_a?(Class) ? model.name : model.class.name
 
-      # Check model-specific rules first
       if @model_specific_associations.key?(model_name)
         return @model_specific_associations[model_name].include?(association_name)
       end
 
-      # Fall back to global rules
       case @association_filter_mode
       when :whitelist
         @association_filter_list.include?(association_name)
@@ -82,35 +131,5 @@ module RealDataTests
         true
       end
     end
-
-    def configure_cleanup(*models)
-      @cleanup_models = models.flatten
-    end
-
-    # Configure limits for specific associations
-    # Example: limit_association 'Patient.visit_notes', 10
-    def limit_association(path, limit)
-      @association_limits[path.to_s] = limit
-    end
-
-    # Prevent loading reciprocal associations
-    # Example: prevent_reciprocal 'VisitNoteType.visit_notes'
-    def prevent_reciprocal(path)
-      @prevent_reciprocal_loading[path.to_s] = true
-    end
-
-    # Get limit for a specific association
-    def get_association_limit(record_class, association_name)
-      path = "#{record_class.name}.#{association_name}"
-      @association_limits[path]
-    end
-
-    # Check if reciprocal loading should be prevented
-    def prevent_reciprocal?(record_class, association_name)
-      path = "#{record_class.name}.#{association_name}"
-      @prevent_reciprocal_loading[path]
-    end
   end
-
-  class Error < StandardError; end
 end

data/lib/real_data_tests/data_anonymizer.rb

@@ -3,8 +3,8 @@ require 'faker'
 
 module RealDataTests
   class DataAnonymizer
-    def initialize(configuration)
-      @configuration = configuration
+    def initialize(preset_config)
+      @preset_config = preset_config
     end
 
     def anonymize_records(records)
@@ -16,38 +16,32 @@ module RealDataTests
     def anonymize_record(record)
       return record unless should_anonymize?(record)
 
-      anonymization_rules = @configuration.anonymization_rules[record.class.name]
-
+      anonymization_rules = @preset_config.anonymization_rules[record.class.name]
       anonymization_rules.each do |attribute, anonymizer|
         begin
           new_value = case anonymizer
                      when String
-                       # Handle legacy string-based Faker calls
                        process_faker_string(anonymizer)
                      when Proc, Lambda
-                       # Handle lambda-based anonymizers
                        anonymizer.call(record)
                      else
                        raise Error, "Unsupported anonymizer type: #{anonymizer.class}"
                      end
-
           record.send("#{attribute}=", new_value)
         rescue => e
           raise Error, "Failed to anonymize #{attribute} using #{anonymizer.inspect}: #{e.message}"
         end
       end
-
       record
     end
 
     private
 
     def should_anonymize?(record)
-      @configuration.anonymization_rules.key?(record.class.name)
+      @preset_config.anonymization_rules.key?(record.class.name)
     end
 
     def process_faker_string(faker_method)
-      # Support legacy string format like "Faker::Name.first_name"
       faker_class, faker_method = faker_method.split('::')[1..].join('::').split('.')
       faker_class = Object.const_get("Faker::#{faker_class}")
       faker_class.send(faker_method)

data/lib/real_data_tests/engine.rb

@@ -7,11 +7,5 @@ module RealDataTests
     config.before_configuration do
       RealDataTests.configuration
     end
-
-    initializer 'real_data_tests.initialize' do |app|
-      if RealDataTests.configuration
-        RealDataTests.configuration.process_delayed_anonymizations
-      end
-    end
   end
 end

data/lib/real_data_tests/pg_dump_generator.rb

@@ -138,8 +138,14 @@ module RealDataTests
         value.to_s
       when :boolean
         value.to_s
-      when :array, :json, :jsonb
-        parse_and_format_special_type(value, column_info)
+      when :jsonb, :json
+        if value.blank?
+          "'{}'"  # Return empty JSON object for blank JSONB/JSON fields
+        else
+          sanitize_string(value.is_a?(String) ? value : value.to_json)
+        end
+      when :array
+        parse_and_format_array(value, column_info[:sql_type])
       else
         if column_info[:array]
           parse_and_format_array(value, column_info[:sql_type])

data/lib/real_data_tests/test_data_builder.rb

@@ -8,19 +8,17 @@ module RealDataTests
     def create_dump_file
       records = RealDataTests::RecordCollector.new(@record).collect
 
-      # Only anonymize if rules are configured
-      if RealDataTests.configuration.anonymization_rules.any?
+      # Only anonymize if rules are configured in the current preset
+      if RealDataTests.configuration.current_preset.anonymization_rules.any?
         puts "\nAnonymizing records..."
-        anonymizer = RealDataTests::DataAnonymizer.new(RealDataTests.configuration)
-        anonymizer.anonymize_records(records)
+        anonymizer = RealDataTests::DataAnonymizer.new(RealDataTests.configuration.current_preset)
+        records = anonymizer.anonymize_records(records)
       end
 
       dump_content = RealDataTests::PgDumpGenerator.new(records).generate
       dump_path = dump_file_path
-
       FileUtils.mkdir_p(File.dirname(dump_path))
       File.write(dump_path, dump_content)
-
       puts "\nDump file created at: #{dump_path}"
       dump_path
     end

data/lib/real_data_tests/version.rb

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

data/lib/real_data_tests.rb

@@ -1,13 +1,14 @@
 # frozen_string_literal: true
 
-require 'real_data_tests/configuration'
-require 'real_data_tests/data_anonymizer'
-require 'real_data_tests/engine' if defined?(Rails)
-require 'real_data_tests/pg_dump_generator'
-require 'real_data_tests/record_collector'
-require 'real_data_tests/rspec_helper'
-require 'real_data_tests/test_data_builder'
-require 'real_data_tests/version'
+require 'rails'
+require_relative 'real_data_tests/version'
+require_relative 'real_data_tests/configuration'
+require_relative 'real_data_tests/data_anonymizer'
+require_relative 'real_data_tests/engine' if defined?(Rails)
+require_relative 'real_data_tests/pg_dump_generator'
+require_relative 'real_data_tests/record_collector'
+require_relative 'real_data_tests/rspec_helper'
+require_relative 'real_data_tests/test_data_builder'
 
 module RealDataTests
   class Error < StandardError; end
@@ -28,6 +29,18 @@ module RealDataTests
       @configuration = Configuration.new
     end
 
+    def use_preset(name)
+      configuration.use_preset(name)
+    end
+
+    def with_preset(name)
+      previous_preset = configuration.current_preset
+      configuration.use_preset(name)
+      yield if block_given?
+    ensure
+      configuration.current_preset = previous_preset
+    end
+
     def create_dump_file(record, name: nil)
       raise ConfigurationError, "Configuration not initialized" unless @configuration
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: real_data_tests
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.1
 platform: ruby
 authors:
 - Kevin Dias
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2025-01-11 00:00:00.000000000 Z
+date: 2025-01-13 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rails
@@ -108,6 +108,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '2.0'
+- !ruby/object:Gem::Dependency
+  name: database_cleaner-active_record
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: faker
   requirement: !ruby/object:Gem::Requirement
@@ -134,6 +148,7 @@ files:
 - CHANGELOG.md
 - CODE_OF_CONDUCT.md
 - Gemfile
+- Gemfile.lock
 - LICENSE.txt
 - README.md
 - Rakefile
@@ -146,7 +161,6 @@ files:
 - lib/real_data_tests/rspec_helper.rb
 - lib/real_data_tests/test_data_builder.rb
 - lib/real_data_tests/version.rb
-- sig/real_data_tests.rbs
 homepage: https://github.com/diasks2/real_data_tests
 licenses:
 - MIT

data/sig/real_data_tests.rbs

@@ -1,4 +0,0 @@
-module RealDataTests
-  VERSION: String
-  # See the writing guide of rbs: https://github.com/ruby/rbs#guides
-end