| # Copyright 2023 The ChromiumOS Authors |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| |
| """Unittests for remoteexec_lib.py""" |
| |
| from chromite.lib import cros_test_lib |
| from chromite.lib import osutils |
| from chromite.lib import remoteexec_lib |
| |
| |
| class TestLogArchiver(cros_test_lib.RunCommandTempDirTestCase): |
| """Tests for remoteexec_lib.""" |
| |
| def setUp(self): |
| self.src_dir = self.tempdir / "src_dir" |
| self.dest_dir = self.tempdir / "dest_dir" |
| |
| osutils.SafeMakedirs(self.src_dir) |
| osutils.SafeMakedirs(self.dest_dir) |
| |
| self.archiver = remoteexec_lib.LogsArchiver(self.dest_dir) |
| self.archiver.src_dir_for_testing = self.src_dir |
| |
| def _create_file(self, package_name: str, filename: str): |
| osutils.WriteFile( |
| self.src_dir / f"reclient-{package_name}" / filename, |
| f"Package: {package_name}\nFile: {filename}", |
| makedirs=True, |
| ) |
| |
| def testArchiveFiles(self): |
| """Test LogArchiver.Archive() method.""" |
| self._create_file("chromeos-chrome", "test.INFO.log") |
| self._create_file("chromeos-chrome", "test.INFO") |
| self._create_file("chromeos-chrome", "reproxy_test.INFO") |
| self._create_file("chromeos-chrome", "reproxy_test.rrpl") |
| |
| log_files = self.archiver.archive() |
| |
| self.assertEqual( |
| log_files, |
| [ |
| "reclient-chromeos-chrome/test.INFO.log.gz", |
| "reclient-chromeos-chrome/reproxy_test.INFO.gz", |
| "reclient-chromeos-chrome/reproxy_test.rrpl.gz", |
| ], |
| ) |
