* Small improvements to the HTML2Text code

[rss2maildir.git] / tests / unittests / UnorderedListTests.py

diff --git a/tests/unittests/UnorderedListTests.py b/tests/unittests/UnorderedListTests.py
old mode 100644 (file)
new mode 100755 (executable)
index af09b7c..a495461
--- a/tests/unittests/UnorderedListTests.py
+++ b/tests/unittests/UnorderedListTests.py
@@ -4,26 +4,20 @@ import unittest
 import sys
 import os
 
-class UnorderedListTests(unittest.TestCase):
-    def setUp(self):
-        self.inputpath = os.path.sep.join(os.path.dirname(os.path.realpath(__file__)).split(os.path.sep)[0:-1])
+import ParsingTests
 
-    def testwellformedlist(self):
-        try:
-            from rss2maildir import HTML2Text
-        except:
-            sys.path.append(os.path.sep.join(self.inputpath.split(os.path.sep)[0:-1]))
-            try:
-                from rss2maildir import HTML2Text
-            except:
-                self.assert_(False)
-        input_path = os.path.sep.join(os.path.dirname(os.path.realpath(__file__)).split(os.path.sep)[0:-1])
-        input = open(os.path.join(input_path, "html", "unorderedlist-wellformed.html")).read()
-        expectedoutput = open(os.path.join(input_path, "expected", "unorderedlist-wellformed.txt")).read()
-        parser = HTML2Text()
-        parser.feed(input)
-        output = parser.gettext()
-        self.assertEqual(output, expectedoutput)
+class UnorderedListTests(ParsingTests.ParsingTest):
+    def testWellFormedList(self):
+        return self.runParsingTest("unordered-wellformed")
+
+    def testBadlyFormedList(self):
+        return self.runParsingTest("unordered-badlyformed")
+
+def suite():
+    suite = unittest.TestSuite()
+    suite.addTest(UnorderedListTests("testWellFormedList"))
+    suite.addTest(UnorderedListTests("testBadlyFormedList"))
+    return suite
 
 if __name__ == "__main__":
     unittest.main()