You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
95 lines
2.9 KiB
Python
95 lines
2.9 KiB
Python
#!/usr/bin/env vpython3
|
|
# Copyright 2023 The ChromiumOS Authors
|
|
# Use of this source code is governed by a BSD-style license that can be
|
|
# found in the LICENSE file.
|
|
|
|
# [VPYTHON:BEGIN]
|
|
# python_version: "3.8"
|
|
# [VPYTHON:END]
|
|
"""Tests for Bazel launcher."""
|
|
|
|
import os
|
|
from pathlib import Path
|
|
import site
|
|
import sys
|
|
import unittest
|
|
|
|
DEPOT_TOOLS_DIR = Path(__file__).resolve().parent.parent
|
|
site.addsitedir(DEPOT_TOOLS_DIR)
|
|
|
|
import bazel
|
|
from testing_support import trial_dir
|
|
|
|
|
|
class FindCrosUnittest(trial_dir.TestCase):
|
|
"""Test the _find_bazel_cros function."""
|
|
def setUp(self):
|
|
"""Create the checkout and chromite files."""
|
|
super().setUp()
|
|
self.checkout_dir = Path(self.root_dir) / "chromiumos"
|
|
self.chromite_dir = self.checkout_dir / "chromite"
|
|
self.launcher = self.chromite_dir / "bin" / "bazel"
|
|
self.launcher.parent.mkdir(exist_ok=True, parents=True)
|
|
self.launcher.write_bytes(b"")
|
|
self.launcher.chmod(0o775)
|
|
self.orig_dir = Path.cwd()
|
|
|
|
def tearDown(self):
|
|
os.chdir(self.orig_dir)
|
|
super().tearDown()
|
|
|
|
def test_at_checkout_base(self):
|
|
"""Test we find the launcher at the base of the checkout."""
|
|
os.chdir(self.checkout_dir)
|
|
self.assertEqual(bazel._find_bazel_cros(), self.launcher)
|
|
|
|
def test_in_checkout_subdir(self):
|
|
"""Test we find the launcher in a subdir of the checkout."""
|
|
os.chdir(self.chromite_dir)
|
|
self.assertEqual(bazel._find_bazel_cros(), self.launcher)
|
|
|
|
def test_out_of_checkout(self):
|
|
"""Test we don't find the launcher outside of the checkout."""
|
|
os.chdir(self.root_dir)
|
|
self.assertIsNone(bazel._find_bazel_cros())
|
|
|
|
|
|
class FindPathUnittest(trial_dir.TestCase):
|
|
"""Test the _find_next_bazel_in_path function."""
|
|
def setUp(self):
|
|
"""Create the checkout and chromite files."""
|
|
super().setUp()
|
|
|
|
self.bin_dir = Path(self.root_dir) / "bin"
|
|
self.bin_dir.mkdir(exist_ok=True, parents=True)
|
|
self.orig_path = os.environ.get("PATH", os.defpath)
|
|
|
|
# DEPOT_TOOLS_DIR is located twice in PATH for spice.
|
|
os.environ["PATH"] = os.pathsep.join([
|
|
str(DEPOT_TOOLS_DIR),
|
|
str(self.bin_dir),
|
|
str(DEPOT_TOOLS_DIR),
|
|
])
|
|
|
|
def tearDown(self):
|
|
"""Restore actions from setUp()."""
|
|
os.environ["PATH"] = self.orig_path
|
|
|
|
def test_not_in_path(self):
|
|
"""Test we don't find anything in PATH when not present."""
|
|
self.assertIsNone(bazel._find_next_bazel_in_path())
|
|
|
|
def test_in_path(self):
|
|
"""Test we find the next Bazel in PATH when present."""
|
|
if sys.platform == "win32":
|
|
launcher = self.bin_dir / "bazel.exe"
|
|
else:
|
|
launcher = self.bin_dir / "bazel"
|
|
launcher.write_bytes(b"")
|
|
launcher.chmod(0o755)
|
|
self.assertEqual(bazel._find_next_bazel_in_path(), launcher)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|