Add a test to check what portage submodules are imported by 'import portage', and assert that it doesn not rise above the current baseline.

1 files changed, 56 insertions, 0 deletions

diff --git a/pym/portage/tests/lazyimport/test_lazy_import_portage_baseline.py b/pym/portage/tests/lazyimport/test_lazy_import_portage_baseline.py
new file mode 100644
index 000000000..d96f83a9b
--- /dev/null
+++ b/pym/portage/tests/lazyimport/test_lazy_import_portage_baseline.py
@@ -0,0 +1,56 @@
+# Copyright 2010 Gentoo Foundation
+# Distributed under the terms of the GNU General Public License v2
+
+import re
+import portage
+from portage import os
+from portage.tests import TestCase
+
+from _emerge.TaskScheduler import TaskScheduler
+from _emerge.PipeReader import PipeReader
+from _emerge.SpawnProcess import SpawnProcess
+
+class LazyImportPortageBaselineTestCase(TestCase):
+
+	_module_re = re.compile(r'^(portage|repoman|_emerge)\.')
+
+	_baseline_imports = frozenset([
+		'portage.const', 'portage.env', 'portage.localization',
+		'portage.proxy', 'portage.proxy.lazyimport',
+		'portage.proxy.objectproxy', 'portage._ensure_encodings',
+	])
+
+	_baseline_import_cmd = [portage._python_interpreter, '-c',
+		'import portage, sys ; ' + \
+		'sys.stdout.write(" ".join(k for k in sys.modules ' + \
+		'if sys.modules[k] is not None))']
+
+	def testLazyImportPortageBaseline(self):
+		"""
+		Check what modules are imported by a baseline module import.
+		"""
+
+		task_scheduler = TaskScheduler()
+		master_fd, slave_fd = os.pipe()
+		master_file = os.fdopen(master_fd, 'rb')
+		slave_file = os.fdopen(slave_fd, 'wb')
+		producer = SpawnProcess(
+			args=self._baseline_import_cmd,
+			env=os.environ, fd_pipes={1:slave_fd},
+			scheduler=task_scheduler.sched_iface)
+		producer.start()
+		slave_file.close()
+
+		consumer = PipeReader(
+			input_files={"producer" : master_file},
+			scheduler=task_scheduler.sched_iface)
+
+		task_scheduler.add(consumer)
+		task_scheduler.run()
+		output = consumer.getvalue().decode('ascii', 'replace').split()
+
+		unexpected_modules = " ".join(sorted(x for x in output \
+			if self._module_re.match(x) is not None and \
+			x not in self._baseline_imports))
+
+		self.assertEqual("", unexpected_modules)