Moved in a test for comments (not supported by the html5lib treebuilder yet.

1 files changed, 10 insertions, 1 deletions

diff --git a/src/beautifulsoup/tests/helpers.py b/src/beautifulsoup/tests/helpers.py
index c62bb48..c8189e3 100644
--- a/src/beautifulsoup/tests/helpers.py
+++ b/src/beautifulsoup/tests/helpers.py
@@ -2,7 +2,7 @@
 
 import unittest
 from beautifulsoup import BeautifulSoup
-from beautifulsoup.element import SoupStrainer
+from beautifulsoup.element import Comment, SoupStrainer
 from beautifulsoup.builder.lxml_builder import LXMLTreeBuilder
 
 class SoupTest(unittest.TestCase):
@@ -63,6 +63,15 @@ class BuilderSmokeTest(SoupTest):
         self.assertSoupEquals(
             "<p>Foo<br/>bar</p>", "<p>Foo<br />bar</p>")
 
+    def test_comment(self):
+        # Comments are represented as Comment objects.
+        markup = "<p>foo<!--foobar-->baz</p>"
+        self.assertSoupEquals(markup)
+
+        soup = self.soup(markup)
+        comment = soup.find(text="foobar")
+        self.assertEquals(comment.__class__, Comment)
+
     def test_nested_inline_elements(self):
         # Inline tags can be nested indefinitely.
         b_tag = "<b>Inside a B tag</b>"