build/fuchsia/binary_sizes_test.py

#!/usr/bin/python
# Copyright 2020 The Chromium 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 copy
import math
import os
import shutil
import subprocess
import tempfile
import time
import unittest

import binary_sizes

from common import DIR_SOURCE_ROOT


class TestBinarySizes(unittest.TestCase):
  tmpdir = None

  @classmethod
  def setUpClass(cls):
    cls.tmpdir = tempfile.mkdtemp()

  @classmethod
  def tearDownClass(cls):
    shutil.rmtree(cls.tmpdir)

  # TODO(crbug.com/1145648): Add tests covering FAR file input and histogram
  # output.

  def testCommitFromBuildProperty(self):
    commit_position = binary_sizes.CommitPositionFromBuildProperty(
        'refs/heads/master@{#819458}')
    self.assertEqual(commit_position, 819458)

  def testCompressedSize(self):
    """Verifies that the compressed file size can be extracted from the zstd
    stderr output."""

    uncompressed_file = tempfile.NamedTemporaryFile(delete=False)
    for line in range(200):
      uncompressed_file.write(
          'Lorem ipsum dolor sit amet, consectetur adipiscing elit. '
          'Sed eleifend')
    uncompressed_file.close()
    compressed_path = uncompressed_file.name + '.zst'
    devnull = open(os.devnull)
    zstd_path = os.path.join(DIR_SOURCE_ROOT, 'third_party', 'zstd', 'bin',
                             'zstd')
    subprocess.call(
        [zstd_path, '-14', uncompressed_file.name, '-o', compressed_path],
        stderr=devnull)
    self.assertEqual(
        binary_sizes.CompressedSize(uncompressed_file.name, ['-14']),
        os.path.getsize(compressed_path))
    os.remove(uncompressed_file.name)
    os.remove(compressed_path)


if __name__ == '__main__':
  unittest.main()