| # Lint as: python2, python3 |
| # Copyright (c) 2021 The Chromium OS Authors. All rights reserved. |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| |
| from autotest_lib.client.common_lib import error |
| from autotest_lib.client.cros.update_engine import nebraska_wrapper |
| from autotest_lib.client.cros.update_engine import update_engine_test |
| |
| |
| class autoupdate_InvalidateSuccessfulUpdate(update_engine_test.UpdateEngineTest |
| ): |
| """Tests installing an update and then invalidating it.""" |
| |
| version = 1 |
| |
| def _apply_update(self, update_url): |
| """ |
| Performs the update and ensures it is successful. |
| |
| @param update_url: The URL to get an update. |
| |
| """ |
| try: |
| self._check_for_update(update_url, |
| critical_update=True, |
| wait_for_completion=True) |
| except error.CmdError as e: |
| raise error.TestFail('Update attempt failed: %s' % |
| self._get_last_error_string()) |
| |
| def _check_invalidated_update(self, update_url): |
| """ |
| Performs an update check and confirms that it results |
| in an invalidated update. |
| |
| @param update_url: The URL to get an update. |
| |
| """ |
| try: |
| self._check_for_update(update_url, |
| critical_update=True, |
| wait_for_completion=False) |
| self._wait_for_update_to_idle(check_kernel_after_update=True, |
| inactive_kernel=False) |
| except error.CmdError as e: |
| raise error.TestFail('Invalidate attempt failed: %s' % |
| self._get_last_error_string()) |
| self._check_update_engine_log_for_entry( |
| 'Invalidating previous update.', |
| raise_error=True, |
| err_str='Failed to invalidate previous update') |
| |
| def run_once(self, payload_url): |
| """ |
| Runs an update and then invalidates it using Nebraska. |
| |
| @param payload_url: Path to a payload on Google storage. |
| |
| """ |
| with nebraska_wrapper.NebraskaWrapper( |
| log_dir=self.resultsdir, payload_url=payload_url) as nebraska: |
| self._apply_update(nebraska.get_update_url()) |
| nebraska.update_config(invalidate_last_update=True) |
| self._check_invalidated_update(nebraska.get_update_url()) |
