tests: Add test for mitogen.ocre.Importer.MITOGEN_PKG_CONTENT

tests/importer_test.py

@@ -1,3 +1,5 @@
+import pkgutil
+import os
 import sys
 import threading
 import types
@@ -9,6 +11,7 @@ try:
 except ImportError:
     import mock
 
+import mitogen
 import mitogen.core
 import mitogen.utils
 from mitogen.core import b
@@ -46,6 +49,15 @@ class ImporterMixin(testlib.RouterMixin):
         super(ImporterMixin, self).tearDown()
 
 
+class MitogenPkgContentTest(testlib.TestCase):
+    def test_matches_filesystem(self):
+        mitogen_pkg_dir = os.path.dirname(mitogen.__file__)
+        self.assertEqual(
+            mitogen.core.Importer.MITOGEN_PKG_CONTENT,
+            sorted(mod.name for mod in pkgutil.iter_modules([mitogen_pkg_dir])),
+        )
+
+
 class InvalidNameTest(ImporterMixin, testlib.TestCase):
     modname = 'trailingdot.'
     # 0:fullname 1:pkg_present 2:path 3:compressed 4:related