require 'contest'
require 'tilt'

begin
  if RUBY_VERSION >= '1.9.0'
    require 'csv'
  else
    require 'fastercsv'
  end

  class CSVTemplateTest < Test::Unit::TestCase

    test "registered for '.rcsv' files" do
      assert Tilt.mappings['rcsv'].include?(Tilt::CSVTemplate)
    end

    test "compiles and evaluates the template on #render" do
      template = Tilt::CSVTemplate.new { "csv << ['hello', 'world']" }
      assert_equal "hello,world\n", template.render
    end

    test "can be rendered more than once" do
      template = Tilt::CSVTemplate.new { "csv << [1,2,3]" }
      3.times { assert_equal "1,2,3\n", template.render }
    end

    test "can pass locals" do
      template = Tilt::CSVTemplate.new { 'csv << [1, name]' }
      assert_equal "1,Joe\n", template.render(Object.new, :name => 'Joe')
    end

    test "evaluating in an object scope" do
      template = Tilt::CSVTemplate.new { 'csv << [1, @name]' }
      scope = Object.new
      scope.instance_variable_set :@name, 'Joe'
      assert_equal "1,Joe\n", template.render(scope)
    end

    test "backtrace file and line reporting" do
      data = File.read(__FILE__).split("\n__END__\n").last
      template = Tilt::CSVTemplate.new('test.csv') { data }
      begin
        template.render
        fail 'should have raised an exception'
      rescue => boom
        assert_kind_of NameError, boom
        line = boom.backtrace.grep(/^test\.csv:/).first
        assert line, "Backtrace didn't contain test.csv"
        file, line, meth = line.split(":")
        assert_equal '4', line
      end
    end

  end

rescue LoadError => boom
  warn "Tilt::CSVTemplate (disabled) please install 'fastercsv' if using ruby 1.8.x"
end


__END__
# header
csv << ['Type', 'Age']

raise NameError

# rows
csv << ['Frog', 2]
csv << ['Cat', 5]