From 81cf7ba9b62a64c84d8806562fdb925c7d9c7949 Mon Sep 17 00:00:00 2001
From: Yorick Peterse <yorickpeterse@gmail.com>
Date: Mon, 18 May 2015 00:32:01 +0200
Subject: [PATCH] Reworked HTML caption closing specs

---
 .../lexer/html_closing_rules/caption_spec.rb  | 60 ++++++++++++++++++-
 1 file changed, 57 insertions(+), 3 deletions(-)

diff --git a/spec/oga/xml/lexer/html_closing_rules/caption_spec.rb b/spec/oga/xml/lexer/html_closing_rules/caption_spec.rb
index e660dac..037cabf 100644
--- a/spec/oga/xml/lexer/html_closing_rules/caption_spec.rb
+++ b/spec/oga/xml/lexer/html_closing_rules/caption_spec.rb
@@ -1,8 +1,52 @@
 require 'spec_helper'
 
 describe Oga::XML::Lexer do
-  describe 'HTML caption elements' do
-    it 'lexes an unclosed <caption> followed by another <caption>' do
+  describe 'using HTML <caption> elements' do
+    it 'lexes an unclosed <caption> followed by a <thead> as separate elements' do
+      lex_html('<caption>foo<thead>bar').should == [
+        [:T_ELEM_NAME, 'caption', 1],
+        [:T_TEXT, 'foo', 1],
+        [:T_ELEM_END, nil, 1],
+        [:T_ELEM_NAME, 'thead', 1],
+        [:T_TEXT, 'bar', 1],
+        [:T_ELEM_END, nil, 1]
+      ]
+    end
+
+    it 'lexes an unclosed <caption> followed by a <tbody> as separate elements' do
+      lex_html('<caption>foo<tbody>bar').should == [
+        [:T_ELEM_NAME, 'caption', 1],
+        [:T_TEXT, 'foo', 1],
+        [:T_ELEM_END, nil, 1],
+        [:T_ELEM_NAME, 'tbody', 1],
+        [:T_TEXT, 'bar', 1],
+        [:T_ELEM_END, nil, 1]
+      ]
+    end
+
+    it 'lexes an unclosed <caption> followed by a <tfoot> as separate elements' do
+      lex_html('<caption>foo<tfoot>bar').should == [
+        [:T_ELEM_NAME, 'caption', 1],
+        [:T_TEXT, 'foo', 1],
+        [:T_ELEM_END, nil, 1],
+        [:T_ELEM_NAME, 'tfoot', 1],
+        [:T_TEXT, 'bar', 1],
+        [:T_ELEM_END, nil, 1]
+      ]
+    end
+
+    it 'lexes an unclosed <caption> followed by a <tr> as separate elements' do
+      lex_html('<caption>foo<tr>bar').should == [
+        [:T_ELEM_NAME, 'caption', 1],
+        [:T_TEXT, 'foo', 1],
+        [:T_ELEM_END, nil, 1],
+        [:T_ELEM_NAME, 'tr', 1],
+        [:T_TEXT, 'bar', 1],
+        [:T_ELEM_END, nil, 1]
+      ]
+    end
+
+    it 'lexes an unclosed <caption> followed by a <caption> as separate elements' do
       lex_html('<caption>foo<caption>bar').should == [
         [:T_ELEM_NAME, 'caption', 1],
         [:T_TEXT, 'foo', 1],
@@ -13,7 +57,7 @@ describe Oga::XML::Lexer do
       ]
     end
 
-    it 'lexes an unclosed <caption> followed by a <colgroup>' do
+    it 'lexes an unclosed <caption> followed by a <colgroup> as separate elements' do
       lex_html('<caption>foo<colgroup>bar').should == [
         [:T_ELEM_NAME, 'caption', 1],
         [:T_TEXT, 'foo', 1],
@@ -23,5 +67,15 @@ describe Oga::XML::Lexer do
         [:T_ELEM_END, nil, 1]
       ]
     end
+
+    it 'lexes an unclosed <caption> followed by a <col> as separate elements' do
+      lex_html('<caption>foo<col>').should == [
+        [:T_ELEM_NAME, 'caption', 1],
+        [:T_TEXT, 'foo', 1],
+        [:T_ELEM_END, nil, 1],
+        [:T_ELEM_NAME, 'col', 1],
+        [:T_ELEM_END, nil, 1]
+      ]
+    end
   end
 end