summaryrefslogtreecommitdiffstats
path: root/ipalib/tests/test_cli.py
blob: e80dee43a9876fbf869722d9a960369e59da3360 (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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
# Authors:
#   Jason Gerard DeRose <jderose@redhat.com>
#
# Copyright (C) 2008  Red Hat
# see file 'COPYING' for use and warranty information
#
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License as
# published by the Free Software Foundation; version 2 only
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA

"""
Unit tests for `ipalib.cli` module.
"""

from tstutil import raises, getitem, no_set, no_del, read_only, ClassChecker
from ipalib import cli


def test_to_cli():
    f = cli.to_cli
    assert f('initialize') == 'initialize'
    assert f('user_add') == 'user-add'


def test_from_cli():
    f = cli.from_cli
    assert f('initialize') == 'initialize'
    assert f('user-add') == 'user_add'


def get_cmd_name(i):
    return 'cmd_%d' % i

class DummyCmd(object):
    def __init__(self, name):
        self.__name = name

    def __get_name(self):
        return self.__name
    name = property(__get_name)

class DummyAPI(object):
    def __init__(self, cnt):
        self.__cmd = tuple(self.__cmd_iter(cnt))

    def __get_cmd(self):
        return self.__cmd
    cmd = property(__get_cmd)

    def __cmd_iter(self, cnt):
        for i in xrange(cnt):
            yield DummyCmd(get_cmd_name(i))

    def finalize(self):
        pass

    def register(self, *args, **kw):
        pass




class test_CLI(ClassChecker):
    """
    Tests the `CLI` class.
    """
    _cls = cli.CLI

    def test_class(self):
        assert type(self.cls.api) is property

    def test_api(self):
        """
        Tests the `api` property.
        """
        api = 'the plugable.API instance'
        o = self.cls(api)
        assert read_only(o, 'api') is api

    def test_parse_kw(self):
        """
        Tests the `parse_kw` method.
        """
        o = self.cls(None)
        kw = dict(
            hello='world',
            how_are='you',
        )
        args = tuple('--%s=%s' % (cli.to_cli(k), v) for (k,v) in kw.items())
        assert dict(o.parse_kw(args)) == kw

    def test_mcl(self):
        """
        Tests the `mcl` (Max Command Length) property .
        """
        cnt = 100
        api = DummyAPI(cnt)
        len(api.cmd) == cnt
        o = self.cls(api)
        assert o.mcl is None
        o.finalize()
        assert o.mcl == 6 # len('cmd_99')

    def test_dict(self):
        """
        Tests the `__contains__` and `__getitem__` methods.
        """
        cnt = 25
        api = DummyAPI(cnt)
        assert len(api.cmd) == cnt
        o = self.cls(api)
        o.finalize()
        for cmd in api.cmd:
            key = cli.to_cli(cmd.name)
            assert key in o
            assert o[key] is cmd
            assert cmd.name not in o
            raises(KeyError, getitem, o, cmd.name)