lib/test_data/manager.rb in test_data-0.2.0 vs lib/test_data/manager.rb in test_data-0.2.1

@@ -12,10 +12,11 @@
       ensure_after_load_save_point_is_active_if_data_is_loaded!
       return rollback_to_after_data_load if save_point_active?(:after_data_load)
 
       create_save_point(:before_data_load)
       @inserts_test_data.call
+      @config.after_test_data_load_hook.call
       record_ar_internal_metadata_that_test_data_is_loaded
       create_save_point(:after_data_load)
     end
 
     def truncate
@@ -39,9 +40,10 @@
         TestData.log.debug("TestData.uses_clean_slate was called, but data was not loaded. Loading data before truncate to preserve the transaction save point ordering")
         load
       end
 
       @truncates_test_data.call
+      @config.after_test_data_truncate_hook.call
       record_ar_internal_metadata_that_test_data_is_truncated
       create_save_point(:after_data_truncate)
     end
 
     def load_custom_data(loader, **options)