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
|
# This file is part of cloud-init. See LICENSE file for license information.
from cloudinit import distros
from cloudinit import util
from cloudinit.tests.helpers import (FilesystemMockingTestCase, mock)
@mock.patch("cloudinit.distros.debian.subp.subp")
class TestDebianApplyLocale(FilesystemMockingTestCase):
def setUp(self):
super(TestDebianApplyLocale, self).setUp()
self.new_root = self.tmp_dir()
self.patchOS(self.new_root)
self.patchUtils(self.new_root)
self.spath = self.tmp_path('etc/default/locale', self.new_root)
cls = distros.fetch("debian")
self.distro = cls("debian", {}, None)
def test_no_rerun(self, m_subp):
"""If system has defined locale, no re-run is expected."""
m_subp.return_value = (None, None)
locale = 'en_US.UTF-8'
util.write_file(self.spath, 'LANG=%s\n' % locale, omode="w")
self.distro.apply_locale(locale, out_fn=self.spath)
m_subp.assert_not_called()
def test_no_regen_on_c_utf8(self, m_subp):
"""If locale is set to C.UTF8, do not attempt to call locale-gen"""
m_subp.return_value = (None, None)
locale = 'C.UTF-8'
util.write_file(self.spath, 'LANG=%s\n' % 'en_US.UTF-8', omode="w")
self.distro.apply_locale(locale, out_fn=self.spath)
self.assertEqual(
[['update-locale', '--locale-file=' + self.spath,
'LANG=%s' % locale]],
[p[0][0] for p in m_subp.call_args_list])
def test_rerun_if_different(self, m_subp):
"""If system has different locale, locale-gen should be called."""
m_subp.return_value = (None, None)
locale = 'en_US.UTF-8'
util.write_file(self.spath, 'LANG=fr_FR.UTF-8', omode="w")
self.distro.apply_locale(locale, out_fn=self.spath)
self.assertEqual(
[['locale-gen', locale],
['update-locale', '--locale-file=' + self.spath,
'LANG=%s' % locale]],
[p[0][0] for p in m_subp.call_args_list])
def test_rerun_if_no_file(self, m_subp):
"""If system has no locale file, locale-gen should be called."""
m_subp.return_value = (None, None)
locale = 'en_US.UTF-8'
self.distro.apply_locale(locale, out_fn=self.spath)
self.assertEqual(
[['locale-gen', locale],
['update-locale', '--locale-file=' + self.spath,
'LANG=%s' % locale]],
[p[0][0] for p in m_subp.call_args_list])
def test_rerun_on_unset_system_locale(self, m_subp):
"""If system has unset locale, locale-gen should be called."""
m_subp.return_value = (None, None)
locale = 'en_US.UTF-8'
util.write_file(self.spath, 'LANG=', omode="w")
self.distro.apply_locale(locale, out_fn=self.spath)
self.assertEqual(
[['locale-gen', locale],
['update-locale', '--locale-file=' + self.spath,
'LANG=%s' % locale]],
[p[0][0] for p in m_subp.call_args_list])
def test_rerun_on_mismatched_keys(self, m_subp):
"""If key is LC_ALL and system has only LANG, rerun is expected."""
m_subp.return_value = (None, None)
locale = 'en_US.UTF-8'
util.write_file(self.spath, 'LANG=', omode="w")
self.distro.apply_locale(locale, out_fn=self.spath, keyname='LC_ALL')
self.assertEqual(
[['locale-gen', locale],
['update-locale', '--locale-file=' + self.spath,
'LC_ALL=%s' % locale]],
[p[0][0] for p in m_subp.call_args_list])
def test_falseish_locale_raises_valueerror(self, m_subp):
"""locale as None or "" is invalid and should raise ValueError."""
with self.assertRaises(ValueError) as ctext_m:
self.distro.apply_locale(None)
m_subp.assert_not_called()
self.assertEqual(
'Failed to provide locale value.', str(ctext_m.exception))
with self.assertRaises(ValueError) as ctext_m:
self.distro.apply_locale("")
m_subp.assert_not_called()
self.assertEqual(
'Failed to provide locale value.', str(ctext_m.exception))
|
