server/site_tests/autoupdate_Lacros/autoupdate_Lacros.py

# Copyright 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.

import logging

from autotest_lib.client.common_lib import error
from autotest_lib.client.common_lib.cros import kernel_utils
from autotest_lib.server.cros.update_engine import update_engine_test


class autoupdate_Lacros(update_engine_test.UpdateEngineTest):
    """Performs a simple AU test and checks lacros."""
    version = 1

    def cleanup(self):
        super(autoupdate_Lacros, self).cleanup()

    def run_once(self, full_payload, job_repo_url=None, running_at_desk=False):
        """
        Performs autoupdate with Nebraska and checks rootfs-lacros.

        @param full_payload: True for full payload, False for delta
        @param job_repo_url: A url pointing to the devserver where the autotest
            package for this build should be staged.
        @param running_at_desk: Indicates test is run locally from workstation.
                                Flag does not work with M2N tests.

        """
        # Login and check rootfs-lacros version
        self._run_client_test_and_check_result('desktopui_RootfsLacros',
                                               tag='before')
        before_version = self._host.run(['cat',
                                         '/tmp/lacros_version.txt']).stdout
        logging.info('rootfs-lacros version before update: %s', before_version)

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

        # Record DUT state before the update.
        active, inactive = kernel_utils.get_kernel_state(self._host)

        # Perform the update.
        self._run_client_test_and_check_result('autoupdate_CannedOmahaUpdate',
                                               payload_url=payload_url)

        # Verify the update completed successfully.
        self._host.reboot()
        kernel_utils.verify_boot_expectations(inactive, host=self._host)
        rootfs_hostlog, _ = self._create_hostlog_files()
        self.verify_update_events(self._FORCED_UPDATE, rootfs_hostlog)

        # Check the rootfs-lacros version again.
        self._run_client_test_and_check_result('desktopui_RootfsLacros',
                                               tag='after',
                                               dont_override_profile=True)
        after_version = self._host.run(['cat',
                                        '/tmp/lacros_version.txt']).stdout
        logging.info('rootfs-lacros version after update: %s', after_version)

        if before_version != after_version:
            raise error.TestFail(
                    'rootfs-lacros versions before and after update do not match'
            )