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

["Michał Górny" <mgorny@gentoo.org>]

commit:     421452d0ae6375db6539147e9e292ec52064617f
Author:     Michał Górny <mgorny <AT> gentoo <DOT> org>
AuthorDate: Tue Jun  7 17:16:43 2011 +0000
Commit:     Michał Górny <mgorny <AT> gentoo <DOT> org>
CommitDate: Tue Jun  7 17:16:43 2011 +0000
URL:        http://git.overlays.gentoo.org/gitweb/?p=proj/pms-test-suite.git;a=commit;h=421452d0

pkgcore: support vardb lookups.

---
 PMSTestSuite/pm/pkgcorepm.py |   21 ++++++++++++++++++---
 1 files changed, 18 insertions(+), 3 deletions(-)

diff --git a/PMSTestSuite/pm/pkgcorepm.py b/PMSTestSuite/pm/pkgcorepm.py
index 738c49b..0d64cd7 100644
--- a/PMSTestSuite/pm/pkgcorepm.py
+++ b/PMSTestSuite/pm/pkgcorepm.py
@@ -63,13 +63,24 @@ class PkgCorePM(PortagePM):
 		raise NotImplementedError('PkgCorePM does not support adding repositories yet.')
 
 	_config = None
+	_vardb = None
+	def _load_config(self):
+		from pkgcore.config import load_config
+		self._config = load_config()
+		self._vardb = self._config.repo['vdb']
+
 	@property
 	def config(self):
 		if self._config is None:
-			from pkgcore.config import load_config
-			self._config = load_config()
+			self._load_config()
 		return self._config
 
+	@property
+	def vardb(self):
+		# reload required to refresh vardb
+		self._load_config()
+		return self._vardb
+
 	def get_repository(self, name):
 		for r in self.config.repo.values():
 			if hasattr(r, 'repo_id') and r.repo_id == name:
@@ -77,5 +88,9 @@ class PkgCorePM(PortagePM):
 		else:
 			raise KeyError('Repository %s not found.' % name)
 
-	# XXX: vardb and stuff
+	def lookup_vardb(self, cpvs):
+		installed_cpvs = set([p.cpvstr for p in self.vardb])
+		installed_cpvs &= set(cpvs)
+		return installed_cpvs
+
 	# portage's get_userpriv_group() required by repoman