Preserve I18n.locale set when after_commit is called (#36)

lib/after_commit_everywhere/wrap.rb

@@ -7,6 +7,7 @@ class Wrap
     def initialize(connection: ActiveRecord::Base.connection, **handlers)
       @connection = connection
       @handlers = handlers
+      @locale = I18n.locale
     end
 
     # rubocop: disable Naming/PredicateName
@@ -16,21 +17,21 @@ def has_transactional_callbacks?
     # rubocop: enable Naming/PredicateName
 
     def before_committed!(*)
-      @handlers[:before_commit]&.call
+      I18n.with_locale(@locale) { @handlers[:before_commit]&.call }
     end
 
     def trigger_transactional_callbacks?
       true
     end
 
     def committed!(should_run_callbacks: true)
-      if should_run_callbacks
-        @handlers[:after_commit]&.call
-      end
+      return unless should_run_callbacks
+
+      I18n.with_locale(@locale) { @handlers[:after_commit]&.call }
     end
 
     def rolledback!(*)
-      @handlers[:after_rollback]&.call
+      I18n.with_locale(@locale) { @handlers[:after_rollback]&.call }
     end
 
     # Required for +transaction(requires_new: true)+

spec/after_commit_everywhere_spec.rb

@@ -108,6 +108,19 @@
         expect(handler_1).not_to have_received(:call)
         expect(handler_2).not_to have_received(:call)
       end
+
+      it "preserves the locale" do
+        I18n.enforce_available_locales = false
+
+        ActiveRecord::Base.transaction do
+          expect(I18n.locale).not_to eq(:de)
+          I18n.with_locale(:de) { example_class.new.after_commit { handler.call(I18n.locale) } }
+          expect(handler).not_to have_received(:call)
+        end
+        expect(handler).to have_received(:call).with(:de)
+
+        I18n.enforce_available_locales = true
+      end
     end
 
     context "without transaction" do
@@ -291,6 +304,19 @@
         end
         expect(handler).not_to have_received(:call)
       end
+
+      it "preserves the locale" do
+        I18n.enforce_available_locales = false
+
+        ActiveRecord::Base.transaction do
+          expect(I18n.locale).not_to eq(:de)
+          I18n.with_locale(:de) { example_class.new.before_commit { handler.call(I18n.locale) } }
+          expect(handler).not_to have_received(:call)
+        end
+        expect(handler).to have_received(:call).with(:de)
+
+        I18n.enforce_available_locales = true
+      end
     end
 
     context "without transaction" do
@@ -439,6 +465,20 @@
         end
         expect(handler).not_to have_received(:call)
       end
+
+      it "preserves the locale" do
+        I18n.enforce_available_locales = false
+
+        ActiveRecord::Base.transaction do
+          expect(I18n.locale).not_to eq(:de)
+          I18n.with_locale(:de) { example_class.new.after_rollback { handler.call(I18n.locale) } }
+          expect(handler).not_to have_received(:call)
+          raise ActiveRecord::Rollback
+        end
+        expect(handler).to have_received(:call).with(:de)
+
+        I18n.enforce_available_locales = true
+      end
     end
 
     context "without transaction" do