Start some lower level unit tests

mitechie · mitechie · commit 0f9da8ace416 · 2012-04-18T21:01:51.000-04:00
diff --git a/src/readability_lxml/readability.py b/src/readability_lxml/readability.py
@@ -82,10 +82,10 @@ class Document:
     TEXT_LENGTH_THRESHOLD = 25
     RETRY_LENGTH = 250
 
-    def __init__(self, input, **options):
+    def __init__(self, input_doc, **options):
         """Generate the document
 
-        :param input: string of the html content.
+        :param input_doc: string of the html content.
 
         kwargs:
             - attributes:
@@ -95,17 +95,20 @@ def __init__(self, input, **options):
             - url: will allow adjusting links to be absolute
 
         """
-        self.input = input
+        if input_doc is None:
+            raise ValueError('You must supply a document to process.')
+
+        self.input_doc = input_doc
         self.options = options
         self.html = None
 
     def _html(self, force=False):
         if force or self.html is None:
-            self.html = self._parse(self.input)
+            self.html = self._parse(self.input_doc)
         return self.html
 
-    def _parse(self, input):
-        doc = build_doc(input)
+    def _parse(self, input_doc):
+        doc = build_doc(input_doc)
         doc = html_cleaner.clean_html(doc)
         base_href = self.options.get('url', None)
         if base_href:
diff --git a/src/tests/test_readability.py b/src/tests/test_readability.py
@@ -0,0 +1,13 @@
+import unittest
+
+from readability_lxml.readability import Document
+
+
+class TestReadabilityDocument(unittest.TestCase):
+    """Test the Document parser."""
+
+    def test_none_input_raises_exception(self):
+        """Feeding a None input to the document should blow up."""
+
+        doc = None
+        self.assertRaises(ValueError, Document, doc)
