99 lines
		
	
	
		
			4.2 KiB
		
	
	
	
		
			Python
		
	
	
	
			
		
		
	
	
			99 lines
		
	
	
		
			4.2 KiB
		
	
	
	
		
			Python
		
	
	
	
| # Copyright 2020 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_WithDLC(update_engine_test.UpdateEngineTest):
 | |
|     """Tests basic DLC installation and n-to-n updating. """
 | |
| 
 | |
|     version = 1
 | |
|     _CLIENT_TEST = 'autoupdate_InstallAndUpdateDLC'
 | |
| 
 | |
|     def initialize(self, host=None):
 | |
|         """Remove all DLCs on the DUT before starting the test. """
 | |
|         super(autoupdate_WithDLC, self).initialize(host=host)
 | |
|         installed = self._dlc_util.list().keys()
 | |
|         for dlc_id in installed:
 | |
|             self._dlc_util.purge(dlc_id)
 | |
|         # DLCs may be present but not mounted, so they won't be purged above.
 | |
|         self._dlc_util.purge(self._dlc_util._SAMPLE_DLC_ID, ignore_status=True)
 | |
| 
 | |
| 
 | |
|     def cleanup(self):
 | |
|         self._save_extra_update_engine_logs(number_of_logs=2)
 | |
|         super(autoupdate_WithDLC, self).cleanup()
 | |
| 
 | |
| 
 | |
|     def run_once(self, full_payload=True, job_repo_url=None):
 | |
|         """
 | |
|         Tests that we can successfully install a DLC, and then update it along
 | |
|         with the OS.
 | |
| 
 | |
|         @param full_payload: True for a full payload. False for delta.
 | |
|         @param job_repo_url: Used for debugging locally. This is used to figure
 | |
|                              out the current build and the devserver to use.
 | |
|                              The test will read this from a host argument
 | |
|                              when run in the lab.
 | |
| 
 | |
|         """
 | |
|         payload_urls = []
 | |
| 
 | |
|         # Payload URL for the platform (OS) update
 | |
|         payload_urls.append(
 | |
|             self.get_payload_for_nebraska(job_repo_url=job_repo_url,
 | |
|                                           full_payload=full_payload))
 | |
| 
 | |
|         # Payload URLs for sample-dlc, a test DLC package.
 | |
|         # We'll always need a full payload for DLC installation,
 | |
|         # and optionally a delta payload if required by the test.
 | |
|         payload_urls.append(
 | |
|             self.get_payload_for_nebraska(job_repo_url=job_repo_url,
 | |
|                                           full_payload=True, is_dlc=True))
 | |
|         if not full_payload:
 | |
|             payload_urls.append(
 | |
|                 self.get_payload_for_nebraska(
 | |
|                     job_repo_url=job_repo_url, full_payload=False,
 | |
|                     is_dlc=True))
 | |
| 
 | |
|         active, inactive = kernel_utils.get_kernel_state(self._host)
 | |
| 
 | |
|         # Install and update sample-dlc, a DLC package made for test purposes.
 | |
|         self._run_client_test_and_check_result(self._CLIENT_TEST,
 | |
|                                                payload_urls=payload_urls,
 | |
|                                                full_payload=full_payload)
 | |
| 
 | |
|         self._host.reboot()
 | |
| 
 | |
|         # Verify the update was successful by checking hostlog and kernel.
 | |
|         rootfs_hostlog, _ = self._create_hostlog_files()
 | |
|         dlc_rootfs_hostlog, _ = self._create_dlc_hostlog_files()
 | |
| 
 | |
|         logging.info('Checking platform update events')
 | |
|         self.verify_update_events(self._FORCED_UPDATE, rootfs_hostlog)
 | |
| 
 | |
|         logging.info('Checking DLC update events')
 | |
|         self.verify_update_events(
 | |
|                 self._FORCED_UPDATE,
 | |
|                 dlc_rootfs_hostlog[self._dlc_util._SAMPLE_DLC_ID])
 | |
| 
 | |
|         kernel_utils.verify_boot_expectations(inactive, host=self._host)
 | |
| 
 | |
|         # After the update and reboot, the DLC will be installed but not yet
 | |
|         # mounted. If the DLC was updated correctly, calling
 | |
|         # |dlcservice_util --install| in this state should mount the DLC
 | |
|         # without hitting Omaha. We can verify this by checking:
 | |
|         # 1. the DLC is not being preloaded (|dlcservice_util --install| will
 | |
|         #    show the same behavior if the DLC is preloaded)
 | |
|         # 2. the install doesn't hit Omaha/Nebraska, by using a bad omaha_url
 | |
|         self._dlc_util.remove_preloaded(self._dlc_util._SAMPLE_DLC_ID)
 | |
|         self._dlc_util.install(self._dlc_util._SAMPLE_DLC_ID,
 | |
|                                omaha_url='fake_url')
 | |
|         if not self._dlc_util.is_installed(self._dlc_util._SAMPLE_DLC_ID):
 | |
|             raise error.TestFail('Dummy DLC was not installed.')
 |