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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
|
# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
from pathlib import Path
from unittest import mock
from unittest.mock import Mock, patch
import conftest # noqa: F401
import pytest
from mozunit import main
import mach.decorators
import mach.registrar
from mach.requirements import MachEnvRequirements
from mach.site import MozSiteMetadata, SitePackagesSource
from mutlh.decorators import Command, CommandArgument, MutlhCommandBase
from mutlh.site import MutlhCommandSiteManager
@pytest.fixture
def registrar(monkeypatch):
test_registrar = mach.registrar.MachRegistrar()
test_registrar.register_category("testing", "Mach unittest", "Testing for mach decorators")
monkeypatch.setattr(mach.decorators, "Registrar", test_registrar)
return test_registrar
def test_register_command_with_argument(registrar):
inner_function = Mock()
context = Mock()
context.cwd = "."
@Command("cmd_foo", category="testing")
@CommandArgument("--arg", default=None, help="Argument help.")
def run_foo(command_context, arg):
inner_function(arg)
registrar.dispatch("cmd_foo", context, arg="argument")
inner_function.assert_called_with("argument")
def test_register_command_sets_up_class_at_runtime(registrar):
inner_function = Mock()
context = Mock()
context.cwd = "."
# We test that the virtualenv is set up properly dynamically on
# the instance that actually runs the command.
@Command("cmd_foo", category="testing", virtualenv_name="env_foo")
def run_foo(command_context):
assert Path(command_context.virtualenv_manager.virtualenv_root).name == "env_foo"
inner_function("foo")
@Command("cmd_bar", category="testing", virtualenv_name="env_bar")
def run_bar(command_context):
assert Path(command_context.virtualenv_manager.virtualenv_root).name == "env_bar"
inner_function("bar")
def from_environment_patch(topsrcdir: str, state_dir: str, virtualenv_name, directory: str):
return MutlhCommandSiteManager(
"",
"",
virtualenv_name,
virtualenv_name,
MozSiteMetadata(0, "mach", SitePackagesSource.VENV, "", ""),
True,
MachEnvRequirements(),
)
with mock.patch.object(MutlhCommandSiteManager, "from_environment", from_environment_patch):
with patch.object(MutlhCommandBase, "activate_virtualenv"):
registrar.dispatch("cmd_foo", context)
inner_function.assert_called_with("foo")
registrar.dispatch("cmd_bar", context)
inner_function.assert_called_with("bar")
if __name__ == "__main__":
main()
|
