1 files changed, 27 insertions, 4 deletions
diff --git a/setup.py b/setup.py
index ea91a0a..a9c1eac 100644
--- a/setup.py
+++ b/setup.py
@@ -1,10 +1,9 @@
 #!/usr/bin/env python
 # -*- coding: utf-8 -*-
 
-import os
 import sys
-
 from setuptools import setup
+from setuptools.command.test import test as TestCommand
 import versioneer
 
 
@@ -12,19 +11,43 @@ required = []
 
 PROJECT = 'obsapi'
 
+versioneer.VCS = 'git'
 versioneer.versionfile_source = '%s/_version.py' % PROJECT
 versioneer.versionfile_build = '%s/_version.py' % PROJECT
 versioneer.tag_prefix = '%s-' % PROJECT  # tags are like 1.2.0
 versioneer.parentdir_prefix = '%s-' % PROJECT  # dirname like 'myproject-1.2.0'
 
+
+class PyTest(TestCommand):
+    user_options = [('pytest-args=', 'a', "Arguments to pass to py.test")]
+
+    def initialize_options(self):
+        TestCommand.initialize_options(self)
+        self.pytest_args = []
+
+    def finalize_options(self):
+        TestCommand.finalize_options(self)
+        self.test_args = []
+        self.test_suite = True
+
+    def run_tests(self):
+        # import here, cause outside the eggs aren't loaded
+        import pytest
+        errno = pytest.main(self.pytest_args)
+        sys.exit(errno)
+
+cmdclass = versioneer.get_cmdclass()
+cmdclass['test'] = PyTest
+
 setup(
     name=PROJECT,
     version=versioneer.get_version(),
-    cmdclass=versioneer.get_cmdclass(),
+    tests_require=['pytest'],
+    cmdclass=cmdclass,
     description='Python library for accessing the Open Build Service api',
     author='Scott Bahling',
     author_email='sbahling@suse.de',
-    packages= [
+    packages=[
         'obsapi',
     ],
     install_requires=required,