[gentoo-commits] proj/pms-test-suite:master commit in: PMSTestSuite/pm/

[Michał Górny]

commit:     5a7d32196d0045b90835ef80a2fecb78125d3c39
Author:     Michał Górny <mgorny <AT> gentoo <DOT> org>
AuthorDate: Tue May 31 11:40:05 2011 +0000
Commit:     Michał Górny <mgorny <AT> gentoo <DOT> org>
CommitDate: Tue May 31 11:40:05 2011 +0000
URL:        http://git.overlays.gentoo.org/gitweb/?p=proj/pms-test-suite.git;a=commit;h=5a7d3219

Support vdb lookups.

---
 PMSTestSuite/pm/__init__.py  |   10 ++++++++++
 PMSTestSuite/pm/portagepm.py |   18 ++++++++++++++++++
 2 files changed, 28 insertions(+), 0 deletions(-)

diff --git a/PMSTestSuite/pm/__init__.py b/PMSTestSuite/pm/__init__.py
index d3bd434..9838822 100644
--- a/PMSTestSuite/pm/__init__.py
+++ b/PMSTestSuite/pm/__init__.py
@@ -71,6 +71,16 @@ class PackageManager(object):
 		"""
 		raise NotImplementedError('Please override the unmerge() method.')
 
+	def lookup_vardb(self, cpvs):
+		"""
+		Lookup the vardb checking whether the packages listed in <cpvs>
+		are installed. The elements of iterable passed as <cpvs> shall be in CPV
+		form, i.e. category/pn-pv.
+
+		Returns an iterable of packages actually merged (a subset of <cpvs>).
+		"""
+		raise NotImplementedError('Please override the lookup_vardb() method.')
+
 def get_package_managers():
 	""" Return the list of supported Package Managers. """
 

diff --git a/PMSTestSuite/pm/portagepm.py b/PMSTestSuite/pm/portagepm.py
index 71672c0..9d5fa0b 100644
--- a/PMSTestSuite/pm/portagepm.py
+++ b/PMSTestSuite/pm/portagepm.py
@@ -44,3 +44,21 @@ class PortagePM(PackageManager):
 		ret = subprocess.call([self.emerge_path, '--unmerge'] + self.pm_options + atoms,
 				env = {'PORTDIR_OVERLAY': ' '.join(self.repo_paths)})
 		return ret == 0
+
+	_vardb = None
+	@property
+	def vardb(self):
+		if not self._vardb:
+			from portage import create_trees
+			trees = create_trees(
+					config_root = os.environ.get('PORTAGE_CONFIGROOT'),
+					target_root = os.environ.get('ROOT'))
+			tree = trees[max(trees)]
+			self._vardb = tree['vartree'].dbapi
+
+		return self._vardb
+
+	def lookup_vardb(self, cpvs):
+		installed_cpvs = set(self.vardb.cpv_all())
+		installed_cpvs &= set(cpvs)
+		return installed_cpvs