Omission included when truncating by break_token.

lib/truncate_html/html_truncator.rb

@@ -14,7 +14,7 @@ def initialize(original_html, options = {})
     def truncate
       return @omission if @chars_remaining < 0
       @original_html.html_tokens.each do |token|
-        if @chars_remaining <= 0 || truncate_token?(token)
+        if @chars_remaining <= 0
           close_open_tags
           break
         else
@@ -50,22 +50,25 @@ def build_output
 
     def process_token(token)
       append_to_result(token)
-      if token.html_tag?
+      if token.html_tag? && @chars_remaining > 0
         if token.open_tag?
           @open_tags << token
         else
           remove_latest_open_tag(token)
         end
       elsif !token.html_comment?
         @chars_remaining -= (@word_boundary ? token.length : token[0, @chars_remaining].length)
-        if @chars_remaining <= 0
-          @truncated_html[-1] = @truncated_html[-1].rstrip + @omission
-        end
+      end
+
+      if @chars_remaining <= 0
+        @truncated_html[-1] = @truncated_html[-1].rstrip + @omission
       end
     end
 
     def append_to_result(token)
-      if token.html_tag? || token.html_comment?
+      if truncate_token?(token)
+        @chars_remaining = 0
+      elsif token.html_tag? || token.html_comment?
         @truncated_html << token
       elsif @word_boundary
         @truncated_html << token if (@chars_remaining - token.length) >= 0

spec/truncate_html/html_truncator_spec.rb

@@ -158,7 +158,7 @@ def truncate(html, opts = {})
       truncate('This is line one. This is line <!-- truncate --> two.', :length => 30, :break_token => '<!-- truncate -->').should == 'This is line one. This is...'
     end
     it 'truncates before the length param if the break_token is before the token at "length"' do
-      truncate('This is line one. <!-- truncate --> This is line two.', :length => 30, :break_token => '<!-- truncate -->').should == 'This is line one.'
+      truncate('This is line one. <!-- truncate --> This is line two.', :length => 30, :break_token => '<!-- truncate -->').should == 'This is line one....'
     end
   end
 
@@ -170,7 +170,7 @@ def truncate(html, opts = {})
       truncate('This is line one. This is line <!-- break --> two.', :length => 30, :break_token => '<!-- break -->').should == 'This is line one. This is...'
     end
     it 'truncates before the length param if the break_token is before the token at "length"' do
-      truncate('This is line one. <!-- break --> This is line two.', :length => 30, :break_token => '<!-- break -->').should == 'This is line one.'
+      truncate('This is line one. <!-- break --> This is line two.', :length => 30, :break_token => '<!-- break -->').should == 'This is line one....'
     end
   end
 
@@ -182,7 +182,7 @@ def truncate(html, opts = {})
       truncate('This is line one. This is line <break /> two.', :length => 30, :break_token => '<break />').should == 'This is line one. This is...'
     end
     it 'truncates before the length param if the break_token is before the token at "length"' do
-      truncate('This is line one. <break /> This is line two.', :length => 30, :break_token => '<break />').should == 'This is line one.'
+      truncate('This is line one. <break /> This is line two.', :length => 30, :break_token => '<break />').should == 'This is line one....'
     end
   end
 
@@ -194,7 +194,7 @@ def truncate(html, opts = {})
       truncate('This is line one. This is line foobar two.', :length => 30, :break_token => 'foobar').should == 'This is line one. This is...'
     end
     it 'truncates before the length param if the break_token is before the token at "length"' do
-      truncate('This is line one. foobar This is line two.', :length => 30, :break_token => 'foobar').should == 'This is line one.'
+      truncate('This is line one. foobar This is line two.', :length => 30, :break_token => 'foobar').should == 'This is line one....'
     end
   end
 
@@ -204,4 +204,10 @@ def truncate(html, opts = {})
         '<h1>hello <!-- stuff --> and <!-- la -->...</h1>'
     end
   end
+
+  context 'when truncating by break_token and using a omission' do
+    it 'includes the omission after the truncation' do
+      truncate('This is the time to truncate this. Do it properly!', :length => 50, :break_token => 'truncate').should == 'This is the time to...'
+    end
+  end
 end