Show contents list when first item has an image

app/presenters/content_item/contents_list.rb

@@ -20,6 +20,7 @@ def show_content_list?
       return false unless contents_items.count > 1
       contents_items.count > 2 ||
         first_item_has_long_table? ||
+        first_item_has_image? ||
         first_item_has_long_content?
     end
 
@@ -67,8 +68,14 @@ def find_table
       end
     end
 
+    def first_item_has_image?
+      element = first_item.next_element
+
+      until element.name == 'h2'
+        if element.name == 'div' && element['class'] == 'img'
+          return true
         else
-          current_element = current_element.next_element
+          element = element.next_element
         end
       end
     end

test/presenters/content_item/contents_list_test.rb

@@ -144,4 +144,16 @@ def body
     end
     refute @contents_list.show_content_list?
   end
+
+  test "#show_content_list? returns true if image is present in the first item" do
+    class << @contents_list
+      def body
+        "<h2 id='one'>One</h2>
+        <div class='img'><img src='www.gov.uk/img.png'></div>
+        <h2 id='two'>Two</h2>
+        <p>#{Faker::Lorem.sentence}</p>"
+      end
+    end
+    assert @contents_list.show_content_list?
+  end
 end