AuxdbTestCase: test eclass inheritance

Signed-off-by: Zac Medico <zmedico@gentoo.org>

2 files changed, 38 insertions, 10 deletions

diff --git a/lib/portage/tests/dbapi/test_auxdb.py b/lib/portage/tests/dbapi/test_auxdb.py
index cfcabc8bb..85d64c15e 100644
--- a/lib/portage/tests/dbapi/test_auxdb.py
+++ b/lib/portage/tests/dbapi/test_auxdb.py
@@ -34,24 +34,44 @@ class AuxdbTestCase(TestCase):
 	def _test_mod(self, auxdbmodule):
 		ebuilds = {
 			"cat/A-1": {
-				"EAPI": "7"
+				"EAPI": "7",
+				"MISC_CONTENT": "inherit foo",
 			},
 			"cat/B-1": {
-				"EAPI": "7"
+				"EAPI": "7",
+				"MISC_CONTENT": "inherit foo",
 			},
 		}
 
-		playground = ResolverPlayground(ebuilds=ebuilds,
+		ebuild_inherited = frozenset(["bar", "foo"])
+		eclass_defined_phases = "prepare"
+		eclass_depend = "bar/foo"
+
+		eclasses = {
+			"foo": (
+				"inherit bar",
+			),
+			"bar": (
+				"EXPORT_FUNCTIONS src_prepare",
+				"DEPEND=\"{}\"".format(eclass_depend),
+				"bar_src_prepare() { default; }",
+			),
+		}
+
+		playground = ResolverPlayground(ebuilds=ebuilds, eclasses=eclasses,
 			user_config={'modules': ('portdbapi.auxdbmodule = %s' % auxdbmodule,)})
 
 		portdb = playground.trees[playground.eroot]["porttree"].dbapi
 
 		loop = asyncio._wrap_loop()
-		loop.run_until_complete(self._test_mod_async(ebuilds, portdb))
+		loop.run_until_complete(self._test_mod_async(ebuilds, ebuild_inherited, eclass_defined_phases, eclass_depend, portdb))
 
 	@coroutine
-	def _test_mod_async(self, ebuilds, portdb):
+	def _test_mod_async(self, ebuilds, ebuild_inherited, eclass_defined_phases, eclass_depend, portdb):
 
 		for cpv, metadata in ebuilds.items():
-			eapi, = yield portdb.async_aux_get(cpv, ['EAPI'])
+			defined_phases, depend, eapi, inherited = yield portdb.async_aux_get(cpv, ['DEFINED_PHASES', 'DEPEND', 'EAPI', 'INHERITED'])
+			self.assertEqual(defined_phases, eclass_defined_phases)
+			self.assertEqual(depend, eclass_depend)
 			self.assertEqual(eapi, metadata['EAPI'])
+			self.assertEqual(frozenset(inherited.split()), ebuild_inherited)
diff --git a/lib/portage/tests/resolver/ResolverPlayground.py b/lib/portage/tests/resolver/ResolverPlayground.py
index 0456ce2e2..98831e000 100644
--- a/lib/portage/tests/resolver/ResolverPlayground.py
+++ b/lib/portage/tests/resolver/ResolverPlayground.py
@@ -67,7 +67,7 @@ class ResolverPlayground(object):
 """
 
 	def __init__(self, ebuilds={}, binpkgs={}, installed={}, profile={}, repo_configs={}, \
-		user_config={}, sets={}, world=[], world_sets=[], distfiles={},
+		user_config={}, sets={}, world=[], world_sets=[], distfiles={}, eclasses={},
 		eprefix=None, targetroot=False, debug=False):
 		"""
 		ebuilds: cpv -> metadata mapping simulating available ebuilds.
@@ -159,7 +159,7 @@ class ResolverPlayground(object):
 		self._create_ebuilds(ebuilds)
 		self._create_binpkgs(binpkgs)
 		self._create_installed(installed)
-		self._create_profile(ebuilds, installed, profile, repo_configs, user_config, sets)
+		self._create_profile(ebuilds, eclasses, installed, profile, repo_configs, user_config, sets)
 		self._create_world(world, world_sets)
 
 		self.settings, self.trees = self._load_config()
@@ -346,7 +346,7 @@ class ResolverPlayground(object):
 				with open(ebuild_path, 'rb') as inputfile:
 					f.write(inputfile.read())
 
-	def _create_profile(self, ebuilds, installed, profile, repo_configs, user_config, sets):
+	def _create_profile(self, ebuilds, eclasses, installed, profile, repo_configs, user_config, sets):
 
 		user_config_dir = os.path.join(self.eroot, USER_CONFIG_PATH)
 
@@ -404,7 +404,15 @@ class ResolverPlayground(object):
 							f.write("masters =\n")
 
 			#Create $profile_dir/eclass (we fail to digest the ebuilds if it's not there)
-			os.makedirs(os.path.join(repo_dir, "eclass"))
+			eclass_dir = os.path.join(repo_dir, "eclass")
+			os.makedirs(eclass_dir)
+
+			for eclass_name, eclass_content in eclasses.items():
+				with open(os.path.join(eclass_dir, "{}.eclass".format(eclass_name)), 'wt') as f:
+					if isinstance(eclass_content, basestring):
+						eclass_content = [eclass_content]
+					for line in eclass_content:
+						f.write("{}\n".format(line))
 
 			# Temporarily write empty value of masters until it becomes default.
 			if not repo_config or "layout.conf" not in repo_config: