def test_run(self): indexer = Autoindexer( project_root=self.project_root, index_rst_location=abspath(join(self.index_rst_dir, 'index_initial.rst')) ) indexer._generate_docs_index = Mock() indexer._sphinx_formatter = Mock() indexer._traverse_modules = Mock() indexer.run() indexer._generate_docs_index.assert_called_once() indexer._sphinx_formatter.assert_called_once() indexer._traverse_modules.assert_called_once()
def test_traverse_modules(self): indexer = Autoindexer( project_root=self.project_root, index_rst_location=abspath(join(self.index_rst_dir, 'index_initial.rst')), module_prefixes=['test_mod'] ) sys.path.append(self.project_root) result = indexer._traverse_modules(self.project_root) self.assertDictEqual( { 'test_mod_1': set(), 'test_mod_1.test_class_1': {'TestClass1'}, 'test_mod_1.test_class_2': {'TestClass2'}, 'test_mod_1.test_submod_1': set(), 'test_mod_1.test_submod_1.test_submodule_class': {'TestSubmoduleClass'}, 'test_mod_1.test_utils': set(), 'test_mod_2': set() }, result )