From 37c056db7daa0d87b42052dcaa9246bd2d5046dd Mon Sep 17 00:00:00 2001
From: Nick Hall <nick__hall@hotmail.com>
Date: Wed, 31 Jul 2013 18:21:42 +0000
Subject: [PATCH] Add command to run unit tests from setup.py

svn: r22787
---
 setup.py | 21 +++++++++++++++++++--
 1 file changed, 19 insertions(+), 2 deletions(-)

diff --git a/setup.py b/setup.py
index ca08f5cb1..9c5d062a1 100644
--- a/setup.py
+++ b/setup.py
@@ -35,7 +35,7 @@ if (sys.version_info < (2, 7) or ( (3,0) <= sys.version_info < (3, 2))):
     raise SystemExit("""Gramps requires Python 2.7 or later, or Python 3.2 or later.""")
 
 from distutils import log
-from distutils.core import setup
+from distutils.core import setup, Command
 from distutils.util import convert_path, newer
 from distutils.command.build import build as _build
 from distutils.command.install import install as _install
@@ -48,6 +48,7 @@ if sys.version_info[0] < 3:
 from stat import ST_MODE
 import io
 from gramps.version import VERSION
+import unittest
 
 ALL_LINGUAS = ('ar', 'bg', 'ca', 'cs', 'da', 'de', 'el', 'en_GB', 'es', 'fi', 'fr',
                'he', 'hr', 'hu', 'it', 'ja', 'lt', 'nb', 'nl', 'nn', 'pl', 'pt_BR',
@@ -255,6 +256,22 @@ class install(_install):
 
         os.remove(resource_file)
 
+class test(Command):
+    """Command to run Gramps unit tests"""
+    description = "run all unit tests"
+    user_options = []
+
+    def initialize_options(self):
+        pass
+
+    def finalize_options(self):
+        pass
+
+    def run(self):
+        os.environ['GRAMPS_RESOURCES'] = '.'
+        all_tests = unittest.TestLoader().discover('.', pattern='*_test.py')
+        unittest.TextTestRunner(verbosity=self.verbose).run(all_tests)
+
 #-------------------------------------------------------------------------
 #
 # Packages
@@ -448,7 +465,7 @@ setup(name = 'gramps',
       url = 'http://gramps-project.org', 
       license = 'GPL v2 or greater', 
       platforms = ['FreeBSD', 'Linux', 'MacOS', 'Windows'],
-      cmdclass = {'build': build, 'install': install},
+      cmdclass = {'build': build, 'install': install, 'test': test},
       packages = packages,
       package_data = {'gramps': package_data},
       data_files = data_files,