tests: Add unit tests for 'virt_bootstrap' module

1 files changed, 39 insertions, 0 deletions

diff --git a/tests/__init__.py b/tests/__init__.py
index e69de29..e82c6d5 100644
--- a/tests/__init__.py
+++ b/tests/__init__.py
@@ -0,0 +1,39 @@
+"""
+    Test suite for virt-bootstrap
+
+    Authors: Radostin Stoyanov <rstoyanov1@gmail.com>
+
+    Copyright (C) 2017 Radostin Stoyanov
+
+    This program is free software: you can redistribute it and/or modify
+    it under the terms of the GNU General Public License as published by
+    the Free Software Foundation, either version 3 of the License, or
+    (at your option) any later version.
+
+    This program is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    GNU General Public License for more details.
+
+    You should have received a copy of the GNU General Public License
+    along with this program.  If not, see <http://www.gnu.org/licenses/>.
+"""
+
+import sys
+import unittest
+
+try:
+    import mock
+except ImportError:
+    import unittest.mock as mock
+
+sys.path += '../src'  # noqa: E402
+
+# pylint: disable=import-error
+from virtBootstrap import virt_bootstrap
+from virtBootstrap import sources
+from virtBootstrap import progress
+from virtBootstrap import utils
+
+__all__ = ['unittest', 'mock',
+           'virt_bootstrap', 'sources', 'progress', 'utils']