mirror of
https://github.com/nodejs/node.git
synced 2025-05-15 08:02:06 +00:00
9d7cd9b864
PR-URL: https://github.com/nodejs/node/pull/54077 Reviewed-By: Jiawen Geng <technicalcute@gmail.com> Reviewed-By: Richard Lau <rlau@redhat.com> Reviewed-By: Joyee Cheung <joyeec9h3@gmail.com> Reviewed-By: Marco Ippolito <marcoippolito54@gmail.com>
132 lines
4.2 KiB
Python
132 lines
4.2 KiB
Python
#!/usr/bin/env python3
|
|
|
|
# Copyright 2023 the V8 project 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 contextlib
|
|
import io
|
|
import os
|
|
import pathlib
|
|
import unittest
|
|
|
|
from tempfile import TemporaryDirectory
|
|
from unittest.mock import patch, mock_open
|
|
|
|
from download_profiles import main, parse_args, retrieve_version
|
|
|
|
|
|
def mock_version_file(major, minor, build, patch):
|
|
return (
|
|
f'#define V8_MAJOR_VERSION {major}\n'
|
|
f'#define V8_MINOR_VERSION {minor}\n'
|
|
f'#define V8_BUILD_NUMBER {build}\n'
|
|
f'#define V8_PATCH_LEVEL {patch}\n'
|
|
)
|
|
|
|
|
|
class TestDownloadProfiles(unittest.TestCase):
|
|
|
|
def _test_cmd(self, cmd, exitcode):
|
|
out = io.StringIO()
|
|
err = io.StringIO()
|
|
with self.assertRaises(SystemExit) as se, \
|
|
contextlib.redirect_stdout(out), \
|
|
contextlib.redirect_stderr(err):
|
|
main(cmd)
|
|
self.assertEqual(se.exception.code, exitcode)
|
|
return out.getvalue(), err.getvalue()
|
|
|
|
def test_validate_profiles(self):
|
|
out, err = self._test_cmd(['validate', '--version', '11.1.0.0'], 0)
|
|
self.assertEqual(len(out), 0)
|
|
self.assertEqual(len(err), 0)
|
|
|
|
def test_download_profiles(self):
|
|
with TemporaryDirectory() as td, \
|
|
patch('download_profiles.PGO_PROFILE_DIR', pathlib.Path(td)):
|
|
out, err = self._test_cmd(['download', '--version', '11.1.0.0'], 0)
|
|
self.assertEqual(len(out), 0)
|
|
self.assertEqual(len(err), 0)
|
|
self.assertGreater(
|
|
len([f for f in os.listdir(td) if f.endswith('.profile')]), 0)
|
|
with open(pathlib.Path(td) / 'profiles_version') as f:
|
|
self.assertEqual(f.read(), '11.1.0.0')
|
|
|
|
# A second download should not be started as profiles exist already
|
|
with patch('download_profiles.call_gsutil') as gsutil:
|
|
out, err = self._test_cmd(['download', '--version', '11.1.0.0'], 0)
|
|
self.assertEqual(
|
|
out,
|
|
'Profiles already downloaded, use --force to overwrite.\n',
|
|
)
|
|
gsutil.assert_not_called()
|
|
|
|
# A forced download should always trigger
|
|
with patch('download_profiles.call_gsutil') as gsutil:
|
|
cmd = ['download', '--version', '11.1.0.0', '--force']
|
|
out, err = self._test_cmd(cmd, 0)
|
|
self.assertEqual(len(out), 0)
|
|
self.assertEqual(len(err), 0)
|
|
gsutil.assert_called_once()
|
|
|
|
def test_arg_quiet(self):
|
|
with patch(
|
|
'builtins.open',
|
|
new=mock_open(read_data=mock_version_file(11, 9, 0, 0))):
|
|
out, err = self._test_cmd(['download'], 0)
|
|
self.assertGreater(len(out), 0)
|
|
|
|
out, err = self._test_cmd(['download', '--quiet'], 0)
|
|
self.assertEqual(len(out), 0)
|
|
|
|
def test_invalid_args(self):
|
|
out, err = self._test_cmd(['invalid-action'], 2)
|
|
self.assertEqual(len(out), 0)
|
|
self.assertGreater(len(err), 0)
|
|
|
|
def test_invalid_depot_tools_path(self):
|
|
out, err = self._test_cmd(
|
|
['validate', '--depot-tools', '/no-depot-tools-path'], 3)
|
|
self.assertEqual(len(out), 0)
|
|
self.assertGreater(len(err), 0)
|
|
|
|
def test_missing_profiles(self):
|
|
out, err = self._test_cmd(['download', '--version', '0.0.0.42'], 4)
|
|
self.assertEqual(len(out), 0)
|
|
self.assertGreater(len(err), 0)
|
|
|
|
|
|
class TestRetrieveVersion(unittest.TestCase):
|
|
def test_retrieve_valid_version(self):
|
|
with patch(
|
|
'builtins.open',
|
|
new=mock_open(read_data=mock_version_file(11, 4, 1, 0))):
|
|
args = parse_args(['download'])
|
|
version = retrieve_version(args)
|
|
|
|
self.assertEqual(version, '11.4.1.0')
|
|
|
|
def test_retrieve_parameter_version(self):
|
|
args = parse_args(['download', '--version', '11.1.1.42'])
|
|
version = retrieve_version(args)
|
|
self.assertEqual(version, '11.1.1.42')
|
|
|
|
def test_retrieve_untagged_version(self):
|
|
out = io.StringIO()
|
|
with patch(
|
|
'builtins.open',
|
|
new=mock_open(read_data=mock_version_file(11, 4, 0, 0))), \
|
|
contextlib.redirect_stdout(out), \
|
|
self.assertRaises(SystemExit) as se:
|
|
args = parse_args(['download'])
|
|
version = retrieve_version(args)
|
|
|
|
self.assertEqual(se.exception.code, 0)
|
|
self.assertEqual(out.getvalue(),
|
|
'The version file specifies 11.4.0.0, which has no profiles.\n')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|