From 2b9dbe2547935f96b165e7f7d67e1af481bada40 Mon Sep 17 00:00:00 2001
From: Radostin Stoyanov <rstoyanov1@gmail.com>
Date: Mon, 24 Jul 2017 09:14:02 +0100
Subject: tests: Add test suite

---
 setup.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

(limited to 'setup.py')

diff --git a/setup.py b/setup.py
index 36de2d5..2f299b6 100755
--- a/setup.py
+++ b/setup.py
@@ -50,7 +50,7 @@ class CheckPylint(Command):
         """
 
         res = 0
-        files = ' '.join(["setup.py", "src/virtBootstrap/*.py"])
+        files = ' '.join(["setup.py", "src/virtBootstrap/*.py", "tests/*.py"])
         output_format = "colorized" if sys.stdout.isatty() else "text"
 
         print(">>> Running pycodestyle ...")
@@ -82,6 +82,7 @@ setup(
     keywords='virtualization container rootfs',
     package_dir={"": "src"},
     packages=['virtBootstrap'],
+    test_suite='tests',
     entry_points={
         'console_scripts': [
             'virt-bootstrap=virtBootstrap.virt_bootstrap:main',
@@ -117,6 +118,8 @@ setup(
     # root password for root file system.
     install_requires=['passlib>=1.6.1'],
 
+    tests_require=['mock>=2.0'],
+
     extras_require={
         'dev': [
             'pylint',
-- 
cgit