Test: Refactor logic of recovering global logger

Signed-off-by: Daijiro Fukuda <[email protected]>
fluent · Feb 16, 2023 · 60c94ad · 60c94ad
1 parent ade4857
commit 60c94ad
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/test/test_logger_initializer.rb b/test/test_logger_initializer.rb
@@ -7,15 +7,15 @@ class LoggerInitializerTest < ::Test::Unit::TestCase
   def setup
     @stored_global_logger = $log
     Dir.mktmpdir do |tmp_dir|
-      begin
-        @tmp_dir = Pathname(tmp_dir)
-        yield
-      ensure
-        $log = @stored_global_logger
-      end
+      @tmp_dir = Pathname(tmp_dir)
+      yield
     end
   end
 
+  def teardown
+    $log = @stored_global_logger
+  end
+
   test 'when path is given' do
     path = @tmp_dir + 'log' + 'fluent_with_path.log'
     logger = Fluent::Supervisor::LoggerInitializer.new(path.to_s, Fluent::Log::LEVEL_DEBUG, nil, nil, {})