[gentoo-commits] proj/gentoopm:master commit in: /, gentoopm/tests/

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

commit:     b751801e385a079c1b59eca10741497e4737d742
Author:     Michał Górny <mgorny <AT> gentoo <DOT> org>
AuthorDate: Fri Jul 15 09:48:43 2011 +0000
Commit:     Michał Górny <mgorny <AT> gentoo <DOT> org>
CommitDate: Fri Jul 15 09:48:43 2011 +0000
URL:        http://git.overlays.gentoo.org/gitweb/?p=proj/gentoopm.git;a=commit;h=b751801e

Add tests for package sets.

---
 gentoopm/tests/__init__.py |    3 +
 gentoopm/tests/pkg.py      |  113 ++++++++++++++++++++++++++++++++++++++++++++
 setup.py                   |    1 +
 3 files changed, 117 insertions(+), 0 deletions(-)

diff --git a/gentoopm/tests/__init__.py b/gentoopm/tests/__init__.py
index eced9c1..7b52fae 100644
--- a/gentoopm/tests/__init__.py
+++ b/gentoopm/tests/__init__.py
@@ -52,5 +52,8 @@ class PackageNames(object):
 	multiple = 'pms'
 	""" Incomplete atom matching multiple packages. """
 
+	empty = 'example/example'
+	""" Atom matching no packages. """
+
 	repository = 'gentoo'
 	""" Repository name guaranteed to match. """

diff --git a/gentoopm/tests/pkg.py b/gentoopm/tests/pkg.py
new file mode 100644
index 0000000..47ca02e
--- /dev/null
+++ b/gentoopm/tests/pkg.py
@@ -0,0 +1,113 @@
+#!/usr/bin/python
+#	vim:fileencoding=utf-8
+# (c) 2011 Michał Górny <mgorny@gentoo.org>
+# Released under the terms of the 2-clause BSD license.
+
+from gentoopm.exceptions import AmbiguousPackageSetError, EmptyPackageSetError
+from gentoopm.tests import PMTestCase, PackageNames
+
+class IterChecker(object):
+	def __init__(self, iterable):
+		self._it = iter(iterable)
+
+	def __nonzero__(self):
+		try:
+			next(self._it)
+		except StopIteration:
+			return False
+		else:
+			return True
+
+class PackageSetsTestCase(PMTestCase):
+	def setUp(self):
+		self._inst = self.pm.installed
+		self._stack = self.pm.stack
+		self._repo = self.pm.repositories[PackageNames.repository]
+		self._repos = (self._inst, self._stack, self._repo)
+
+	def test_filter_atom(self):
+		at = PackageNames.single_complete
+		for r in self._repos:
+			self.assertTrue(IterChecker(r.filter(at)))
+
+	def test_filter_atom_incomplete(self):
+		at = PackageNames.single
+		for r in self._repos:
+			self.assertTrue(IterChecker(r.filter(at)))
+
+	def test_filter_atom_multiple(self):
+		""" Check whether multi-matching atoms return multiple packages. """
+		at = PackageNames.multiple
+		for r in self._repos:
+			keys = set()
+			for p in r.filter(at):
+				keys.add(p.key)
+			self.assertTrue(len(keys) > 1)
+
+	def test_filter_atom_empty(self):
+		at = PackageNames.empty
+		for r in self._repos:
+			self.assertFalse(IterChecker(r.filter(at)))
+
+	def test_select_atom(self):
+		at = PackageNames.single_complete
+		for r in self._repos:
+			self.assertTrue(r.select(at))
+
+	def test_select_atom_incomplete(self):
+		at = PackageNames.single
+		for r in self._repos:
+			self.assertTrue(r.select(at))
+
+	def test_select_atom_multiple(self):
+		at = PackageNames.multiple
+		for r in self._repos:
+			self.assertRaises(AmbiguousPackageSetError, r.select, at)
+
+	def test_select_atom_empty(self):
+		at = PackageNames.empty
+		for r in self._repos:
+			self.assertRaises(EmptyPackageSetError, r.select, at)
+
+	def test_getitem_atom(self):
+		at = PackageNames.single_complete
+		for r in (self._stack, self._repo):
+			self.assertRaises(AmbiguousPackageSetError, lambda r, at: r[at], r, at)
+		self.assertTrue(self._inst[at])
+
+	def test_getitem_atom_empty(self):
+		at = PackageNames.empty
+		for r in self._repos:
+			self.assertRaises(EmptyPackageSetError, lambda r, at: r[at], r, at)
+
+	def test_nonzero_true(self):
+		at = PackageNames.single_complete
+		for r in self._repos:
+			pset = r.filter(at)
+			self.assertTrue(IterChecker(pset))
+			self.assertTrue(pset)
+
+	def test_nonzero_false(self):
+		at = PackageNames.empty
+		for r in self._repos:
+			pset = r.filter(at)
+			self.assertFalse(IterChecker(pset))
+			self.assertFalse(pset)
+
+	def test_contains_atom(self):
+		at = PackageNames.single_complete
+		for r in self._repos:
+			self.assertTrue(at in r)
+
+	def test_contains_atom_multiple(self):
+		at = PackageNames.multiple
+		for r in self._repos:
+			self.assertTrue(at in r)
+
+	def test_contains_atom_empty(self):
+		at = PackageNames.empty
+		for r in self._repos:
+			self.assertFalse(at in r)
+
+	def tearDown(self):
+		pass

diff --git a/setup.py b/setup.py
index 6c5a481..4318a8e 100755
--- a/setup.py
+++ b/setup.py
@@ -55,6 +55,7 @@ class TestCommand(Command):
 				testsuite = unittest.TestSuite()
 				testsuite.addTests(l.loadTestsFromModule('gentoopm.tests.atom'))
 				testsuite.addTests(l.loadTestsFromModule('gentoopm.tests.config'))
+				testsuite.addTests(l.loadTestsFromModule('gentoopm.tests.pkg'))
 				testsuite.addTests(l.loadTestsFromModule('gentoopm.tests.repo'))
 				maintestsuite.addTests(testsuite)