summaryrefslogtreecommitdiffstats
path: root/ipalib/tests/test_public.py
blob: 6bff9d8953a62817b8a611ea6411c13b0508ba88 (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
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
# 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.public` module.
"""

from tstutil import raises, getitem, no_set, no_del, read_only
from ipalib import public, plugable, errors


def test_RULE_FLAG():
    assert public.RULE_FLAG == 'validation_rule'


def test_rule():
    flag = public.RULE_FLAG
    rule = public.rule
    def my_func():
        pass
    assert not hasattr(my_func, flag)
    rule(my_func)
    assert getattr(my_func, flag) is True
    @rule
    def my_func2():
        pass
    assert getattr(my_func2, flag) is True


def test_is_rule():
    is_rule = public.is_rule
    flag = public.RULE_FLAG

    class no_call(object):
        def __init__(self, value):
            if value is not None:
                assert value in (True, False)
                setattr(self, flag, value)

    class call(no_call):
        def __call__(self):
            pass

    assert is_rule(call(True))
    assert not is_rule(no_call(True))
    assert not is_rule(call(False))
    assert not is_rule(call(None))


class ClassChecker(object):
    __cls = None
    __subcls = None

    def __get_cls(self):
        if self.__cls is None:
            self.__cls = self._cls
        return self.__cls
    cls = property(__get_cls)

    def __get_subcls(self):
        if self.__subcls is None:
            self.__subcls = self.get_subcls()
        return self.__subcls
    subcls = property(__get_subcls)

    def get_subcls(self):
        raise NotImplementedError(
            self.__class__.__name__,
            'get_subcls()'
        )


class test_option(ClassChecker):
    _cls = public.option

    def get_subcls(self):
        rule = public.rule
        class int_opt(self.cls):
            type = int
            @rule
            def rule_0(self, value):
                if value == 0:
                    return 'cannot be 0'
            @rule
            def rule_1(self, value):
                if value == 1:
                    return 'cannot be 1'
            @rule
            def rule_2(self, value):
                if value == 2:
                    return 'cannot be 2'
        return int_opt

    def test_class(self):
        """
        Perform some tests on the class (not an instance).
        """
        #assert issubclass(cls, plugable.ReadOnly)
        assert type(self.cls.rules) is property

    def test_normalize(self):
        assert 'normalize' in self.cls.__public__
        o = self.subcls()
        # Test with values that can't be converted:
        nope = (
            '7.0'
            'whatever',
            object,
            None,
        )
        for val in nope:
            e = raises(errors.NormalizationError, o.normalize, val)
            assert isinstance(e, errors.ValidationError)
            assert e.name == 'int_opt'
            assert e.value == val
            assert e.error == "not <type 'int'>"
            assert e.type is int
        # Test with values that can be converted:
        okay = (
            7,
            7.0,
            7.2,
            7L,
            '7',
            ' 7 ',
        )
        for val in okay:
            assert o.normalize(val) == 7

    def test_validate(self):
        """
        Test the validate method.
        """
        assert 'validate' in self.cls.__public__
        o = self.subcls()
        o.validate(9)
        for i in xrange(3):
            e = raises(errors.RuleError, o.validate, i)
            assert e.error == 'cannot be %d' % i
            assert e.value == i

    def test_rules(self):
        """
        Test the rules property.
        """
        o = self.subcls()
        assert len(o.rules) == 3
        def get_rule(i):
            return getattr(o, 'rule_%d' % i)
        rules = tuple(get_rule(i) for i in xrange(3))
        assert o.rules == rules

    def test_default(self):
        assert 'default' in self.cls.__public__
        assert self.cls().default() is None





def test_cmd():
    cls = public.cmd
    assert issubclass(cls, plugable.Plugin)


def test_obj():
    cls = public.obj
    assert issubclass(cls, plugable.Plugin)



def test_attr():
    cls = public.attr
    assert issubclass(cls, plugable.Plugin)

    class api(object):
        obj = dict(user='the user obj')

    class user_add(cls):
        pass

    i = user_add()
    assert read_only(i, 'obj_name') == 'user'
    assert read_only(i, 'attr_name') == 'add'
    assert read_only(i, 'obj') is None
    i.finalize(api)
    assert read_only(i, 'api') is api
    assert read_only(i, 'obj') == 'the user obj'


def test_mthd():
    cls = public.mthd
    assert issubclass(cls, public.attr)
    assert issubclass(cls, public.cmd)


def test_prop():
    cls = public.prop
    assert issubclass(cls, public.attr)


def test_PublicAPI():
    cls = public.PublicAPI
    assert issubclass(cls, plugable.API)

    api = cls()

    class cmd1(public.cmd):
        pass
    api.register(cmd1)

    class cmd2(public.cmd):
        pass
    api.register(cmd2)

    api()