1import os
2import unittest
3from nose.plugins.doctests import Doctest
4from nose.plugins import PluginTester
5
6support = os.path.join(os.path.dirname(__file__), 'support')
7
8class TestDoctestPlugin(PluginTester, unittest.TestCase):
9    activate = '--with-doctest'
10    args = ['-v']
11    plugins = [Doctest()]
12    suitepath = os.path.join(support, 'dtt')
13
14    def runTest(self):
15        print str(self.output)
16
17        assert 'Doctest: some_mod ... ok' in self.output
18        assert 'Doctest: some_mod.foo ... ok' in self.output
19        assert 'Ran 2 tests' in self.output
20        assert str(self.output).strip().endswith('OK')
21
22
23class TestDoctestFiles(PluginTester, unittest.TestCase):
24    activate = '--with-doctest'
25    args = ['-v', '--doctest-extension=.txt']
26    plugins = [Doctest()]
27    suitepath = os.path.join(support, 'dtt', 'docs')
28
29    def runTest(self):
30        print str(self.output)
31
32        expect = [
33            'Doctest: doc.txt ... ok',
34            'Doctest: errdoc.txt ... FAIL'
35            ]
36        for line in self.output:
37            if not line.strip():
38                continue
39            if line.startswith('='):
40                break
41            self.assertEqual(line.strip(), expect.pop(0))
42
43if __name__ == '__main__':
44    unittest.main()
45