build/fuchsia/binary_sizes_test.py

   1 #!/usr/bin/env vpython3
   2 # Copyright 2020 The Chromium Authors
   3 # Use of this source code is governed by a BSD-style license that can be
   4 # found in the LICENSE file.
   5
   6 import json
   7 import os
   8 import shutil
   9 import tempfile
  10 import unittest
  11
  12 import binary_sizes
  13
  14
  15 _EXAMPLE_BLOBS = """
  16 {
  17   "web_engine": [
  18     {
  19       "merkle": "77e876447dd2daaaab7048d646e87fe8b6d9fecef6cbfcc4af30b8fbfa50b881",
  20       "path": "locales/ta.pak",
  21       "bytes": 17916,
  22       "is_counted": true,
  23       "size": 16384
  24     },
  25     {
  26       "merkle": "5f1932b8c9fe954f3c3fdb34ab2089d2af34e5a0cef90cad41a1cd37d92234bf",
  27       "path": "lib/libEGL.so",
  28       "bytes": 226960,
  29       "is_counted": true,
  30       "size": 90112
  31     },
  32     {
  33       "merkle": "9822fc0dd95cdd1cc46b5c6632a928a6ad19b76ed0157397d82a2f908946fc34",
  34       "path": "meta.far",
  35       "bytes": 24576,
  36       "is_counted": true,
  37       "size": 16384
  38     },
  39     {
  40       "merkle": "090aed4593c4f7d04a3ad80e9971c0532dd5b1d2bdf4754202cde510a88fd220",
  41       "path": "locales/ru.pak",
  42       "bytes": 11903,
  43       "is_counted": true,
  44       "size": 16384
  45     }
  46   ]
  47 }
  48 """
  49
  50
  51 class TestBinarySizes(unittest.TestCase):
  52   tmpdir = None
  53
  54   @classmethod
  55   def setUpClass(cls):
  56     cls.tmpdir = tempfile.mkdtemp()
  57
  58   @classmethod
  59   def tearDownClass(cls):
  60     shutil.rmtree(cls.tmpdir)
  61
  62
  63   def testReadAndWritePackageBlobs(self):
  64     # TODO(1309977): Disabled on Windows because Windows doesn't allow opening a
  65     # NamedTemporaryFile by name.
  66     if os.name == 'nt':
  67       return
  68     with tempfile.NamedTemporaryFile(mode='w') as tmp_file:
  69       tmp_file.write(_EXAMPLE_BLOBS)
  70       tmp_file.flush()
  71
  72       package_blobs = binary_sizes.ReadPackageBlobsJson(tmp_file.name)
  73
  74     tmp_package_file = tempfile.NamedTemporaryFile(mode='w', delete=False)
  75     tmp_package_file.close()
  76     try:
  77       binary_sizes.WritePackageBlobsJson(tmp_package_file.name, package_blobs)
  78
  79       self.assertEqual(binary_sizes.ReadPackageBlobsJson(tmp_package_file.name),
  80                        package_blobs)
  81     finally:
  82       os.remove(tmp_package_file.name)
  83
  84   def testReadAndWritePackageSizes(self):
  85     # TODO(1309977): Disabled on Windows because Windows doesn't allow opening a
  86     # NamedTemporaryFile by name.
  87     if os.name == 'nt':
  88       return
  89     with tempfile.NamedTemporaryFile(mode='w') as tmp_file:
  90       tmp_file.write(_EXAMPLE_BLOBS)
  91       tmp_file.flush()
  92       blobs = binary_sizes.ReadPackageBlobsJson(tmp_file.name)
  93
  94     sizes = binary_sizes.GetPackageSizes(blobs)
  95
  96     new_sizes = {}
  97     with tempfile.NamedTemporaryFile(mode='w') as tmp_file:
  98       binary_sizes.WritePackageSizesJson(tmp_file.name, sizes)
  99       new_sizes = binary_sizes.ReadPackageSizesJson(tmp_file.name)
 100       self.assertEqual(new_sizes, sizes)
 101       self.assertIn('web_engine', new_sizes)
 102
 103   def testGetPackageSizesUsesBlobMerklesForCount(self):
 104     # TODO(1309977): Disabled on Windows because Windows doesn't allow opening a
 105     # NamedTemporaryFile by name.
 106     if os.name == 'nt':
 107       return
 108     blobs = json.loads(_EXAMPLE_BLOBS)
 109
 110     # Make a duplicate of the last blob.
 111     last_blob = dict(blobs['web_engine'][-1])
 112     blobs['cast_runner'] = []
 113     last_blob['path'] = 'foo'  # Give a non-sense name, but keep merkle.
 114
 115     # If the merkle is the same, the blob_count increases by 1.
 116     # This effectively reduces the size of the blobs size by half.
 117     # In both packages, despite it appearing in both and under different
 118     # names.
 119     blobs['cast_runner'].append(last_blob)
 120
 121     with tempfile.NamedTemporaryFile(mode='w') as tmp_file:
 122       tmp_file.write(json.dumps(blobs))
 123       tmp_file.flush()
 124       blobs = binary_sizes.ReadPackageBlobsJson(tmp_file.name)
 125
 126     sizes = binary_sizes.GetPackageSizes(blobs)
 127
 128     self.assertEqual(sizes['cast_runner'].compressed, last_blob['size'] / 2)
 129
 130
 131 if __name__ == '__main__':
 132   unittest.main()