blob: 1129eba476c98d5c9ed72bdf01143f15b2486d7e (
plain)
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
|
# 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/.
import imp
import sys
from pathlib import Path
from unittest.mock import patch
from mozunit import main
from mach.base import MachError
from mach.test.conftest import TestBase
class Entry:
"""Stub replacement for pkg_resources.EntryPoint"""
def __init__(self, providers):
self.providers = providers
def load(self):
def _providers():
return self.providers
return _providers
class TestEntryPoints(TestBase):
"""Test integrating with setuptools entry points"""
provider_dir = Path(__file__).parent.resolve() / "providers"
def _run_help(self):
return self._run_mach(["help"], entry_point="mach.providers")
@patch("pkg_resources.iter_entry_points")
def test_load_entry_point_from_directory(self, mock):
# Ensure parent module is present otherwise we'll (likely) get
# an error due to unknown parent.
if "mach.commands" not in sys.modules:
mod = imp.new_module("mach.commands")
sys.modules["mach.commands"] = mod
mock.return_value = [Entry([self.provider_dir])]
# Mach error raised due to conditions_invalid.py
with self.assertRaises(MachError):
self._run_help()
@patch("pkg_resources.iter_entry_points")
def test_load_entry_point_from_file(self, mock):
mock.return_value = [Entry([self.provider_dir / "basic.py"])]
result, stdout, stderr = self._run_help()
self.assertIsNone(result)
self.assertIn("cmd_foo", stdout)
if __name__ == "__main__":
main()
|
