Add coverage option to setup.py

setup.py

@@ -1,5 +1,6 @@
 """Setup for ResurceSync library and client implementation."""
-from setuptools import setup
+from setuptools import setup, Command
+import os
 # setuptools used instead of distutils.core so that 
 # dependencies can be handled automatically
 
@@ -16,6 +17,27 @@ if match:
 else:
     raise RuntimeError("Unable to find version string in %s." % (VERSIONFILE))
 
+class Coverage(Command):
+    """Class to allow coverage run from setup."""
+
+    description = "run coverage"
+    user_options = []
+
+    def initialize_options(self):
+        """Empty initialize_options."""
+        pass
+
+    def finalize_options(self):
+        """Empty finalize_options."""
+        pass
+
+    def run(self):
+        """Run coverage program."""
+        os.system("coverage run --source=resync setup.py test")
+        os.system("coverage report")
+        os.system("coverage html")
+        print("See htmlcov/index.html for details.")
+
 setup(
     name='resync',
     version=version,
@@ -43,4 +65,7 @@ setup(
         "testfixtures"
     ],
     test_suite="tests",
+    cmdclass={
+        'coverage': Coverage,
+    },
 )