TaggedLogging not broadcasting tags

[mtomiyoshi]

This is similar to #43291, but I am calling the method with a block.

When not using a block it works fine, demonstrated by the test mentioned in the issue:

rails/activesupport/test/tagged_logging_test.rb

Line 221 in f95c0b7

@logger.extend(ActiveSupport::Logger.broadcast(broadcast_logger))

Steps to reproduce

# frozen_string_literal: true

require 'bundler/inline'

gemfile(true) do
  source 'https://rubygems.org'

  git_source(:github) { |repo| "https://github.com/#{repo}.git" }

  gem 'rails', github: 'rails/rails', branch: 'main'
end

require 'active_support'
require 'active_support/core_ext/object/blank'
require 'minitest/autorun'

class BugTest < Minitest::Test
  class MyLogger < ::ActiveSupport::Logger
    def flush(*)
      info '[FLUSHED]'
    end
  end

  def test_keep_broadcast_with_block
    @output = StringIO.new
    @logger = ActiveSupport::TaggedLogging.new(MyLogger.new(@output))

    broadcast_output = StringIO.new
    broadcast_logger = ActiveSupport::TaggedLogging.new(Logger.new(broadcast_output))
    @logger.extend(ActiveSupport::Logger.broadcast(broadcast_logger))

    @logger.tagged('OMG') { |logger| logger.info 'Broadcasting...' }

    assert_equal "[OMG] Broadcasting...\n", @output.string
    assert_equal "[OMG] Broadcasting...\n", broadcast_output.string
  end
end

Expected behavior

Both loggers should log "[OMG] Broadcasting...\n"

Actual behavior

One of them logs "[OMG] Broadcasting...\n", the other logs Broadcasting...\n instead.

Result of the test script:

F

Failure:
BugTest#test_keep_broadcast_with_block [tagged_logger_bug.rb:50]:
--- expected
+++ actual
@@ -1,2 +1,2 @@
-"[OMG] Broadcasting...
+"Broadcasting...
 "

System configuration

Rails version: main

Ruby version: 3.1.1

[Edouard-chin]

Fixed by #44695