diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/fetchimg_test.py | 14 |
1 files changed, 13 insertions, 1 deletions
diff --git a/tests/fetchimg_test.py b/tests/fetchimg_test.py index 683c9701..5d482567 100644 --- a/tests/fetchimg_test.py +++ b/tests/fetchimg_test.py @@ -17,6 +17,18 @@ class TestClass: 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, 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)) |