blob: faf165863db61111319beef4052c06229e65533c (
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
60
61
62
63
64
65
|
# 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 __future__ import absolute_import
from __future__ import unicode_literals
import os
import unittest
from io import StringIO
import pytest
from mozunit import main
from six import string_types
from mach.base import CommandContext
from mach.registrar import Registrar
here = os.path.abspath(os.path.dirname(__file__))
@pytest.mark.usefixtures("get_mach", "run_mach")
class TestDispatcher(unittest.TestCase):
"""Tests dispatch related code"""
def get_parser(self, config=None):
mach = self.get_mach("basic.py")
for provider in Registrar.settings_providers:
mach.settings.register_provider(provider)
if config:
if isinstance(config, string_types):
config = StringIO(config)
mach.settings.load_fps([config])
context = CommandContext(settings=mach.settings)
return mach.get_argument_parser(context)
def test_command_aliases(self):
config = """
[alias]
foo = cmd_foo
bar = cmd_bar
baz = cmd_bar --baz
cmd_bar = cmd_bar --baz
"""
parser = self.get_parser(config=config)
args = parser.parse_args(["foo"])
self.assertEquals(args.command, "cmd_foo")
def assert_bar_baz(argv):
args = parser.parse_args(argv)
self.assertEquals(args.command, "cmd_bar")
self.assertTrue(args.command_args.baz)
# The following should all result in |cmd_bar --baz|
assert_bar_baz(["bar", "--baz"])
assert_bar_baz(["baz"])
assert_bar_baz(["cmd_bar"])
if __name__ == "__main__":
main()
|
