| # -*- coding: utf-8 -*- |
| # Copyright (c) 2012 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. |
| |
| """Unittests for the binpkg.py module.""" |
| |
| from __future__ import print_function |
| |
| import os |
| import sys |
| |
| from chromite.api.gen.chromiumos import common_pb2 |
| from chromite.lib import binpkg |
| from chromite.lib import build_target_lib |
| from chromite.lib import cros_test_lib |
| from chromite.lib import gs_unittest |
| from chromite.lib import osutils |
| from chromite.lib import sysroot_lib |
| |
| |
| assert sys.version_info >= (3, 6), 'This module requires Python 3.6+' |
| |
| |
| PACKAGES_CONTENT = """USE: test |
| |
| CPV: chromeos-base/shill-0.0.1-r1 |
| |
| CPV: chromeos-base/test-0.0.1-r1 |
| DEBUG_SYMBOLS: yes |
| """ |
| |
| class FetchTarballsTest(cros_test_lib.MockTempDirTestCase): |
| """Tests for GSContext that go over the network.""" |
| |
| def testFetchFakePackages(self): |
| """Pretend to fetch binary packages.""" |
| gs_mock = self.StartPatcher(gs_unittest.GSContextMock()) |
| gs_mock.SetDefaultCmdResult() |
| uri = 'gs://foo/bar' |
| packages_uri = '{}/Packages'.format(uri) |
| packages_file = """URI: gs://foo |
| |
| CPV: boo/baz |
| PATH boo/baz.tbz2 |
| """ |
| gs_mock.AddCmdResult(['cat', packages_uri], output=packages_file) |
| |
| binpkg.FetchTarballs([uri], self.tempdir) |
| |
| @cros_test_lib.NetworkTest() |
| def testFetchRealPackages(self): |
| """Actually fetch a real binhost from the network.""" |
| uri = 'gs://chromeos-prebuilt/board/lumpy/paladin-R37-5905.0.0-rc2/packages' |
| binpkg.FetchTarballs([uri], self.tempdir) |
| |
| |
| class DebugSymbolsTest(cros_test_lib.TempDirTestCase): |
| """Tests for the debug symbols handling in binpkg.""" |
| |
| def testDebugSymbolsDetected(self): |
| """When generating the Packages file, DEBUG_SYMBOLS is updated.""" |
| osutils.WriteFile(os.path.join(self.tempdir, |
| 'chromeos-base/shill-0.0.1-r1.debug.tbz2'), |
| 'hello', makedirs=True) |
| osutils.WriteFile(os.path.join(self.tempdir, 'Packages'), |
| PACKAGES_CONTENT) |
| |
| index = binpkg.GrabLocalPackageIndex(self.tempdir) |
| self.assertEqual(index.packages[0]['CPV'], 'chromeos-base/shill-0.0.1-r1') |
| self.assertEqual(index.packages[0].get('DEBUG_SYMBOLS'), 'yes') |
| self.assertFalse('DEBUG_SYMBOLS' in index.packages[1]) |
| |
| |
| class PackageIndexTest(cros_test_lib.TempDirTestCase): |
| """Package index tests.""" |
| |
| def testReadWrite(self): |
| """Sanity check that the read and write method work properly.""" |
| packages1 = os.path.join(self.tempdir, 'Packages1') |
| packages2 = os.path.join(self.tempdir, 'Packages2') |
| |
| # Set some data. |
| pkg_index = binpkg.PackageIndex() |
| pkg_index.header['A'] = 'B' |
| pkg_index.packages = [ |
| {'CPV': 'foo/bar', |
| 'KEY': 'value'}, |
| {'CPV': 'cat/pkg', |
| 'KEY': 'also_value'}, |
| ] |
| |
| # Write the package index files using each writing method. |
| pkg_index.modified = True |
| pkg_index.WriteFile(packages1) |
| with open(packages2, 'w') as f: |
| pkg_index.Write(f) |
| tmpf = pkg_index.WriteToNamedTemporaryFile() |
| |
| # Make sure the files are the same. |
| fc1 = osutils.ReadFile(packages1) |
| fc2 = osutils.ReadFile(packages2) |
| fc3 = tmpf.read() |
| self.assertEqual(fc1, fc2) |
| self.assertEqual(fc1, fc3) |
| |
| # Make sure it parses out the same data we wrote. |
| with open(packages1) as f: |
| read_index = binpkg.PackageIndex() |
| read_index.Read(f) |
| |
| self.assertDictEqual(pkg_index.header, read_index.header) |
| self.assertCountEqual(pkg_index.packages, read_index.packages) |
| |
| class PackageIndexInfoTest(cros_test_lib.TempDirTestCase): |
| """Package index info tests.""" |
| |
| def _make_instance(self, sha, number, board, profile_name, location): |
| """Return a binpkg.PackageIndexInfo instance.""" |
| return binpkg.PackageIndexInfo( |
| snapshot_sha=sha, snapshot_number=number, |
| build_target=build_target_lib.BuildTarget( |
| name=board) if board else None, |
| profile=sysroot_lib.Profile( |
| name=profile_name) if profile_name else None, |
| location=location) |
| |
| def _make_proto(self, sha, number, board, profile_name, location): |
| """Return a common_pb2.PackageIndexInfo protobuf.""" |
| proto = common_pb2.PackageIndexInfo() |
| if sha: |
| proto.snapshot_sha = sha |
| if number: |
| proto.snapshot_number = number |
| if board: |
| proto.build_target.name = board |
| if profile_name: |
| proto.profile.name = profile_name |
| if location: |
| proto.location = location |
| return proto |
| |
| def testConversion(self): |
| """Sanity check converting to/from protobuf works.""" |
| |
| info = self._make_instance('SHA5', 5, 'target', 'profile', 'LOCATION') |
| proto = self._make_proto('SHA5', 5, 'target', 'profile', 'LOCATION') |
| |
| self.assertEqual(str(info.as_protobuf), str(proto)) |
| self.assertEqual(info.as_protobuf, proto) |
| |
| self.assertEqual(binpkg.PackageIndexInfo.from_protobuf(proto), info) |
| |
| def testEquality(self): |
| """Test that equality checks work.""" |
| info = self._make_instance('SHA5', 5, 'target', 'profile', 'LOCATION') |
| self.assertEqual( |
| info, self._make_instance('SHA5', 5, 'target', 'profile', 'LOCATION')) |
| self.assertNotEqual( |
| info, self._make_instance('XXXX', 5, 'target', 'profile', 'LOCATION')) |
| self.assertNotEqual( |
| info, self._make_instance('SHA5', 6, 'target', 'profile', 'LOCATION')) |
| self.assertNotEqual( |
| info, self._make_instance('SHA5', 5, 'xxxxxx', 'profile', 'LOCATION')) |
| self.assertNotEqual( |
| info, self._make_instance('SHA5', 5, 'target', 'xxxxxxx', 'LOCATION')) |
| self.assertNotEqual( |
| info, self._make_instance('SHA5', 5, 'target', 'profile', 'XXXXXXXX')) |