#!/usr/bin/env ruby

require 'test/unit'
require File.dirname(__FILE__) + '/../lib/haml/helpers'

class HelperTest < Test::Unit::TestCase
  include Haml::Helpers

  def test_find_and_flatten
    assert_equal(find_and_flatten("<br/><textarea></textarea><br/>"),
                                  "<br/><textarea></textarea><br/>")

    assert_equal(find_and_flatten("<code lang='ruby'>TEST!</code>\t\t<p></p>"),
                                  "<code lang='ruby'>TEST!</code>\t\t<p></p>")

    assert_equal(find_and_flatten("<pre>Hello\nWorld!\nYOU ARE \rFLAT?\n\rOMGZ!</pre></br>"),
                                  "<pre>Hello&#x000A;World!&#x000A;YOU ARE FLAT?&#x000A;OMGZ!</pre></br>")

    assert_equal(find_and_flatten("<div class='text_area_test_area'>\n  <textarea>Two\n  lines</textarea>\n</div>\n"),
                                  "<div class='text_area_test_area'>\n  <textarea>Two&#x000A;  lines</textarea>\n</div>\n")

    assert_equal(find_and_flatten("<code>Two\nlines</code><pre>a\nb\nc</pre>"),
                                  "<code>Two&#x000A;lines</code><pre>a&#x000A;b&#x000A;c</pre>")

    assert_equal(find_and_flatten("<pre>Two\nlines</pre>\n<pre>a\nb\nc</pre>"),
                                  "<pre>Two&#x000A;lines</pre>\n<pre>a&#x000A;b&#x000A;c</pre>")
  end

  def test_list_of_should_render_correctly
    assert_equal("<li>1</li>\n<li>2</li>", (list_of([1, 2]) { |i| i.to_s}))
    assert_equal("<li>1</li>", (list_of([[1]]) { |i| i.first}))
  end
end