[virt-tools-list] [virt-bootstrap] [PATCH v3 06/12] tests: Add test suite

[Radostin Stoyanov]

Tests can be started with: `./setup.py test`
---
 setup.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index b2e17ac..c6a35ad 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.7.1'],
 
+    tests_require=['mock>=2.0'],
+
     extras_require={
         'dev': [
             'pylint',
-- 
2.9.4