From 3a133efc13152763971dce7543a4d451cd073107 Mon Sep 17 00:00:00 2001
From: Takashi Kokubun <takashikkbn@gmail.com>
Date: Sat, 29 Dec 2018 12:40:17 +0900
Subject: [PATCH] Fix ERB warning for Ruby 2.6+ in benchmark.rb

---
 benchmark.rb | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/benchmark.rb b/benchmark.rb
index bdc8959e..dcf1f9f4 100644
--- a/benchmark.rb
+++ b/benchmark.rb
@@ -38,7 +38,11 @@ RBench.run(times) do
     obj = Object.new
 
     Haml::Engine.new(haml_template).def_method(obj, :haml)
-    obj.instance_eval("def erb; #{ERB.new(erb_template, nil, '-').src}; end")
+    if ERB.instance_method(:initialize).parameters.assoc(:key) # Ruby 2.6+
+      obj.instance_eval("def erb; #{ERB.new(erb_template, trim_mode: '-').src}; end")
+    else
+      obj.instance_eval("def erb; #{ERB.new(erb_template, nil, '-').src}; end")
+    end
     obj.instance_eval("def erubi; #{Erubi::Engine.new(erb_template).src}; end")
 
     haml      { obj.haml }