File: test_extract.py

package info (click to toggle)
python-django-babel 0.6.2-7
  • links: PTS, VCS
  • area: main
  • in suites: sid, trixie
  • size: 264 kB
  • sloc: python: 662; makefile: 162
file content (220 lines) | stat: -rw-r--r-- 8,967 bytes parent folder | download | duplicates (4)
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
# -*- coding: utf-8 -*-
import unittest

import pytest

from babel.messages import extract
from babel._compat import BytesIO

import django
from django_babel.extract import extract_django


default_keys = extract.DEFAULT_KEYWORDS.keys()


class ExtractDjangoTestCase(unittest.TestCase):
    # TODO: translator comments are not yet supported!

    def test_extract_no_tags(self):
        buf = BytesIO(b'nothing')
        messages = list(extract_django(buf, default_keys, [], {}))
        self.assertEqual([], messages)

    def test_extract_simple_double_quotes(self):
        buf = BytesIO(b'{% trans "Bunny" %}')
        messages = list(extract_django(buf, default_keys, [], {}))
        self.assertEqual([(1, None, u'Bunny', [])], messages)

    def test_extract_simple_single_quotes(self):
        buf = BytesIO(b"{% trans 'Bunny' %}")
        messages = list(extract_django(buf, default_keys, [], {}))
        self.assertEqual([(1, None, u'Bunny', [])], messages)

    def test_extract_simple_with_context_single_quotes(self):
        buf = BytesIO(b"{% trans 'Bunny' context 'carrot' %}")
        messages = list(extract_django(buf, default_keys, [], {}))
        self.assertEqual([(1, 'pgettext',
                           [u'carrot', u'Bunny'], [])], messages)

    def test_extract_simple_with_context_double_quotes(self):
        buf = BytesIO(b"{% trans 'Bunny' context \"carrot\" %}")
        messages = list(extract_django(buf, default_keys, [], {}))
        self.assertEqual([(1, 'pgettext',
                           [u'carrot', u'Bunny'], [])], messages)

    def test_extract_simple_with_context_with_single_quotes(self):
        buf = BytesIO(b"{% trans 'Bunny' context \"'carrot'\" %}")
        messages = list(extract_django(buf, default_keys, [], {}))
        self.assertEqual([(1, 'pgettext',
                           [u'\'carrot\'', u'Bunny'], [])], messages)

    def test_extract_simple_with_context_with_double_quotes(self):
        buf = BytesIO(b"{% trans 'Bunny' context '\"carrot\"' %}")
        messages = list(extract_django(buf, default_keys, [], {}))
        self.assertEqual([(1, 'pgettext',
                           [u'"carrot"', u'Bunny'], [])], messages)

    def test_extract_var(self):
        buf = BytesIO(b'{% blocktrans %}{{ anton }}{% endblocktrans %}')
        messages = list(extract_django(buf, default_keys, [], {}))
        self.assertEqual([(1, None, u'%(anton)s', [])], messages)

    def test_extract_filter_with_filter(self):
        test_tmpl = (
            b'{% blocktrans with berta=anton|lower %}'
            b'{{ berta }}{% endblocktrans %}'
        )
        buf = BytesIO(test_tmpl)
        messages = list(extract_django(buf, default_keys, [], {}))
        self.assertEqual([(1, None, u'%(berta)s', [])], messages)

    def test_extract_with_interpolation(self):
        buf = BytesIO(b'{% blocktrans %}xxx{{ anton }}xxx{% endblocktrans %}')
        messages = list(extract_django(buf, default_keys, [], {}))
        self.assertEqual([(1, None, u'xxx%(anton)sxxx', [])], messages)

    def test_extract_unicode(self):
        buf = BytesIO(u'{% trans "@ſðæ314“ſſ¶ÐĐÞ→SÆ^ĸŁ" %}'.encode('utf8'))
        messages = list(extract_django(buf, default_keys, [], {}))
        self.assertEqual([(1, None, u'@ſðæ314“ſſ¶ÐĐÞ→SÆ^ĸŁ', [])], messages)

    def test_extract_unicode_blocktrans(self):
        test_tmpl = u'{% blocktrans %}@ſðæ314“ſſ¶ÐĐÞ→SÆ^ĸŁ{% endblocktrans %}'
        buf = BytesIO(test_tmpl.encode('utf8'))
        messages = list(extract_django(buf, default_keys, [], {}))
        self.assertEqual([(1, None, u'@ſðæ314“ſſ¶ÐĐÞ→SÆ^ĸŁ', [])], messages)

    # TODO: Yet expected to not extract the comments.
    def test_extract_ignored_comment(self):
        buf = BytesIO(
            b'{# ignored comment #1 #}{% trans "Translatable literal #9a" %}',
        )
        messages = list(extract_django(buf, default_keys, [], {}))
        self.assertEqual(
            [(1, None, u'Translatable literal #9a', [])], messages
        )

    def test_extract_ignored_comment2(self):
        test_tmpl = (
            b'{# Translators: ignored i18n comment #1 #}'
            b'{% trans "Translatable literal #9a" %}'
        )
        buf = BytesIO(test_tmpl)
        messages = list(extract_django(buf, default_keys, [], {}))
        self.assertEqual(
            [(1, None, u'Translatable literal #9a', [])], messages
        )

    def test_extract_valid_comment(self):
        test_tmpl = (
            b'{# ignored comment #6 #}'
            b'{% trans "Translatable literal #9h" %}'
            b'{# Translators: valid i18n comment #7 #}'
        )
        buf = BytesIO(test_tmpl)
        messages = list(extract_django(buf, default_keys, [], {}))
        self.assertEqual(
            [(1, None, u'Translatable literal #9h', [])], messages
        )

    def test_extract_singular_form(self):
        test_tmpl = (
            b'{% blocktrans count counter=number %}'
            b'singular{% plural %}{{ counter }} plural'
            b'{% endblocktrans %}'
        )
        buf = BytesIO(test_tmpl)
        messages = list(extract_django(buf, default_keys, [], {}))
        self.assertEqual(
            [(1, 'ngettext', (u'singular', u'%(counter)s plural'), [])],
            messages
        )

    def test_trans_blocks_must_not_include_other_block_tags(self):
        buf = BytesIO(b'{% blocktrans %}{% other_tag %}{% endblocktrans %}')
        gen = extract_django(buf, default_keys, [], {})
        with pytest.raises(SyntaxError):
            next(gen)

    def test_extract_var_other(self):
        buf = BytesIO(b'{{ book }}')
        messages = list(extract_django(buf, default_keys, [], {}))
        self.assertEqual([], messages)

    def test_extract_filters_default_translatable(self):
        buf = BytesIO(b'{{ book.author|default:_("Unknown") }}')
        messages = list(extract_django(buf, default_keys, [], {}))
        self.assertEqual([(1, None, u'Unknown', [])], messages)

    def test_extract_filters_default_translatable_single_quotes(self):
        buf = BytesIO(b"{{ book.author|default:_('Unknown') }}")
        messages = list(extract_django(buf, default_keys, [], {}))
        self.assertEqual([(1, None, u'Unknown', [])], messages)

    def test_extract_constant_single_quotes(self):
        buf = BytesIO(b"{{ _('constant') }}")
        messages = list(extract_django(buf, default_keys, [], {}))
        self.assertEqual([(1, None, u'constant', [])], messages)

    def test_extract_constant_double_quotes(self):
        buf = BytesIO(b'{{ _("constant") }}')
        messages = list(extract_django(buf, default_keys, [], {}))
        self.assertEqual([(1, None, u'constant', [])], messages)

    def test_extract_constant_block(self):
        buf = BytesIO(b'{% _("constant") %}')
        messages = list(extract_django(buf, default_keys, [], {}))
        self.assertEqual([(1, None, u'constant', [])], messages)

    def test_extract_constant_in_block(self):
        test_tmpl = (
            b'{% blocktrans foo=_("constant") %}{{ foo }}{% endblocktrans %}'
        )
        buf = BytesIO(test_tmpl)
        messages = list(extract_django(buf, default_keys, [], {}))
        self.assertEqual(
            [(1, None, u'constant', []), (1, None, u'%(foo)s', [])],
            messages,
        )

    def test_extract_context_in_block(self):
        test_tmpl = (
            b'{% blocktrans context "banana" %}{{ foo }}{% endblocktrans %}'
        )
        buf = BytesIO(test_tmpl)
        messages = list(extract_django(buf, default_keys, [], {}))
        self.assertEqual(
            [(1, 'pgettext', [u'banana', u'%(foo)s'], [])],
            messages,
        )

    def test_extract_context_in_plural_block(self):
        test_tmpl = (
            b'{% blocktrans context "banana" %}{{ foo }}'
            b'{% plural %}{{ bar }}{% endblocktrans %}'
        )
        buf = BytesIO(test_tmpl)
        messages = list(extract_django(buf, default_keys, [], {}))
        self.assertEqual(
            [(1, 'npgettext', [u'banana', u'%(foo)s', u'%(bar)s'], [])],
            messages,
        )

    def test_blocktrans_with_whitespace_not_trimmed(self):
        test_tmpl = (
            b'{% blocktrans %}\n\tfoo\n\tbar\n{% endblocktrans %}'
        )
        buf = BytesIO(test_tmpl)
        messages = list(extract_django(buf, default_keys, [], {}))
        self.assertEqual([(4, None, u'\n\tfoo\n\tbar\n', [])], messages)

    @pytest.mark.skipif(django.VERSION < (1, 7),
                        reason='Trimmed whitespace is a Django >= 1.7 feature')
    def test_blocktrans_with_whitespace_trimmed(self):
        test_tmpl = (
            b'{% blocktrans trimmed %}\n\tfoo\n\tbar\n{% endblocktrans %}'
        )
        buf = BytesIO(test_tmpl)
        messages = list(extract_django(buf, default_keys, [], {}))
        self.assertEqual([(4, None, u'foo bar', [])], messages)