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
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
|
#!/usr/bin/python2
#
# Copyright 2014 Google Inc. All Rights Reserved
"""Download image unittest."""
from __future__ import print_function
import os
import mock
import unittest
import download_images
from cros_utils import command_executer
from cros_utils import logger
import test_flag
MOCK_LOGGER = logger.GetLogger(log_dir='', mock=True)
class ImageDownloaderTestcast(unittest.TestCase):
"""The image downloader test class."""
def __init__(self, *args, **kwargs):
super(ImageDownloaderTestcast, self).__init__(*args, **kwargs)
self.called_download_image = False
self.called_uncompress_image = False
self.called_get_build_id = False
@mock.patch.object(os, 'makedirs')
@mock.patch.object(os.path, 'exists')
def test_download_image(self, mock_path_exists, mock_mkdirs):
# Set mock and test values.
mock_cmd_exec = mock.Mock(spec=command_executer.CommandExecuter)
test_chroot = '/usr/local/home/chromeos'
test_build_id = 'lumpy-release/R36-5814.0.0'
image_path = ('gs://chromeos-image-archive/%s/chromiumos_test_image.tar.xz'
% test_build_id)
downloader = download_images.ImageDownloader(logger_to_use=MOCK_LOGGER,
cmd_exec=mock_cmd_exec)
# Set os.path.exists to always return False and run downloader
mock_path_exists.return_value = False
test_flag.SetTestMode(True)
downloader.DownloadImage(test_chroot, test_build_id, image_path)
# Verify os.path.exists was called twice, with proper arguments.
self.assertEqual(mock_path_exists.call_count, 2)
mock_path_exists.assert_called_with(
'/usr/local/home/chromeos/chroot/tmp/lumpy-release/'
'R36-5814.0.0/chromiumos_test_image.bin')
mock_path_exists.assert_any_call(
'/usr/local/home/chromeos/chroot/tmp/lumpy-release/R36-5814.0.0')
# Verify we called os.mkdirs
self.assertEqual(mock_mkdirs.call_count, 1)
mock_mkdirs.assert_called_with(
'/usr/local/home/chromeos/chroot/tmp/lumpy-release/R36-5814.0.0')
# Verify we called ChrootRunCommand once, with proper arguments.
self.assertEqual(mock_cmd_exec.ChrootRunCommand.call_count, 1)
mock_cmd_exec.ChrootRunCommand.assert_called_with(
'/usr/local/home/chromeos',
'gsutil cp '
'gs://chromeos-image-archive/lumpy-release/R36-5814.0.0/'
'chromiumos_test_image.tar.xz'
' /tmp/lumpy-release/R36-5814.0.0')
# Reset the velues in the mocks; set os.path.exists to always return True.
mock_path_exists.reset_mock()
mock_cmd_exec.reset_mock()
mock_path_exists.return_value = True
# Run downloader
downloader.DownloadImage(test_chroot, test_build_id, image_path)
# Verify os.path.exists was called twice, with proper arguments.
self.assertEqual(mock_path_exists.call_count, 2)
mock_path_exists.assert_called_with(
'/usr/local/home/chromeos/chroot/tmp/lumpy-release/'
'R36-5814.0.0/chromiumos_test_image.bin')
mock_path_exists.assert_any_call(
'/usr/local/home/chromeos/chroot/tmp/lumpy-release/R36-5814.0.0')
# Verify we made no RunCommand or ChrootRunCommand calls (since
# os.path.exists returned True, there was no work do be done).
self.assertEqual(mock_cmd_exec.RunCommand.call_count, 0)
self.assertEqual(mock_cmd_exec.ChrootRunCommand.call_count, 0)
@mock.patch.object(os.path, 'exists')
def test_uncompress_image(self, mock_path_exists):
# set mock and test values.
mock_cmd_exec = mock.Mock(spec=command_executer.CommandExecuter)
test_chroot = '/usr/local/home/chromeos'
test_build_id = 'lumpy-release/R36-5814.0.0'
downloader = download_images.ImageDownloader(logger_to_use=MOCK_LOGGER,
cmd_exec=mock_cmd_exec)
# Set os.path.exists to always return False and run uncompress.
mock_path_exists.return_value = False
downloader.UncompressImage(test_chroot, test_build_id)
# Verify os.path.exists was called once, with correct arguments.
self.assertEqual(mock_path_exists.call_count, 1)
mock_path_exists.assert_called_with(
'/usr/local/home/chromeos/chroot/tmp/lumpy-release/'
'R36-5814.0.0/chromiumos_test_image.bin')
# Verify ChrootRunCommand was called, with correct arguments.
self.assertEqual(mock_cmd_exec.ChrootRunCommand.call_count, 1)
mock_cmd_exec.ChrootRunCommand.assert_called_with(
'/usr/local/home/chromeos',
'cd /tmp/lumpy-release/R36-5814.0.0 ;unxz '
'chromiumos_test_image.tar.xz; tar -xvf chromiumos_test_image.tar')
# Set os.path.exists to always return False and run uncompress.
mock_path_exists.reset_mock()
mock_cmd_exec.reset_mock()
mock_path_exists.return_value = True
downloader.UncompressImage(test_chroot, test_build_id)
# Verify os.path.exists was called once, with correct arguments.
self.assertEqual(mock_path_exists.call_count, 1)
mock_path_exists.assert_called_with(
'/usr/local/home/chromeos/chroot/tmp/lumpy-release/'
'R36-5814.0.0/chromiumos_test_image.bin')
# Verify ChrootRunCommand was not called.
self.assertEqual(mock_cmd_exec.ChrootRunCommand.call_count, 0)
def test_run(self):
# Set test arguments
test_chroot = '/usr/local/home/chromeos'
test_build_id = 'remote/lumpy/latest-dev'
# Set values to test/check.
self.called_download_image = False
self.called_uncompress_image = False
self.called_get_build_id = False
# Define fake stub functions for Run to call
def FakeGetBuildID(unused_root, unused_xbuddy_label):
self.called_get_build_id = True
return 'lumpy-release/R36-5814.0.0'
def GoodDownloadImage(root, build_id, image_path):
if root or build_id or image_path:
pass
self.called_download_image = True
return 'chromiumos_test_image.bin'
def BadDownloadImage(root, build_id, image_path):
if root or build_id or image_path:
pass
self.called_download_image = True
return None
def FakeUncompressImage(root, build_id):
if root or build_id:
pass
self.called_uncompress_image = True
return 0
# Initialize downloader
downloader = download_images.ImageDownloader(logger_to_use=MOCK_LOGGER)
# Set downloader to call fake stubs.
downloader.GetBuildID = FakeGetBuildID
downloader.UncompressImage = FakeUncompressImage
downloader.DownloadImage = GoodDownloadImage
# Call Run.
downloader.Run(test_chroot, test_build_id)
# Make sure it called both _DownloadImage and _UncompressImage
self.assertTrue(self.called_download_image)
self.assertTrue(self.called_uncompress_image)
# Reset values; Now use fake stub that simulates DownloadImage failing.
self.called_download_image = False
self.called_uncompress_image = False
downloader.DownloadImage = BadDownloadImage
# Call Run again.
downloader.Run(test_chroot, test_build_id)
# Verify that UncompressImage was not called, since _DownloadImage "failed"
self.assertTrue(self.called_download_image)
self.assertFalse(self.called_uncompress_image)
if __name__ == '__main__':
unittest.main()
|