• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1# Copyright 2018 The Chromium OS Authors. All rights reserved.
2# Use of this source code is governed by a BSD-style license that can be
3# found in the LICENSE file.
4
5"""Test the install module."""
6
7import mock
8import unittest
9
10import common
11from autotest_lib.site_utils.deployment import install
12from autotest_lib.site_utils.stable_images import build_data
13
14
15class AFEMock(object):
16    """Mock frontend.AFE."""
17    CROS_IMAGE_TYPE = 'cros'
18    FIRMWARE_IMAGE_TYPE = 'firmware'
19
20    def __init__(self, cros_version=None, fw_version=None):
21        self.cros_version_map = mock.Mock()
22        self.cros_version_map.get_version.return_value = cros_version
23        self.fw_version_map = mock.Mock()
24        self.fw_version_map.get_version.return_value = fw_version
25
26    def get_stable_version_map(self, image_type):
27        if image_type == self.CROS_IMAGE_TYPE:
28            return self.cros_version_map
29        elif image_type == self.FIRMWARE_IMAGE_TYPE:
30             return self.fw_version_map
31
32
33class UpdateBuildTests(unittest.TestCase):
34    """Tests for _update_build."""
35
36    OMAHA_VERSION = 'R64-10176.65.0'
37    CROS_VERSION = 'R64-10175.65.0'
38
39    WOLF_BOARD = 'wolf'
40    WOLF_FW_VERSION = 'Google_Wolf.4389.24.62'
41    WOLF_NEW_FW_VERSION = 'Google_Wolf.4390.24.62'
42    WOLF_FW_VERSION_MAP = {WOLF_BOARD: WOLF_NEW_FW_VERSION}
43
44    CORAL_BOARD = 'coral'
45    CORAL_FW_VERSION = 'Google_Coral.10068.37.0'
46    CORAL_FW_VERSION_MAP = {
47            'blue': 'Google_Coral.10068.39.0',
48            'robo360': 'Google_Coral.10068.34.0',
49            'porbeagle': None,
50    }
51
52    def setUp(self):
53        self.report_log_mock = mock.Mock()
54        self.patchers = []
55
56    def _set_patchers(self,
57                      omaha_version=OMAHA_VERSION,
58                      firmware_versions=WOLF_FW_VERSION_MAP):
59        patcher1 = mock.patch.object(
60                install, '_get_omaha_build',
61                return_value=omaha_version)
62        patcher2 = mock.patch.object(
63                build_data, 'get_firmware_versions',
64                return_value=firmware_versions)
65
66        self.patchers.extend([patcher1, patcher2])
67
68        for p in self.patchers:
69            p.start()
70
71    def tearDown(self):
72        for p in self.patchers:
73            p.stop()
74
75    def test_update_build_cros_and_fw_version_on_non_unibuild(self):
76        """Update non-unibuild with old cros_version and fw_version in AFE."""
77        afe_mock = AFEMock(
78                cros_version=self.CROS_VERSION, fw_version=self.WOLF_FW_VERSION)
79
80        self._set_patchers()
81        arguments = mock.Mock(board=self.WOLF_BOARD, dry_run=False, build=None)
82
83        afe_mock.cros_version_map.set_version.assert_not_called()
84        afe_mock.fw_version_map.set_version.assert_not_called()
85
86    def test_update_build_without_omaha_version_on_non_unibuild(self):
87        """Do not update non-unibuild as no OMAHA_VERSION found."""
88        afe_mock = AFEMock(
89                cros_version=self.CROS_VERSION, fw_version=self.WOLF_FW_VERSION)
90
91        self._set_patchers(omaha_version=None)
92        arguments = mock.Mock(board=self.WOLF_BOARD, dry_run=False, build=None)
93
94        afe_mock.cros_version_map.set_version.assert_not_called()
95        afe_mock.cros_version_map.delete_version.assert_not_called()
96        afe_mock.fw_version_map.set_version.assert_not_called()
97        afe_mock.fw_version_map.delete_version.assert_not_called()
98
99    def test_update_build_cros_on_non_unibuild(self):
100        """Update non-unibuild with old cros_version in AFE."""
101        afe_mock = AFEMock(
102                cros_version=self.CROS_VERSION, fw_version=self.WOLF_FW_VERSION)
103        self._set_patchers(
104                firmware_versions={self.WOLF_BOARD: self.WOLF_FW_VERSION})
105        arguments = mock.Mock(board=self.WOLF_BOARD, dry_run=False, build=None)
106
107        afe_mock.cros_version_map.set_version.assert_not_called()
108        afe_mock.fw_version_map.set_version.assert_not_called()
109        afe_mock.fw_version_map.delete_version.assert_not_called()
110
111    def test_update_build_none_cros_and_fw_version_on_non_unibuild(self):
112        """Update Non-unibuild with None cros_version & fw_version in AFE."""
113        afe_mock = AFEMock(cros_version=None, fw_version=None)
114        self._set_patchers()
115        arguments = mock.Mock(board=self.WOLF_BOARD, dry_run=False, build=None)
116
117        afe_mock.cros_version_map.set_version.assert_not_called()
118        afe_mock.fw_version_map.set_version.assert_not_called()
119
120    def test_update_build_cros_and_fw_version_on_unibuild(self):
121        """Update unibuild with old cros_version and fw_versions."""
122        afe_mock = AFEMock(
123                cros_version=self.CROS_VERSION,
124                fw_version=self.CORAL_FW_VERSION)
125        self._set_patchers(
126                firmware_versions=self.CORAL_FW_VERSION_MAP)
127        arguments = mock.Mock(board=self.CORAL_BOARD, dry_run=False,
128                              build=None)
129
130        afe_mock.cros_version_map.set_version.assert_not_called()
131        afe_mock.fw_version_map.set_version.assert_not_called()
132        afe_mock.fw_version_map.set_version.assert_not_called()
133        afe_mock.fw_version_map.delete_version.assert_not_called()
134
135
136if __name__ == '__main__':
137    unittest.main()
138