1 files changed, 34 insertions, 0 deletions
diff --git a/tests/fetchimg_test.py b/tests/fetchimg_test.py
new file mode 100644
index 00000000..5d482567
--- /dev/null
+++ b/tests/fetchimg_test.py
@@ -0,0 +1,34 @@
+import mock
+from func.fetchimg import FetchImg
+
+
+class TestClass:
+    @mock.patch('func.fetchimg.os')
+    @mock.patch('func.fetchimg.os.path')
+    def test_fetch_img_success(self, mock_path, mock_os):
+        mock_os.system.return_value = True
+        mock_path.isfile.return_value = True
+        img = FetchImg()
+        img.download()
+
+    @mock.patch('func.fetchimg.time')
+    @mock.patch('func.fetchimg.os.system')
+    @mock.patch('func.fetchimg.os.path')
+    def test_fetch_img_fail(self, mock_path, mock_system, mock_time):
+        img = FetchImg()
+        mock_system.return_value = True
+        mock_path.isfile.side_effect = [False, False, True]
+        img.download()
+        assert mock_time.sleep.call_count == 2
+
+    @mock.patch('func.fetchimg.time')
+    @mock.patch('func.fetchimg.os.system')
+    @mock.patch('func.fetchimg.os.path')
+    def test_fetch_temp_success(self, mock_path, mock_system, mock_time):
+        img = FetchImg()
+        mock_system.return_value = True
+        mock_path.isfile.side_effect = [True]
+        img.download()
+        filepath = './Temp_Img/QTIP_CentOS.qcow2'
+        imgstorepath = "/home/opnfv/imgstore/QTIP_CentOS.qcow2"
+        mock_system.assert_called_with("ln -s %s %s" % (imgstorepath, filepath))