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
)