From 59c64439d8cb474c0986993023a08ebbe88656c4 Mon Sep 17 00:00:00 2001
From: Jacques Distler <distler@golem.ph.utexas.edu>
Date: Sat, 24 Feb 2007 00:41:35 -0600
Subject: [PATCH] More unit tests.

---
 test/unit/sanitize_test.rb | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/test/unit/sanitize_test.rb b/test/unit/sanitize_test.rb
index 7a2fd813..db0d4bc2 100644
--- a/test/unit/sanitize_test.rb
+++ b/test/unit/sanitize_test.rb
@@ -40,6 +40,13 @@ class SanitizeTest < Test::Unit::TestCase
     end
   end
 
+  Sanitize::ALLOWED_PROTOCOLS.each do |protocol|
+    define_method "test_should_allow_#{protocol}_uris" do
+      assert_equal "<a href=\"#{protocol}\">foo</a>",
+        sanitize_html(%(<a href="#{protocol}">foo</a>))
+    end
+  end
+
   def test_should_allow_anchors
     assert_equal "<a href=\"foo\">&lt;script>baz&lt;/script></a>",
      sanitize_html("<a href='foo' onclick='bar'><script>baz</script></a>")
@@ -72,7 +79,7 @@ class SanitizeTest < Test::Unit::TestCase
     end
   end
 
-[%(<img src="javascript:alert('XSS');" />), 
+  [%(<img src="javascript:alert('XSS');" />), 
    %(<img src=javascript:alert('XSS') />), 
    %(<img src="JaVaScRiPt:alert('XSS')" />), 
    %(<img src='javascript:alert(&quot;XSS&quot;)' />),