Get the flat_rootname right for nested CodeUnits. Thanks, Christian Heimes.

1 files changed, 60 insertions, 0 deletions

diff --git a/test/test_codeunit.py b/test/test_codeunit.py
new file mode 100644
index 00000000..d03ff8d9
--- /dev/null
+++ b/test/test_codeunit.py
@@ -0,0 +1,60 @@
+"""Tests for coverage.codeunit"""
+
+import os, sys
+
+from coverage.codeunit import code_unit_factory
+from coverage.files import FileLocator
+from coveragetest import CoverageTest
+
+# pylint: disable-msg=F0401
+# Unable to import 'aa' (No module named aa)
+
+class CodeUnitTest(CoverageTest):
+    """Tests for coverage.codeunit"""
+
+    def setUp(self):
+        super(CodeUnitTest, self).setUp()
+        # Parent class saves and restores sys.path, we can just modify it.
+        testmods = self.nice_file(os.path.dirname(__file__), 'modules')
+        sys.path.append(testmods)
+        
+    def test_filenames(self):
+        acu = code_unit_factory("aa/afile.py", FileLocator())
+        bcu = code_unit_factory("aa/bb/bfile.py", FileLocator())
+        ccu = code_unit_factory("aa/bb/cc/cfile.py", FileLocator())
+        self.assertEqual(acu[0].name, "aa/afile")
+        self.assertEqual(bcu[0].name, "aa/bb/bfile")
+        self.assertEqual(ccu[0].name, "aa/bb/cc/cfile")
+        self.assertEqual(acu[0].flat_rootname(), "aa_afile")
+        self.assertEqual(bcu[0].flat_rootname(), "aa_bb_bfile")
+        self.assertEqual(ccu[0].flat_rootname(), "aa_bb_cc_cfile")
+        self.assertEqual(acu[0].source_file().read(), "# afile.py\n")
+        self.assertEqual(bcu[0].source_file().read(), "# bfile.py\n")
+        self.assertEqual(ccu[0].source_file().read(), "# cfile.py\n")
+
+    def test_modules(self):
+        import aa, aa.bb, aa.bb.cc
+        cu = code_unit_factory([aa, aa.bb, aa.bb.cc], FileLocator())
+        self.assertEqual(cu[0].name, "aa")
+        self.assertEqual(cu[1].name, "aa.bb")
+        self.assertEqual(cu[2].name, "aa.bb.cc")
+        self.assertEqual(cu[0].flat_rootname(), "aa")
+        self.assertEqual(cu[1].flat_rootname(), "aa_bb")
+        self.assertEqual(cu[2].flat_rootname(), "aa_bb_cc")
+        self.assertEqual(cu[0].source_file().read(), "# aa\n")
+        self.assertEqual(cu[1].source_file().read(), "# bb\n")
+        self.assertEqual(cu[2].source_file().read(), "")  # yes, empty
+
+    def test_module_files(self):
+        import aa.afile, aa.bb.bfile, aa.bb.cc.cfile
+        cu = code_unit_factory([aa.afile, aa.bb.bfile, aa.bb.cc.cfile],
+                                                                FileLocator())
+        self.assertEqual(cu[0].name, "aa.afile")
+        self.assertEqual(cu[1].name, "aa.bb.bfile")
+        self.assertEqual(cu[2].name, "aa.bb.cc.cfile")
+        self.assertEqual(cu[0].flat_rootname(), "aa_afile")
+        self.assertEqual(cu[1].flat_rootname(), "aa_bb_bfile")
+        self.assertEqual(cu[2].flat_rootname(), "aa_bb_cc_cfile")
+        self.assertEqual(cu[0].source_file().read(), "# afile.py\n")
+        self.assertEqual(cu[1].source_file().read(), "# bfile.py\n")
+        self.assertEqual(cu[2].source_file().read(), "# cfile.py\n")