# Lint as: python2, python3
# Copyright 2021 The ChromiumOS Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

from autotest_lib.server.cros.update_engine import update_engine_test


class autoupdate_InvalidateUpdateBeforeReboot(
        update_engine_test.UpdateEngineTest):
    """Tests to see if we can invalidate an update before a reboot."""
    version = 1
    _CLIENT_TEST = 'autoupdate_InvalidateSuccessfulUpdate'

    def run_once(self, full_payload=True, running_at_desk=False, build=None):
        """
        Runs the invalidate successful update test.

        @param full_payload: True for full payload, False for delta.
        @param running_at_desk: indicates test is run locally from a workstation.
        @param build: An optional parameter to specify the target build for the
                      update when running locally. If no build is supplied, the
                      current version on the DUT will be used.

        """
        # Get a payload to use for the test.
        payload_url = self.get_payload_for_nebraska(
                build=build,
                full_payload=full_payload,
                public_bucket=running_at_desk)

        # Perform an update, invalidate it and verify successful invalidation.
        self._run_client_test_and_check_result(self._CLIENT_TEST,
                                               payload_url=payload_url)

        # Verify via the logs the update was applied.
        self._check_update_engine_log_for_entry(
                'Update successfully applied, waiting to reboot.',
                raise_error=True)

        # Verify via the logs the update was invalidated.
        self._check_update_engine_log_for_entry(
                'Invalidating previous update.', raise_error=True)