From ee0eee05325308548e1e10a795be9fa74ede6044 Mon Sep 17 00:00:00 2001 From: Alex Willmer Date: Mon, 17 Nov 2025 08:30:02 +0000 Subject: [PATCH] tests: Add test for mitogen.ocre.Importer.MITOGEN_PKG_CONTENT --- tests/importer_test.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/tests/importer_test.py b/tests/importer_test.py index 33aafe93..7894181c 100644 --- a/tests/importer_test.py +++ b/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