ghazel-erubis 2.6.5.1

data/test/test-main.rb

@@ -0,0 +1,721 @@
+##
+## $Rev$
+## $Release: 2.6.5 $
+## $Date$
+##
+
+require  "#{File.dirname(__FILE__)}/test.rb"
+
+require 'tempfile'
+require 'erubis/main'
+
+
+$script = File.dirname(TESTDIR) + '/bin/erubis'
+#if test(?f, 'bin/erubis')
+#  $script = 'bin/erubis'
+#elsif test(?f, '../bin/erubis')
+#  $script = '../bin/erubis'
+#end
+
+
+class StringWriter < String
+  def write(arg)
+    self << arg
+  end
+  def flush(*args)
+    # pass
+  end
+end
+
+class Erubis::Main
+  public :usage
+  public :show_properties
+  public :show_enhancers
+end
+
+
+class MainTest < Test::Unit::TestCase
+
+  INPUT = <<'END'
+list:
+<% list = ['<aaa>', 'b&b', '"ccc"']
+   for item in list %>
+  - <%= item %>
+<% end %>
+user: <%= defined?(user) ? user : "(none)" %>
+END
+  INPUT2 = INPUT.gsub(/\blist([^:])/, '@list\1').gsub(/\buser([^:])/, '@user\1')
+
+#  SRC = <<'END'
+#_buf = ''; _buf << "list:\n"
+# list = ['<aaa>', 'b&b', '"ccc"']
+#   for item in list
+#_buf << "  - "; _buf << ( item ).to_s; _buf << "\n"
+# end
+#_buf << "user: "; _buf << ( defined?(user) ? user : "(none)" ).to_s; _buf << "\n"
+#_buf
+#END
+  SRC = <<'END'
+_buf = ''; _buf << 'list:
+'; list = ['<aaa>', 'b&b', '"ccc"']
+   for item in list 
+ _buf << '  - '; _buf << ( item ).to_s; _buf << '
+'; end 
+ _buf << 'user: '; _buf << ( defined?(user) ? user : "(none)" ).to_s; _buf << '
+';
+_buf.to_s
+END
+#  SRC2 = SRC.gsub(/\blist /, '@list ').gsub(/\buser /, '@user ')
+
+  OUTPUT = <<'END'
+list:
+  - <aaa>
+  - b&b
+  - "ccc"
+user: (none)
+END
+
+  ESCAPED_OUTPUT = <<'END'
+list:
+  - &lt;aaa&gt;
+  - b&amp;b
+  - &quot;ccc&quot;
+user: (none)
+END
+
+
+  PI_INPUT = <<'END'
+<ul>
+  <?rb @list = ['<aaa>', 'b&b', '"ccc"']
+   for item in @list ?>
+  <li>@{item}@ / @!{item}@
+      <%= item %> / <%== item %></li>
+  <?rb end ?>
+<ul>
+END
+
+  PI_SRC = <<'END'
+_buf = ''; _buf << '<ul>
+';   @list = ['<aaa>', 'b&b', '"ccc"']
+   for item in @list 
+ _buf << '  <li>'; _buf << Erubis::XmlHelper.escape_xml(item); _buf << ' / '; _buf << (item).to_s; _buf << '
+      '; _buf << ( item ).to_s; _buf << ' / '; _buf << Erubis::XmlHelper.escape_xml( item ); _buf << '</li>
+';   end 
+ _buf << '<ul>
+';
+_buf.to_s
+END
+
+  PI_ESCAPED_SRC = <<'END'
+_buf = ''; _buf << '<ul>
+';   @list = ['<aaa>', 'b&b', '"ccc"']
+   for item in @list 
+ _buf << '  <li>'; _buf << (item).to_s; _buf << ' / '; _buf << Erubis::XmlHelper.escape_xml(item); _buf << '
+      '; _buf << Erubis::XmlHelper.escape_xml( item ); _buf << ' / '; _buf << ( item ).to_s; _buf << '</li>
+';   end 
+ _buf << '<ul>
+';
+_buf.to_s
+END
+
+  PI_OUTPUT = <<'END'
+<ul>
+  <li>&lt;aaa&gt; / <aaa>
+      <aaa> / &lt;aaa&gt;</li>
+  <li>b&amp;b / b&b
+      b&b / b&amp;b</li>
+  <li>&quot;ccc&quot; / "ccc"
+      "ccc" / &quot;ccc&quot;</li>
+<ul>
+END
+
+  PI_ESCAPED_OUTPUT = <<'END'
+<ul>
+  <li><aaa> / &lt;aaa&gt;
+      &lt;aaa&gt; / <aaa></li>
+  <li>b&b / b&amp;b
+      b&amp;b / b&b</li>
+  <li>"ccc" / &quot;ccc&quot;
+      &quot;ccc&quot; / "ccc"</li>
+<ul>
+END
+
+  def _test()
+    if @filename.nil?
+      method = (caller[0] =~ /in `(.*)'/) && $1    #'
+      method =~ /block in (.*)/ and method = $1    # for Ruby 1.9
+      @filename = "tmp.#{method}"
+    end
+    File.open(@filename, 'w') {|f| f.write(@input) } if @filename
+    begin
+      argv = @options.is_a?(Array) ? @options.dup : @options.split
+      argv << @filename if @filename
+      $stdout = output = StringWriter.new
+      Erubis::Main.new.execute(argv)
+    ensure
+      $stdout = STDOUT
+      File.unlink(@filename) if @filename && test(?f, @filename)
+    end
+    assert_text_equal(@expected, output)
+  end
+
+  def _error_test(errclass, errmsg)
+    ex = assert_raise(errclass) { _test() }
+    assert_equal(errmsg, ex.message)
+  end
+
+
+  def test_help      # -h
+    @options = '-h'
+    m = Erubis::Main.new
+    @expected = m.usage() + "\n" + m.show_properties() + m.show_enhancers()
+    @filename = false
+    _test()
+  end
+
+  def test_version    # -v
+    @options = '-v'
+    @expected = (("$Release: 2.6.5 $" =~ /[.\d]+/) && $&) + "\n"
+    @filename = false
+    _test()
+  end
+
+
+  def test_basic1
+    @input    = INPUT
+    @expected = OUTPUT
+    @options  = ''
+    _test()
+  end
+
+
+  def test_source1    # -x
+    @input    = INPUT
+    @expected = SRC
+    @options  = '-x'
+    _test()
+  end
+
+
+  def _with_dummy_file
+    bindir = File.join(File.dirname(File.dirname(__FILE__)), 'bin')
+    env_path = ENV['PATH']
+    env__    = ENV['_']
+    begin
+      ENV['PATH'] = bindir + File::PATH_SEPARATOR + ENV['PATH']
+      ENV['_'] = 'erubis'
+      Tempfile.open(self.name.gsub(/[^\w]/,'_')) do |f|
+        f.write(INPUT)
+        f.flush
+        yield(f.path)
+      end
+    ensure
+      ENV['PATH'] = env_path
+      ENV['_']    = env__    if env__
+    end
+  end
+
+
+  def test_syntax1    # -z (syntax ok)
+    @input    = INPUT
+    @expected = "Syntax OK\n"
+    @options  = '-z'
+    _test()
+    #
+    _with_dummy_file do |filepath|
+      actual = `erubis #{@options} #{filepath}`
+      assert_equal @expected, actual
+    end
+  end
+
+
+  def test_syntax2    # -z (syntax error)
+    inputs = []
+    inputs << <<'END'
+<ul>
+<% for item in list %>
+  <li><%= item[:name]] %></li>
+<% end %>
+</ul>
+END
+    inputs << <<'END'
+<ul>
+<% for item in list %>
+  <li><%= item[:name] %></li>
+<% edn %>
+</ul>
+END
+    basename = 'tmp.test_syntax2_%d.rhtml'
+    filenames = [ basename % 0, basename % 1 ]
+    errmsgs = []
+    if ruby19?
+      errmsgs << <<'END'
+3: syntax error, unexpected ']', expecting ')'
+ _buf << '  <li>'; _buf << ( item[:name]] ).to_s; _buf << '</li>
+                                         ^
+-:4: syntax error, unexpected keyword_end, expecting ')'
+'; end 
+      ^
+-:7: syntax error, unexpected $end, expecting ')'
+END
+      errmsgs << <<'END'
+7: syntax error, unexpected $end, expecting keyword_end
+END
+    else
+      errmsgs << <<'END'
+3: syntax error, unexpected ']', expecting ')'
+ _buf << '  <li>'; _buf << ( item[:name]] ).to_s; _buf << '</li>
+                                         ^
+-:4: syntax error, unexpected kEND, expecting ')'
+'; end 
+      ^
+-:7: syntax error, unexpected $end, expecting ')'
+END
+      errmsgs << <<'END'
+7: syntax error, unexpected $end, expecting kEND
+END
+    end
+    #
+    max = inputs.length
+    (0...max).each do |i|
+      @input    = inputs[i]
+      @expected = "tmp.test_syntax2:#{errmsgs[i]}"
+      @options  = '-z'
+      _test()
+    end
+    #
+    begin
+      (0...max).each do |i|
+        File.open(filenames[i], 'w') {|f| f.write(inputs[i]) }
+      end
+      @input = '<ok/>'
+      @expected = ''
+      @options = '-z'
+      (0...max).each do |i|
+        @expected << "#{filenames[i]}:#{errmsgs[i]}"
+        @options << " #{filenames[i]}"
+      end
+      _test()
+    ensure
+      (0...max).each do |i|
+        File.unlink(filenames[i]) if test(?f, filenames[i])
+      end
+    end
+  end
+
+
+  def test_pattern1   # -p
+    @input    = INPUT.gsub(/<%/, '<!--%').gsub(/%>/, '%-->')
+    @expected = OUTPUT
+    #@options  = "-p '<!--% %-->'"
+    @options  = ["-p", "<!--% %-->"]
+    _test()
+  end
+
+
+#  def test_class1     # -C
+#    @input    = INPUT
+#    @expected = OUTPUT.gsub(/<aaa>/, '&lt;aaa&gt;').gsub(/b&b/, 'b&amp;b').gsub(/"ccc"/, '&quot;ccc&quot;')
+#    @options  = "-C XmlEruby"
+#    _test()
+#  end
+
+
+  def test_notrim1    # --trim=false
+    @input   = INPUT
+    @expected = <<'END'
+list:
+
+  - <aaa>
+
+  - b&b
+
+  - "ccc"
+
+user: (none)
+END
+    @options = "--trim=false"  # -T
+    _test()
+  end
+
+
+  def test_notrim2    # --trim=false
+    @input    = INPUT
+#    @expected = <<'END'
+#_buf = ''; _buf << "list:\n"
+# list = ['<aaa>', 'b&b', '"ccc"']
+#   for item in list ; _buf << "\n"
+#_buf << "  - "; _buf << ( item ).to_s; _buf << "\n"
+# end ; _buf << "\n"
+#_buf << "user: "; _buf << ( defined?(user) ? user : "(none)" ).to_s; _buf << "\n"
+#_buf
+#END
+    @expected = <<'END'
+_buf = ''; _buf << 'list:
+'; list = ['<aaa>', 'b&b', '"ccc"']
+   for item in list ; _buf << '
+'; _buf << '  - '; _buf << ( item ).to_s; _buf << '
+'; end ; _buf << '
+'; _buf << 'user: '; _buf << ( defined?(user) ? user : "(none)" ).to_s; _buf << '
+';
+_buf.to_s
+END
+    @options = "-x --trim=false"   # -xT
+    _test()
+  end
+
+
+  #--
+  #def test_context1
+  #  @input    = INPUT
+  #  @expected = OUTPUT.gsub(/\(none\)/, 'Hello')
+  #  @options  = '--user=Hello'
+  #  _test()
+  #end
+  #++
+
+
+  def test_datafile1      # -f data.yaml
+    datafile = "test.context1.yaml"
+    @input    = INPUT2
+    @expected = OUTPUT.gsub(/\(none\)/, 'Hello')
+    @options  = "-f #{datafile}"
+    #
+    str = <<-END
+    user:  Hello
+    password:  world
+    END
+    File.open(datafile, 'w') {|f| f.write(str) }
+    begin
+      _test()
+    ensure
+      File.unlink(datafile) if test(?f, datafile)
+    end
+  end
+
+
+  def test_datafile2      # -f data.rb
+    datafile = "test.context1.rb"
+    @input    = INPUT2
+    @expected = OUTPUT.gsub(/\(none\)/, 'Hello')
+    @options  = "-f #{datafile}"
+    #
+    str = <<-END
+    @user = 'Hello'
+    @password = 'world'
+    END
+    File.open(datafile, 'w') {|f| f.write(str) }
+    begin
+      _test()
+    ensure
+      File.unlink(datafile) if test(?f, datafile)
+    end
+  end
+
+
+  def test_untabify1  # -t (obsolete)
+    yamlfile = "test.context2.yaml"
+    @input    = INPUT2
+    @expected = OUTPUT.gsub(/\(none\)/, 'Hello')
+    @options  = "-tf #{yamlfile}"
+    #
+    yaml = <<-END
+    user:	Hello
+    password:	world
+    END
+    File.open(yamlfile, 'w') {|f| f.write(yaml) }
+    begin
+      _test()
+    ensure
+      File.unlink(yamlfile) if test(?f, yamlfile)
+    end
+  end
+
+
+  def test_untabify2  # -T
+    yamlfile = "test.context2.yaml"
+    @input    = INPUT2
+    @expected = OUTPUT.gsub(/\(none\)/, 'Hello')
+    @options  = "-Tf #{yamlfile}"
+    #
+    yaml = <<-END
+    user: Hello
+    items:
+	- aaa
+	- bbb
+	- ccc
+    END
+    File.open(yamlfile, 'w') {|f| f.write(yaml) }
+    assert_raise(ArgumentError) do
+      _test()
+    end
+    File.open(yamlfile, 'w') {|f| f.write(yaml.gsub(/\t/, ' '*8)) }
+    _test()
+  ensure
+      File.unlink(yamlfile) if test(?f, yamlfile)
+  end
+
+
+  def test_symbolify1 # -S
+    yamlfile = "test.context3.yaml"
+    @input    = <<END
+<% for h in @list %>
+<tr>
+ <td><%= h[:name] %></td><td><%= h[:mail] %></td>
+</tr>
+<% end %>
+END
+    @expected = <<END
+<tr>
+ <td>foo</td><td>foo@mail.com</td>
+</tr>
+<tr>
+ <td>bar</td><td>bar@mail.org</td>
+</tr>
+END
+    @options  = "-f #{yamlfile} -S"
+    #
+    yaml = <<-END
+list:
+  - name:  foo
+    mail:  foo@mail.com
+  - name:  bar
+    mail:  bar@mail.org
+END
+    File.open(yamlfile, 'w') { |f| f.write(yaml) }
+    begin
+      _test()
+    ensure
+      File.unlink(yamlfile) if test(?f, yamlfile)
+    end
+  end
+
+
+  def test_result1   # -B
+    yamlfile = "test.context4.yaml"
+    #
+    @input = <<'END'
+user = <%= user %>
+<% for item in list %>
+ - <%= item %>
+<% end %>
+END
+    @expected = <<'END'
+user = World
+ - aaa
+ - bbb
+ - ccc
+END
+    @options = "-f #{yamlfile} -B "
+    #
+    yaml = <<-END
+user: World
+list:
+  - aaa
+  - bbb
+  - ccc
+END
+    File.open(yamlfile, 'w') {|f| f.write(yaml) }
+    begin
+      _test()
+    ensure
+      File.unlink(yamlfile) if test(?f, yamlfile)
+    end
+  end
+
+
+  def test_context1   # -c
+    @input = <<'END'
+user = <%= @user %>
+<% for item in @list %>
+ - <%= item %>
+<% end %>
+END
+    @expected = <<'END'
+user = World
+ - aaa
+ - bbb
+ - ccc
+END
+    #
+    @options = ['-c', '{user: World, list: [aaa, bbb, ccc]}']
+    _test()
+    @options = ['-c', '@user="World"; @list=%w[aaa bbb ccc]']
+    _test()
+  end
+
+
+  def test_include1   # -I
+    dir = 'foo'
+    lib = 'bar'
+    Dir.mkdir dir unless test(?d, dir)
+    filename = "#{dir}/#{lib}.rb"
+    File.open(filename, 'w') do |f|
+      f.write <<-'END'
+        def escape(str)
+          return "<#{str.upcase}>"
+        end
+      END
+    end
+    #
+    @input    = "<% require '#{lib}' %>\n" + INPUT.gsub(/<%= item %>/, '<%= escape(item) %>')
+    @expected = OUTPUT.gsub(/<aaa>/, '<<AAA>>').gsub(/b\&b/, '<B&B>').gsub(/"ccc"/, '<"CCC">')
+    @options  = "-I #{dir}"
+    #
+    begin
+      _test()
+    ensure
+      File.unlink filename if test(?f, filename)
+      Dir.rmdir dir if test(?d, dir)
+    end
+  end
+
+
+  def test_require1   # -r
+    dir = 'foo'
+    lib = 'bar'
+    Dir.mkdir dir unless test(?d, dir)
+    filename = "#{dir}/#{lib}.rb"
+    File.open(filename, 'w') do |f|
+      f.write <<-'END'
+        def escape(str)
+          return "<#{str.upcase}>"
+        end
+      END
+    end
+    #
+    @input    = INPUT.gsub(/<%= item %>/, '<%= escape(item) %>')
+    @expected = OUTPUT.gsub(/<aaa>/, '<<AAA>>').gsub(/b\&b/, '<B&B>').gsub(/"ccc"/, '<"CCC">')
+    @options  = "-I #{dir} -r #{lib}"
+    #
+    begin
+      _test()
+    ensure
+      File.unlink filename if test(?f, filename)
+      Dir.rmdir dir if test(?d, dir)
+    end
+  end
+
+
+  def test_enhancers1 # -E
+    @input   = <<END
+<% list = %w[<aaa> b&b "ccc"] %>
+% for item in list
+ - <%= item %> : <%== item %>
+ - [= item =] : [== item =]
+% end
+END
+    @expected = <<END
+ - &lt;aaa&gt; : <aaa>
+ - &lt;aaa&gt; : <aaa>
+ - b&amp;b : b&b
+ - b&amp;b : b&b
+ - &quot;ccc&quot; : "ccc"
+ - &quot;ccc&quot; : "ccc"
+END
+    @options = "-E Escape,PercentLine,HeaderFooter,BiPattern"
+    _test()
+  end
+
+
+  def test_bodyonly1  # -b
+    @input = INPUT
+    @expected = SRC.sub(/\A_buf = '';/,'').sub(/\n_buf.to_s\n\z/,'')
+    @options = '-b -x'
+    _test()
+  end
+
+
+  def test_escape1  # -e
+    @input = INPUT
+    @expected = SRC.gsub(/<< \((.*?)\).to_s;/, '<< Erubis::XmlHelper.escape_xml(\1);')
+    @options = '-ex'
+    _test()
+  end
+
+
+  def test_invalid_option  # -1 (invalid option)
+    @input = INPUT
+    @options = '-1'
+    _error_test(Erubis::CommandOptionError, "-1: unknown option.")
+  end
+
+
+  def test_invalid_enhancer  # -E hoge
+    @options = '-E hoge'
+    errmsg = "hoge: no such Enhancer (try '-h' to show all enhancers)."
+    _error_test(Erubis::CommandOptionError, errmsg)
+  end
+
+
+  def test_invalid_lang  # -l hoge
+    @options = '-l hoge'
+    errmsg = "-l hoge: invalid language name (class Erubis::Ehoge not found)."
+    _error_test(Erubis::CommandOptionError, errmsg)
+  end
+
+
+  def test_missing_argument  # -E
+    @filename = false
+    @options = '-E'
+    _error_test(Erubis::CommandOptionError, "-E: enhancers required.")
+    @options = '-l'
+    _error_test(Erubis::CommandOptionError, "-l: lang required.")
+  end
+
+
+  def test_pi1  # --pi -x
+    @input = PI_INPUT
+    @expected = PI_SRC
+    @options = '-x --pi'
+    _test()
+  end
+
+  def test_pi2  # --pi -x --escape=false
+    @input = PI_INPUT
+    @expected = PI_ESCAPED_SRC
+    @options = '-x --pi --escape=false'
+    _test()
+  end
+
+  def test_pi3  # --pi
+    @input = PI_INPUT
+    @expected = PI_OUTPUT
+    @options = '--pi'
+    _test()
+  end
+
+  def test_pi4  # --pi --escape=false
+    @input = PI_INPUT
+    @expected = PI_ESCAPED_OUTPUT
+    @options = '--pi --escape=false'
+    _test()
+  end
+
+  def test_pi5  # --pi=ruby -x
+    @input = PI_INPUT.gsub(/<\?rb/, '<?ruby')
+    @expected = PI_SRC
+    @options = '--pi=ruby -x'
+    _test()
+  end
+
+  def test_pi6  # --pi -xl java
+    @input = <<'END'
+<?java for (int i = 0; i < arr.length; i++) { ?>
+  - @{arr[i]}@ / @!{arr[i]}@
+<?java } ?>
+END
+    @expected = <<'END'
+StringBuffer _buf = new StringBuffer(); for (int i = 0; i < arr.length; i++) { 
+_buf.append("  - "); _buf.append(escape(arr[i])); _buf.append(" / "); _buf.append(arr[i]); _buf.append("\n");
+ } 
+return _buf.toString();
+END
+    @options = '--pi -xl java'
+    _test()
+  end
+
+
+  self.post_definition()
+
+end