Run tests using 'setup.py test'.

1 files changed, 38 insertions, 2 deletions

diff --git a/setup.py b/setup.py
index af06fec..271d13f 100755
--- a/setup.py
+++ b/setup.py
@@ -3,7 +3,7 @@
 # (c) 2011 Michał Górny <mgorny@gentoo.org>
 # Released under the terms of the 2-clause BSD license.
 
-from distutils.core import setup
+from distutils.core import setup, Command
 
 import os.path, sys
 
@@ -13,6 +13,38 @@ try:
 except ImportError:
 	PV = 'unknown'
 
+class TestCommand(Command):
+	user_options = []
+
+	def initialize_options(self):
+		pass
+
+	def finalize_options(self):
+		pass
+
+	def run(self):
+		import unittest, doctest
+
+		import PMSTestSuite.cli
+		import PMSTestSuite.library
+		import PMSTestSuite.pm
+		import PMSTestSuite.pm.pkgcorepm
+		import PMSTestSuite.pm.portagepm
+		import PMSTestSuite.repository
+
+		tests = unittest.TestSuite()
+		tests.addTests(doctest.DocTestSuite(PMSTestSuite.cli))
+		tests.addTests(doctest.DocTestSuite(PMSTestSuite.library))
+		tests.addTests(doctest.DocTestSuite(PMSTestSuite.library.case))
+		tests.addTests(doctest.DocTestSuite(PMSTestSuite.pm))
+		tests.addTests(doctest.DocTestSuite(PMSTestSuite.pm.pkgcorepm))
+		tests.addTests(doctest.DocTestSuite(PMSTestSuite.pm.portagepm))
+		tests.addTests(doctest.DocTestSuite(PMSTestSuite.repository))
+
+		r = unittest.TextTestRunner()
+		res = r.run(tests)
+		sys.exit(0 if res.wasSuccessful() else 1)
+
 setup(
 		name = 'pms-test-suite',
 		version = PV,
@@ -43,5 +75,9 @@ setup(
 			'Topic :: Software Development :: Quality Assurance',
 			'Topic :: Software Development :: Testing',
 			'Topic :: System :: Installation/Setup'
-		]
+		],
+
+		cmdclass = {
+			'test': TestCommand
+		}
 )