1 files changed, 120 insertions, 0 deletions
diff --git a/tests/test_filereporter.py b/tests/test_filereporter.py
new file mode 100644
index 00000000..473bfe49
--- /dev/null
+++ b/tests/test_filereporter.py
@@ -0,0 +1,120 @@
+"""Tests for coverage.codeunit"""
+
+import os
+import sys
+
+from coverage.plugin import FileReporter
+from coverage.python import PythonFileReporter
+
+from tests.coveragetest import CoverageTest
+
+# pylint: disable=import-error
+# Unable to import 'aa' (No module named aa)
+
+
+def native(filename):
+    """Make `filename` into a native form."""
+    return filename.replace("/", os.sep)
+
+
+class FileReporterTest(CoverageTest):
+    """Tests for FileReporter classes."""
+
+    run_in_temp_dir = False
+
+    def setUp(self):
+        super(FileReporterTest, 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 = PythonFileReporter("aa/afile.py")
+        bcu = PythonFileReporter("aa/bb/bfile.py")
+        ccu = PythonFileReporter("aa/bb/cc/cfile.py")
+        self.assertEqual(acu.name, "aa/afile.py")
+        self.assertEqual(bcu.name, "aa/bb/bfile.py")
+        self.assertEqual(ccu.name, "aa/bb/cc/cfile.py")
+        self.assertEqual(acu.flat_rootname(), "aa_afile_py")
+        self.assertEqual(bcu.flat_rootname(), "aa_bb_bfile_py")
+        self.assertEqual(ccu.flat_rootname(), "aa_bb_cc_cfile_py")
+        self.assertEqual(acu.source(), "# afile.py\n")
+        self.assertEqual(bcu.source(), "# bfile.py\n")
+        self.assertEqual(ccu.source(), "# cfile.py\n")
+
+    def test_odd_filenames(self):
+        acu = PythonFileReporter("aa/afile.odd.py")
+        bcu = PythonFileReporter("aa/bb/bfile.odd.py")
+        b2cu = PythonFileReporter("aa/bb.odd/bfile.py")
+        self.assertEqual(acu.name, "aa/afile.odd.py")
+        self.assertEqual(bcu.name, "aa/bb/bfile.odd.py")
+        self.assertEqual(b2cu.name, "aa/bb.odd/bfile.py")
+        self.assertEqual(acu.flat_rootname(), "aa_afile_odd_py")
+        self.assertEqual(bcu.flat_rootname(), "aa_bb_bfile_odd_py")
+        self.assertEqual(b2cu.flat_rootname(), "aa_bb_odd_bfile_py")
+        self.assertEqual(acu.source(), "# afile.odd.py\n")
+        self.assertEqual(bcu.source(), "# bfile.odd.py\n")
+        self.assertEqual(b2cu.source(), "# bfile.py\n")
+
+    def test_modules(self):
+        import aa
+        import aa.bb
+        import aa.bb.cc
+
+        acu = PythonFileReporter(aa)
+        bcu = PythonFileReporter(aa.bb)
+        ccu = PythonFileReporter(aa.bb.cc)
+        self.assertEqual(acu.name, native("aa.py"))
+        self.assertEqual(bcu.name, native("aa/bb.py"))
+        self.assertEqual(ccu.name, native("aa/bb/cc.py"))
+        self.assertEqual(acu.flat_rootname(), "aa_py")
+        self.assertEqual(bcu.flat_rootname(), "aa_bb_py")
+        self.assertEqual(ccu.flat_rootname(), "aa_bb_cc_py")
+        self.assertEqual(acu.source(), "# aa\n")
+        self.assertEqual(bcu.source(), "# bb\n")
+        self.assertEqual(ccu.source(), "")  # yes, empty
+
+    def test_module_files(self):
+        import aa.afile
+        import aa.bb.bfile
+        import aa.bb.cc.cfile
+
+        acu = PythonFileReporter(aa.afile)
+        bcu = PythonFileReporter(aa.bb.bfile)
+        ccu = PythonFileReporter(aa.bb.cc.cfile)
+        self.assertEqual(acu.name, native("aa/afile.py"))
+        self.assertEqual(bcu.name, native("aa/bb/bfile.py"))
+        self.assertEqual(ccu.name, native("aa/bb/cc/cfile.py"))
+        self.assertEqual(acu.flat_rootname(), "aa_afile_py")
+        self.assertEqual(bcu.flat_rootname(), "aa_bb_bfile_py")
+        self.assertEqual(ccu.flat_rootname(), "aa_bb_cc_cfile_py")
+        self.assertEqual(acu.source(), "# afile.py\n")
+        self.assertEqual(bcu.source(), "# bfile.py\n")
+        self.assertEqual(ccu.source(), "# cfile.py\n")
+
+    def test_comparison(self):
+        acu = FileReporter("aa/afile.py")
+        acu2 = FileReporter("aa/afile.py")
+        zcu = FileReporter("aa/zfile.py")
+        bcu = FileReporter("aa/bb/bfile.py")
+        assert acu == acu2 and acu <= acu2 and acu >= acu2
+        assert acu < zcu and acu <= zcu and acu != zcu
+        assert zcu > acu and zcu >= acu and zcu != acu
+        assert acu < bcu and acu <= bcu and acu != bcu
+        assert bcu > acu and bcu >= acu and bcu != acu
+
+    def test_egg(self):
+        # Test that we can get files out of eggs, and read their source files.
+        # The egg1 module is installed by an action in igor.py.
+        import egg1
+        import egg1.egg1
+
+        # Verify that we really imported from an egg.  If we did, then the
+        # __file__ won't be an actual file, because one of the "directories"
+        # in the path is actually the .egg zip file.
+        self.assert_doesnt_exist(egg1.__file__)
+
+        ecu = PythonFileReporter(egg1)
+        eecu = PythonFileReporter(egg1.egg1)
+        self.assertEqual(ecu.source(), u"")
+        self.assertEqual(eecu.source().split("\n")[0], u"# My egg file!")