global_config_unittest.py revision 42ffd3d0a235f8ab14af52142176c7bd7f299487 
1#!/usr/bin/python
2
3import os
4import mox
5import types
6import unittest
7
8import common
9from autotest_lib.client.common_lib import autotemp
10from autotest_lib.client.common_lib import global_config
11from autotest_lib.client.common_lib import lsbrelease_utils
12
13
14global_config_ini_contents = """
15[SECTION_A]
16value_1: 6.0
17value_2: hello
18value_3: true
19value_4: FALSE
20value_5: tRuE
21value_6: falsE
22
23[SECTION_B]
24value_1: -5
25value_2: 2.3
26value_3: 0
27value_4: 7
28
29[SECTION_C]
30value_1: nobody@localhost
31
32[SECTION_D]
33value_1: 1
34
35[SECTION_E]
36value_1: 1
37value_2: 2
38value_a: A
39random: 1
40wireless_ssid_1.2.3.4/24: ssid_1
41wireless_ssid_4.3.2.1/16: ssid_2
42"""
43
44moblab_config_ini_contents = """
45[SECTION_C]
46value_1: moblab@remotehost
47
48[SECTION_D]
49value_1: 2
50"""
51
52shadow_config_ini_contents = """
53[SECTION_C]
54value_1: somebody@remotehost
55"""
56
57
58def create_config_files():
59    """Create config files to be used for test."""
60    global_temp = autotemp.tempfile("global", ".ini", text=True)
61    os.write(global_temp.fd, global_config_ini_contents)
62
63    moblab_temp = autotemp.tempfile("moblab", ".ini", text=True)
64    os.write(moblab_temp.fd, moblab_config_ini_contents)
65
66    shadow_temp = autotemp.tempfile("shadow", ".ini", text=True)
67    os.write(shadow_temp.fd, shadow_config_ini_contents)
68
69    return (global_temp, shadow_temp, moblab_temp)
70
71
72class global_config_test(mox.MoxTestBase):
73    """Test class"""
74    # grab the singelton
75    conf = global_config.global_config
76
77    def setUp(self):
78        """Setup config files for test."""
79        super(global_config_test, self).setUp()
80        # set the config files to our test files
81        (self.global_temp, self.shadow_temp,
82                self.moblab_temp) = create_config_files()
83
84        self.conf.set_config_files(self.global_temp.name, self.shadow_temp.name,
85                                   self.moblab_temp.name)
86
87
88    def tearDown(self):
89        """Cleanup and reset config settings."""
90        self.shadow_temp.clean()
91        self.moblab_temp.clean()
92        self.global_temp.clean()
93        self.conf.set_config_files(global_config.DEFAULT_CONFIG_FILE,
94                                   global_config.DEFAULT_SHADOW_FILE,
95                                   global_config.DEFAULT_MOBLAB_FILE)
96
97
98    def test_float(self):
99        """Test converting float value."""
100        val = self.conf.get_config_value("SECTION_A", "value_1", float)
101        self.assertEquals(type(val), types.FloatType)
102        self.assertEquals(val, 6.0)
103
104
105    def test_int(self):
106        """Test converting int value."""
107        val = self.conf.get_config_value("SECTION_B", "value_1", int)
108        self.assertEquals(type(val), types.IntType)
109        self.assertTrue(val < 0)
110        val = self.conf.get_config_value("SECTION_B", "value_3", int)
111        self.assertEquals(val, 0)
112        val = self.conf.get_config_value("SECTION_B", "value_4", int)
113        self.assertTrue(val > 0)
114
115
116    def test_string(self):
117        """Test converting string value."""
118        val = self.conf.get_config_value("SECTION_A", "value_2")
119        self.assertEquals(type(val),types.StringType)
120        self.assertEquals(val, "hello")
121
122
123    def setIsMoblab(self, is_moblab):
124        """Set lsbrelease_utils.is_moblab result.
125
126        @param is_moblab: Value to have lsbrelease_utils.is_moblab to return.
127        """
128        self.mox.StubOutWithMock(lsbrelease_utils, 'is_moblab')
129        lsbrelease_utils.is_moblab().AndReturn(is_moblab)
130
131
132    def test_override_non_moblab(self):
133        """Test value overriding works in non-moblab setup."""
134        self.setIsMoblab(False)
135        self.mox.ReplayAll()
136
137        self.conf.reset_config_values()
138
139        # Confirm shadow config overrides global config.
140        val = self.conf.get_config_value("SECTION_C", "value_1")
141        self.assertEquals(val, "somebody@remotehost")
142
143        # Confirm moblab config should be ignored in non-moblab environment..
144        val = self.conf.get_config_value("SECTION_D", "value_1")
145        self.assertEquals(val, "1")
146
147
148    def test_override_moblab(self):
149        """Test value overriding works in moblab setup."""
150        self.setIsMoblab(True)
151        self.mox.ReplayAll()
152
153        self.conf.reset_config_values()
154
155        # Confirm shadow config overrides both moblab and global config.
156        val = self.conf.get_config_value("SECTION_C", "value_1")
157        self.assertEquals(val, "somebody@remotehost")
158
159        # Confirm moblab config should override global config in moblab.
160        val = self.conf.get_config_value("SECTION_D", "value_1")
161        self.assertEquals(val, "2")
162
163
164    def test_exception(self):
165        """Test exception to be raised on invalid config value."""
166        error = 0
167        try:
168            val = self.conf.get_config_value("SECTION_B", "value_2", int)
169        except:
170            error = 1
171        self.assertEquals(error, 1)
172
173
174    def test_boolean(self):
175        """Test converting boolean value."""
176        val = self.conf.get_config_value("SECTION_A", "value_3", bool)
177        self.assertEquals(val, True)
178        val = self.conf.get_config_value("SECTION_A", "value_4", bool)
179        self.assertEquals(val, False)
180        val = self.conf.get_config_value("SECTION_A", "value_5", bool)
181        self.assertEquals(val, True)
182        val = self.conf.get_config_value("SECTION_A", "value_6", bool)
183        self.assertEquals(val, False)
184
185
186    def test_defaults(self):
187        """Test default value works."""
188        val = self.conf.get_config_value("MISSING", "foo", float, 3.6)
189        self.assertEquals(val, 3.6)
190        val = self.conf.get_config_value("SECTION_A", "novalue", str, "default")
191        self.assertEquals(val, "default")
192
193
194    def test_fallback_key(self):
195        """Test fallback value works."""
196        val = self.conf.get_config_value_with_fallback(
197                "SECTION_A", "value_3", "value_4", bool)
198        self.assertEquals(val, True)
199
200        val = self.conf.get_config_value_with_fallback(
201                "SECTION_A", "not_existing", "value_4", bool)
202        self.assertEquals(val, False)
203
204        val = self.conf.get_config_value_with_fallback(
205                "SECTION_A", "not_existing", "value_4",
206                fallback_section='SECTION_B')
207        self.assertEquals(val, '7')
208
209        self.assertRaises(
210                Exception, self.conf.get_config_value_with_fallback,
211                "SECTION_A", "not_existing", "also_not_existing", bool)
212
213
214    def test_get_config_value_regex(self):
215        """Test get_config_value_regex works."""
216        configs = self.conf.get_config_value_regex('SECTION_E', 'value_\d+',
217                                                   int)
218        self.assertEquals(configs, {'value_1': 1, 'value_2': 2})
219        configs = self.conf.get_config_value_regex('SECTION_E', 'value_.*')
220        self.assertEquals(configs, {'value_1': '1', 'value_2': '2',
221                                    'value_a': 'A'})
222        configs = self.conf.get_config_value_regex('SECTION_E',
223                                                   'wireless_ssid_.*')
224        self.assertEquals(configs, {'wireless_ssid_1.2.3.4/24': 'ssid_1',
225                                    'wireless_ssid_4.3.2.1/16': 'ssid_2'})
226
227
228# this is so the test can be run in standalone mode
229if __name__ == '__main__':
230    """Main"""
231    unittest.main()
232