Fixed bugs in doctest loading from non-module files. Added functional test.

1 files changed, 20 insertions, 0 deletions

diff --git a/functional_tests/test_doctest_plugin.py b/functional_tests/test_doctest_plugin.py
index 0204eb0..c91ecc5 100644
--- a/functional_tests/test_doctest_plugin.py
+++ b/functional_tests/test_doctest_plugin.py
@@ -20,5 +20,25 @@ class TestDoctestPlugin(PluginTester, unittest.TestCase):
         assert str(self.output).strip().endswith('OK')
 
 
+class TestDoctestFiles(PluginTester, unittest.TestCase):
+    activate = '--with-doctest'
+    args = ['-v', '--doctest-extension=.txt']
+    plugins = [Doctest()]
+    suitepath = os.path.join(support, 'dtt', 'docs')
+    
+    def runTest(self):
+        print str(self.output)
+
+        expect = [
+            'Doctest: doc.txt ... ok',
+            'Doctest: errdoc.txt ... FAIL'
+            ]
+        for line in self.output:
+            if not line.strip():
+                continue
+            if line.startswith('='):
+                break
+            self.assertEqual(line.strip(), expect.pop(0))
+
 if __name__ == '__main__':
     unittest.main()