1 files changed, 22 insertions, 0 deletions
diff --git a/test/rexml/test_document.rb b/test/rexml/test_document.rb
index 72d0ff696f..02e4a69754 100644
--- a/test/rexml/test_document.rb
+++ b/test/rexml/test_document.rb
@@ -223,4 +223,26 @@ EOX
       end
     end
   end
+
+  class BomTest < self
+    def test_utf_16le
+      xml = <<-EOX.encode("UTF-16LE").force_encoding("ASCII-8BIT")
+<?xml version="1.0" encoding="UTF-16"?>
+<message>Hello world!</message>
+EOX
+      bom = "\ufeff".encode("UTF-16LE").force_encoding("ASCII-8BIT")
+      document = REXML::Document.new(bom + xml)
+      assert_equal("UTF-16", document.encoding)
+    end
+
+    def test_utf_16be
+      xml = <<-EOX.encode("UTF-16BE").force_encoding("ASCII-8BIT")
+<?xml version="1.0" encoding="UTF-16"?>
+<message>Hello world!</message>
+EOX
+      bom = "\ufeff".encode("UTF-16BE").force_encoding("ASCII-8BIT")
+      document = REXML::Document.new(bom + xml)
+      assert_equal("UTF-16", document.encoding)
+    end
+  end
 end