1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
|
# Copyright 2021 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""Test the zmake re-exec functionality."""
import os
import sys
import unittest.mock as mock
import pytest
import zmake.__main__ as main
@pytest.fixture
def fake_env(monkeypatch):
environ = {}
monkeypatch.setattr(os, "environ", environ)
return environ
@pytest.fixture
def mock_execve():
with mock.patch("os.execve", autospec=True) as mocked_function:
yield mocked_function
def test_out_of_chroot(fake_env, mock_execve):
# When CROS_WORKON_SRCROOT is not set, we should not re-exec.
main.maybe_reexec(["--help"])
mock_execve.assert_not_called()
def test_pythonpath_set(fake_env, mock_execve):
# With PYTHONPATH set, we should not re-exec.
fake_env["CROS_WORKON_SRCROOT"] = "/mnt/host/source"
fake_env["PYTHONPATH"] = "/foo/bar/baz"
main.maybe_reexec(["--help"])
mock_execve.assert_not_called()
def test_zmake_does_not_exist(fake_env, mock_execve):
# When zmake is not at src/platform/ec/zephyr/zmake, don't re-exec.
fake_env["CROS_WORKON_SRCROOT"] = "/this/does/not/exist"
main.maybe_reexec(["--help"])
mock_execve.assert_not_called()
def test_zmake_reexec(fake_env, mock_execve):
# Nothing else applies? The re-exec should happen.
fake_env["CROS_WORKON_SRCROOT"] = "/mnt/host/source"
main.maybe_reexec(["--help"])
new_env = dict(fake_env)
new_env["PYTHONPATH"] = "/mnt/host/source/src/platform/ec/zephyr/zmake"
mock_execve.assert_called_once_with(
sys.executable,
[sys.executable, "-m", "zmake", "--help"],
new_env,
)
|
