blob: 5d4825673ab43be29c6d7b7331e818f50bf8295b (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
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))
|
