Add a couple more tests for #128.

rgrove · Feb 4, 2015 · 2639414 · 2639414
1 parent 7a43f1d
commit 2639414
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/test/test_sanitize.rb b/test/test_sanitize.rb
@@ -22,6 +22,10 @@
         @s.document(input)
         input.must_equal('<!DOCTYPE html><b>foo</b>')
       end
+
+      it 'should not choke on frozen documents' do
+        @s.document('<!doctype html><html><b>foo</b>'.freeze).must_equal "<html>foo</html>\n"
+      end
     end
 
     describe '#fragment' do
@@ -43,7 +47,7 @@
         @s.fragment('<!DOCTYPE html><html><body><b>foo</b></body></html>').must_equal 'foo'
       end
 
-      it 'should not choke on fragments which are frozen' do
+      it 'should not choke on frozen fragments' do
         @s.fragment('<b>foo</b>'.freeze).must_equal 'foo'
       end
     end

diff --git a/test/test_unicode.rb b/test/test_unicode.rb
@@ -11,6 +11,17 @@
       @s = Sanitize.new(Sanitize::Config::RELAXED)
     end
 
+    it 'should not modify the input string' do
+      fragment = "a\u0340b\u0341c"
+      document = "a\u0340b\u0341c"
+
+      @s.document(document)
+      @s.fragment(fragment)
+
+      fragment.must_equal "a\u0340b\u0341c"
+      document.must_equal "a\u0340b\u0341c"
+    end
+
     it 'should strip deprecated grave and acute clones' do
       @s.document("a\u0340b\u0341c").must_equal "<html><head></head><body>abc</body></html>\n"
       @s.fragment("a\u0340b\u0341c").must_equal 'abc'