* MANIFEST: add test_erb.rb

        * lib/erb.rb, test/erb/test_erb.rb: don't forget filename,
          if both filename and safe_level given. [ruby-dev:23050]


git-svn-id: http://svn.ruby-lang.org/repos/ruby/trunk@5846 b2dd03c8-39d4-4d8f-98ff-823fe69b080e

1 files changed, 40 insertions, 0 deletions

diff --git a/test/erb/test_erb.rb b/test/erb/test_erb.rb
new file mode 100644
index 000000000..864119741
--- /dev/null
+++ b/test/erb/test_erb.rb
@@ -0,0 +1,40 @@
+require 'test/unit'
+require 'erb'
+
+class TestERB < Test::Unit::TestCase
+  class MyError < RuntimeError ; end
+
+  def test_without_filename
+    erb = ERB.new("<% raise ::TestERB::MyError %>")
+    e = assert_raise(MyError) {
+      erb.result
+    }
+    assert_equal("(erb):1", e.backtrace[0])
+  end
+
+  def test_with_filename
+    erb = ERB.new("<% raise ::TestERB::MyError %>")
+    erb.filename = "test filename"
+    e = assert_raise(MyError) {
+      erb.result
+    }
+    assert_equal("test filename:1", e.backtrace[0])
+  end
+
+  def test_without_filename_with_safe_level
+    erb = ERB.new("<% raise ::TestERB::MyError %>", 1)
+    e = assert_raise(MyError) {
+      erb.result
+    }
+    assert_equal("(erb):1", e.backtrace[0])
+  end
+
+  def test_with_filename_and_safe_level
+    erb = ERB.new("<% raise ::TestERB::MyError %>", 1)
+    erb.filename = "test filename"
+    e = assert_raise(MyError) {
+      erb.result
+    }
+    assert_equal("test filename:1", e.backtrace[0])
+  end
+end