Fix ParserTestDriver, FormatterTestDriver to let plugin instance alwa…

…ys to have its class name
fluent · Jul 25, 2016 · c474488 · c474488
1 parent de13f7e
commit c474488
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/lib/fluent/test/formatter_test.rb b/lib/fluent/test/formatter_test.rb
@@ -26,7 +26,9 @@ def initialize(klass_or_str, proc=nil, &block)
           if block
             # Create new class for test w/ overwritten methods
             #   klass.dup is worse because its ancestors does NOT include original class name
+            klass_name = klass_or_str.name
             klass_or_str = Class.new(klass_or_str)
+            klass_or_str.define_singleton_method("name") { klass_name }
             klass_or_str.module_eval(&block)
           end
           @instance = klass_or_str.new

diff --git a/lib/fluent/test/parser_test.rb b/lib/fluent/test/parser_test.rb
@@ -25,7 +25,9 @@ def initialize(klass_or_str, format=nil, conf={}, &block)
           if block
             # Create new class for test w/ overwritten methods
             #   klass.dup is worse because its ancestors does NOT include original class name
+            klass_name = klass_or_str.name
             klass_or_str = Class.new(klass_or_str)
+            klass_or_str.define_singleton_method("name") { klass_name }
             klass_or_str.module_eval(&block)
           end
           case klass_or_str.instance_method(:initialize).arity