Working toward reading source from eggs, but this isn't right on Py3k yet.

4 files changed, 19 insertions, 0 deletions

diff --git a/test/eggsrc/egg1/__init__.py b/test/eggsrc/egg1/__init__.py
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/test/eggsrc/egg1/__init__.py
diff --git a/test/eggsrc/egg1/egg1.py b/test/eggsrc/egg1/egg1.py
new file mode 100644
index 0000000..3fadde3
--- /dev/null
+++ b/test/eggsrc/egg1/egg1.py
@@ -0,0 +1,4 @@
+# My egg file!
+
+walrus = "Eggman"
+says = "coo-coo cachoo"
diff --git a/test/eggsrc/setup.py b/test/eggsrc/setup.py
new file mode 100644
index 0000000..6a88a58
--- /dev/null
+++ b/test/eggsrc/setup.py
@@ -0,0 +1,6 @@
+from setuptools import setup
+
+setup(
+    name="covtestegg1",
+    packages=['egg1'],
+    )
diff --git a/test/test_codeunit.py b/test/test_codeunit.py
index 36ca1fc..0c2fc98 100644
--- a/test/test_codeunit.py
+++ b/test/test_codeunit.py
@@ -71,3 +71,12 @@ class CodeUnitTest(CoverageTest):
         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):
+        import egg1, egg1.egg1
+        cu = code_unit_factory([egg1, egg1.egg1], FileLocator())
+        self.assertEqual(cu[0].source_file().read(), "")
+        self.assertEqual(cu[1].source_file().read().split("\n")[0],
+                "# My egg file!"
+                )
+        
\ No newline at end of file