summaryrefslogtreecommitdiffstats
path: root/tests/test_ipalib/test_request.py
blob: f26c270a7fd243b75dd48aefd34d4ca127ec86eb (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
# Authors:
#   Jason Gerard DeRose <jderose@redhat.com>
#
# Copyright (C) 2008  Red Hat
# see file 'COPYING' for use and warranty contextrmation
#
# 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

"""
Test the `ipalib.request` module.
"""

import threading
import locale
from tests.util import raises, assert_equal
from tests.util import TempDir, dummy_ugettext, dummy_ungettext
from ipalib.constants import OVERRIDE_ERROR
from ipalib import request


def test_ugettext():
    """
    Test the `ipalib.request.ugettext` function.
    """
    f = request.ugettext
    context = request.context
    message = 'Hello, world!'

    # Test with no context.ugettext:
    assert not hasattr(context, 'ugettext')
    assert_equal(f(message), u'Hello, world!')

    # Test with dummy context.ugettext:
    assert not hasattr(context, 'ugettext')
    dummy = dummy_ugettext()
    context.ugettext = dummy
    assert f(message) is dummy.translation
    assert dummy.message is message

    # Cleanup
    del context.ugettext
    assert not hasattr(context, 'ugettext')


def test_ungettext():
    """
    Test the `ipalib.request.ungettext` function.
    """
    f = request.ungettext
    context = request.context
    singular = 'Goose'
    plural = 'Geese'

    # Test with no context.ungettext:
    assert not hasattr(context, 'ungettext')
    assert_equal(f(singular, plural, 1), u'Goose')
    assert_equal(f(singular, plural, 2), u'Geese')

    # Test singular with dummy context.ungettext
    assert not hasattr(context, 'ungettext')
    dummy = dummy_ungettext()
    context.ungettext = dummy
    assert f(singular, plural, 1) is dummy.translation_singular
    assert dummy.singular is singular
    assert dummy.plural is plural
    assert dummy.n == 1
    del context.ungettext
    assert not hasattr(context, 'ungettext')

    # Test plural with dummy context.ungettext
    assert not hasattr(context, 'ungettext')
    dummy = dummy_ungettext()
    context.ungettext = dummy
    assert f(singular, plural, 2) is dummy.translation_plural
    assert dummy.singular is singular
    assert dummy.plural is plural
    assert dummy.n == 2
    del context.ungettext
    assert not hasattr(context, 'ungettext')


def test_set_languages():
    """
    Test the `ipalib.request.set_languages` function.
    """
    f = request.set_languages
    c = request.context
    langs = ('ru', 'en')

    # Test that StandardError is raised if languages has already been set:
    assert not hasattr(c, 'languages')
    c.languages = None
    e = raises(StandardError, f, *langs)
    assert str(e) == OVERRIDE_ERROR % ('context', 'languages', None, langs)
    del c.languages

    # Test setting the languages:
    assert not hasattr(c, 'languages')
    f(*langs)
    assert c.languages == langs
    del c.languages

    # Test setting language from locale.getdefaultlocale()
    assert not hasattr(c, 'languages')
    f()
    assert c.languages == locale.getdefaultlocale()[:1]
    del c.languages
    assert not hasattr(c, 'languages')


def test_create_translation():
    """
    Test the `ipalib.request.create_translation` function.
    """
    f = request.create_translation
    c = request.context
    t = TempDir()

    # Test that StandardError is raised if ugettext or ungettext:
    assert not (hasattr(c, 'ugettext') or hasattr(c, 'ungettext'))
    for name in ('ugettext', 'ungettext'):
        setattr(c, name, None)
        e = raises(StandardError, f, 'ipa', None)
        assert str(e) == (
            'create_translation() already called in thread %r' %
            threading.currentThread().getName()
        )
        delattr(c, name)

    # Test using default language:
    assert not hasattr(c, 'ugettext')
    assert not hasattr(c, 'ungettext')
    assert not hasattr(c, 'languages')
    f('ipa', t.path)
    assert hasattr(c, 'ugettext')
    assert hasattr(c, 'ungettext')
    assert c.languages == locale.getdefaultlocale()[:1]
    del c.ugettext
    del c.ungettext
    del c.languages

    # Test using explicit languages:
    langs = ('de', 'es')
    f('ipa', t.path, *langs)
    assert hasattr(c, 'ugettext')
    assert hasattr(c, 'ungettext')
    assert c.languages == langs
    del c.ugettext
    del c.ungettext
    del c.languages