require 'erb'
require File.expand_path('../../../../spec_helper', __FILE__)

describe "ERB::DefMethod.def_erb_method" do


  input = <<'END'
<% for item in @items %>
<b><%= item %></b>
<% end %>
END


  it "define method to render eRuby file as an instance method of current module" do
    expected = <<'END'

<b>10</b>

<b>20</b>

<b>30</b>

END
    #
    begin
      file = tmp('_example.rhtml')
      File.open(file, 'w') {|f| f.write(input) }
      klass = Class.new do
        extend ERB::DefMethod
        def_erb_method('render()', file)
        def initialize(items)
          @items = items
        end
      end
      klass.new([10,20,30]).render().should == expected
    ensure
      rm_r file
    end

  end


  it "define method to render eRuby object as an instance method of current module" do
    expected = <<'END'
<b>10</b>
<b>20</b>
<b>30</b>
END
    #
    MY_INPUT4_FOR_ERB = input
    class MyClass4ForErb
      extend ERB::DefMethod
      if RUBY_VERSION >= '2.6'
        erb = ERB.new(MY_INPUT4_FOR_ERB, trim_mode: '<>')
      else
        erb = ERB.new(MY_INPUT4_FOR_ERB, nil, '<>')
      end
      def_erb_method('render()', erb)
      def initialize(items)
        @items = items
      end
    end
    MyClass4ForErb.new([10,20,30]).render().should == expected
  end


end