mirror of
https://github.com/klzgrad/naiveproxy.git
synced 2024-11-25 14:56:10 +03:00
149 lines
4.3 KiB
Python
Executable File
149 lines
4.3 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# Copyright 2023 The Chromium Authors
|
|
# Use of this source code is governed by a BSD-style license that can be
|
|
# found in the LICENSE file.
|
|
|
|
import os
|
|
import pathlib
|
|
import shutil
|
|
import sys
|
|
import tempfile
|
|
import textwrap
|
|
import unittest
|
|
from unittest import mock
|
|
|
|
import clobber
|
|
|
|
|
|
class TestExtractBuildCommand(unittest.TestCase):
|
|
def setUp(self):
|
|
self.build_ninja_file, self.build_ninja_path = tempfile.mkstemp(text=True)
|
|
|
|
def tearDown(self):
|
|
os.close(self.build_ninja_file)
|
|
os.remove(self.build_ninja_path)
|
|
|
|
def test_normal_extraction(self):
|
|
build_ninja_file_contents = textwrap.dedent("""
|
|
ninja_required_version = 1.7.2
|
|
|
|
rule gn
|
|
command = ../../buildtools/gn --root=../.. -q --regeneration gen .
|
|
pool = console
|
|
description = Regenerating ninja files
|
|
|
|
build build.ninja.stamp: gn
|
|
generator = 1
|
|
depfile = build.ninja.d
|
|
|
|
build build.ninja: phony build.ninja.stamp
|
|
generator = 1
|
|
|
|
pool build_toolchain_action_pool
|
|
depth = 72
|
|
|
|
pool build_toolchain_link_pool
|
|
depth = 23
|
|
|
|
subninja toolchain.ninja
|
|
subninja clang_newlib_x64/toolchain.ninja
|
|
subninja glibc_x64/toolchain.ninja
|
|
subninja irt_x64/toolchain.ninja
|
|
subninja nacl_bootstrap_x64/toolchain.ninja
|
|
subninja newlib_pnacl/toolchain.ninja
|
|
|
|
build blink_python_tests: phony obj/blink_python_tests.stamp
|
|
build blink_tests: phony obj/blink_tests.stamp
|
|
|
|
default all
|
|
""") # Based off of a standard linux build dir.
|
|
with open(self.build_ninja_path, 'w') as f:
|
|
f.write(build_ninja_file_contents)
|
|
|
|
expected_build_ninja_file_contents = textwrap.dedent("""
|
|
ninja_required_version = 1.7.2
|
|
|
|
rule gn
|
|
command = ../../buildtools/gn --root=../.. -q --regeneration gen .
|
|
pool = console
|
|
description = Regenerating ninja files
|
|
|
|
build build.ninja.stamp: gn
|
|
generator = 1
|
|
depfile = build.ninja.d
|
|
|
|
build build.ninja: phony build.ninja.stamp
|
|
generator = 1
|
|
|
|
""")
|
|
|
|
self.assertEqual(clobber.extract_gn_build_commands(self.build_ninja_path),
|
|
expected_build_ninja_file_contents)
|
|
|
|
def test_unexpected_format(self):
|
|
# No "build build.ninja:" line should make it return an empty string.
|
|
build_ninja_file_contents = textwrap.dedent("""
|
|
ninja_required_version = 1.7.2
|
|
|
|
rule gn
|
|
command = ../../buildtools/gn --root=../.. -q --regeneration gen .
|
|
pool = console
|
|
description = Regenerating ninja files
|
|
|
|
subninja toolchain.ninja
|
|
|
|
build blink_python_tests: phony obj/blink_python_tests.stamp
|
|
build blink_tests: phony obj/blink_tests.stamp
|
|
|
|
""")
|
|
with open(self.build_ninja_path, 'w') as f:
|
|
f.write(build_ninja_file_contents)
|
|
|
|
self.assertEqual(clobber.extract_gn_build_commands(self.build_ninja_path),
|
|
'')
|
|
|
|
|
|
class TestDelete(unittest.TestCase):
|
|
def setUp(self):
|
|
self.build_dir = tempfile.mkdtemp()
|
|
|
|
pathlib.Path(os.path.join(self.build_dir, 'build.ninja')).touch()
|
|
pathlib.Path(os.path.join(self.build_dir, 'build.ninja.d')).touch()
|
|
|
|
def tearDown(self):
|
|
shutil.rmtree(self.build_dir)
|
|
|
|
def test_delete_build_dir_full(self):
|
|
# Create a dummy file in the build dir and ensure it gets removed.
|
|
dummy_file = os.path.join(self.build_dir, 'dummy')
|
|
pathlib.Path(dummy_file).touch()
|
|
|
|
clobber.delete_build_dir(self.build_dir)
|
|
|
|
self.assertFalse(os.path.exists(dummy_file))
|
|
|
|
def test_delete_build_dir_fail(self):
|
|
# Make delete_dir() throw to ensure it's handled gracefully.
|
|
|
|
with mock.patch('clobber._clean_dir', side_effect=OSError):
|
|
with self.assertRaises(OSError):
|
|
clobber.delete_build_dir(self.build_dir)
|
|
|
|
@unittest.skipIf(sys.platform == 'win32', 'Symlinks are not allowed on Windows by default')
|
|
def test_delete_build_dir_link(self):
|
|
with tempfile.TemporaryDirectory() as tmpdir:
|
|
# create a symlink.
|
|
build_dir = os.path.join(tmpdir, 'link')
|
|
os.symlink(self.build_dir, build_dir)
|
|
|
|
# create a dummy file.
|
|
dummy_file = os.path.join(build_dir, 'dummy')
|
|
pathlib.Path(dummy_file).touch()
|
|
clobber.delete_build_dir(build_dir)
|
|
|
|
self.assertFalse(os.path.exists(dummy_file))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|