Fix Docs::Parse#document? when document has no doctype

Thibaut · Thibaut · commit d366e14ea77d · 2016-04-10T10:16:24.000-04:00
diff --git a/lib/docs/core/parser.rb b/lib/docs/core/parser.rb
@@ -11,7 +11,7 @@ def html
     private
 
     def document?
-      @content =~ /\A\s*<!doctype/i
+      @content =~ /\A\s*<(?:\!doctype|html)/i
     end
 
     def parse_as_document
diff --git a/test/lib/docs/core/parser_test.rb b/test/lib/docs/core/parser_test.rb
@@ -22,6 +22,9 @@ def parser(content)
       it "returns the <body>" do
         body = '<!doctype html><meta charset=utf-8><title></title><div>Test</div>'
         assert_equal '<div>Test</div>', parser(body).html.inner_html
+
+        body = '<html><meta charset=utf-8><title></title><div>Test</div></html>'
+        assert_equal '<div>Test</div>', parser(body).html.inner_html
       end
     end
   end
