| Tatsuhisa Yamaguchi | d30a969 | 2023-08-17 19:48:43 +0900 | [diff] [blame] | 1 | # Copyright 2023 The ChromiumOS Authors |
| 2 | # Use of this source code is governed by a BSD-style license that can be |
| 3 | # found in the LICENSE file. |
| 4 | |
| 5 | """Unittests for ChromeLkgm operations.""" |
| 6 | |
| 7 | from chromite.api import api_config |
| 8 | from chromite.api.controller import chrome_lkgm |
| 9 | from chromite.api.gen.chromite.api import chrome_lkgm_pb2 |
| 10 | from chromite.lib import chrome_lkgm as chrome_lkgm_lib |
| 11 | from chromite.lib import cros_test_lib |
| 12 | |
| 13 | |
| 14 | class FindLkgmTest(cros_test_lib.MockTestCase, api_config.ApiConfigMixin): |
| 15 | """Unittests for FindLkgm.""" |
| 16 | |
| 17 | LKGM_VERSION = "123.0.0.4566" |
| 18 | FALLBACK_VERSION = "123.0.0.4562" |
| 19 | |
| 20 | def setUp(self): |
| 21 | self.request = chrome_lkgm_pb2.FindLkgmRequest() |
| 22 | self.request.build_target.name = "newboard" |
| 23 | self.request.chrome_src = "/home/user/chromium/src" |
| 24 | self.request.fallback_versions = 10 |
| 25 | self.response = chrome_lkgm_pb2.FindLkgmResponse() |
| 26 | self.finder_mock = self.PatchObject( |
| 27 | chrome_lkgm_lib, "ChromeOSVersionFinder" |
| 28 | ) |
| 29 | self.instance = self.finder_mock.return_value |
| 30 | self.instance.config_name = f"{self.request.build_target.name}/release" |
| 31 | self.get_full_version_mock = self.PatchObject( |
| 32 | self.instance, |
| 33 | "GetFullVersionFromLatest", |
| 34 | return_value=self.FALLBACK_VERSION, |
| 35 | ) |
| 36 | |
| 37 | def testInvalidLkgm(self): |
| 38 | """LKGM version file found, but not successfully parsed.""" |
| 39 | |
| 40 | self.PatchObject(chrome_lkgm_lib, "GetChromeLkgm", return_value=None) |
| 41 | |
| 42 | chrome_lkgm.FindLkgm(self.request, self.response, self.api_config) |
| 43 | self.assertTrue(self.response.error) |
| 44 | self.get_full_version_mock.assert_not_called() |
| 45 | |
| 46 | def testLkgmNotFound(self): |
| 47 | """LKGM version file not found.""" |
| 48 | |
| 49 | self.PatchObject( |
| 50 | chrome_lkgm_lib, |
| 51 | "GetChromeLkgm", |
| 52 | side_effect=FileNotFoundError("CHROMEOS_LKGM not found"), |
| 53 | ) |
| 54 | |
| 55 | chrome_lkgm.FindLkgm(self.request, self.response, self.api_config) |
| 56 | self.assertTrue(self.response.error) |
| 57 | self.get_full_version_mock.assert_not_called() |
| 58 | |
| 59 | def testLkgmFound(self): |
| 60 | """LKGM version found.""" |
| 61 | |
| 62 | self.PatchObject( |
| 63 | chrome_lkgm_lib, "GetChromeLkgm", return_value=self.LKGM_VERSION |
| 64 | ) |
| 65 | self.instance.config_name = f"{self.request.build_target.name}/release" |
| 66 | |
| 67 | chrome_lkgm.FindLkgm(self.request, self.response, self.api_config) |
| 68 | self.assertFalse(self.response.error) |
| 69 | self.assertEqual(self.FALLBACK_VERSION, self.response.full_version) |
| 70 | self.assertEqual("newboard/release", self.response.config_name) |
| 71 | self.assertEqual(self.LKGM_VERSION, self.response.chromeos_lkgm) |
| 72 | self.get_full_version_mock.assert_called_with(self.LKGM_VERSION) |
| 73 | |
| 74 | def testFailToGetFullVersion(self): |
| 75 | """LKGM version found, but fallbacked full version wasn't found.""" |
| 76 | |
| 77 | self.PatchObject( |
| 78 | chrome_lkgm_lib, "GetChromeLkgm", return_value=self.LKGM_VERSION |
| 79 | ) |
| 80 | self.PatchObject( |
| 81 | self.instance, "GetFullVersionFromLatest", return_value=None |
| 82 | ) |
| 83 | |
| 84 | chrome_lkgm.FindLkgm(self.request, self.response, self.api_config) |
| 85 | self.assertTrue(self.response.error) |