Revision cdeadadc kamaki/cli/test.py

b/kamaki/cli/test.py
279 279
        GL.assert_called_once_with('my logger name')
280 280

  
281 281

  
282
class UtilsMethods(TestCase):
283

  
284
    def assert_dicts_are_equal(self, d1, d2):
285
        for k, v in d1.items():
286
            self.assertTrue(k in d2)
287
            if isinstance(v, dict):
288
                self.assert_dicts_are_equal(v, d2[k])
289
            else:
290
                self.assertEqual(unicode(v), unicode(d2[k]))
291

  
292
    def test_guess_mime_type(self):
293
        from kamaki.cli.utils import guess_mime_type
294
        from mimetypes import guess_type
295
        for args in product(
296
                ('file.txt', 'file.png', 'file.zip', 'file.gz', None, 'X'),
297
                ('a type', None),
298
                ('an/encoding', None)):
299
            filename, ctype, cencoding = args
300
            if filename:
301
                exp_type, exp_enc = guess_type(filename)
302
                self.assertEqual(
303
                    guess_mime_type(*args),
304
                    (exp_type or ctype, exp_enc or cencoding))
305
            else:
306
                self.assertRaises(AssertionError, guess_mime_type, *args)
307

  
308
    def test_pretty_keys(self):
309
        from kamaki.cli.utils import pretty_keys
310
        for args, exp in (
311
                (
312
                    ({'k1': 'v1', 'k1_k2': 'v2'}, ),
313
                    {'k1': 'v1', 'k1 k2': 'v2'}),
314
                (
315
                    ({'k1': 'v1', 'k1_k2': 'v2'}, '1'),
316
                    {'k': 'v1', 'k _k2': 'v2'}),
317
                (
318
                    ({'k1_k2': 'v1', 'k1': {'k2': 'v2', 'k2_k3': 'v3'}}, ),
319
                    {'k1 k2': 'v1', 'k1': {'k2': 'v2', 'k2_k3': 'v3'}}),
320
                (
321
                    (
322
                        {'k1_k2': 'v1', 'k1': {'k2': 'v2', 'k2_k3': 'v3'}},
323
                        '_',
324
                        True),
325
                    {'k1 k2': 'v1', 'k1': {'k2': 'v2', 'k2 k3': 'v3'}}),
326
                (
327
                    (
328
                        {
329
                            'k1_k2': {'k_1': 'v_1', 'k_2': {'k_3': 'v_3'}},
330
                            'k1': {'k2': 'v2', 'k2_k3': 'v3'}},
331
                        '_',
332
                        True),
333
                    {
334
                        'k1 k2': {'k 1': 'v_1', 'k 2': {'k 3': 'v_3'}},
335
                        'k1': {'k2': 'v2', 'k2 k3': 'v3'}}),
336
                (
337
                    (
338
                        {
339
                            'k1_k2': {'k_1': 'v_1', 'k_2': {'k_3': 'v_3'}},
340
                            'k1': {'k2': 'v2', 'k2_k3': 'v3'}},
341
                        '1',
342
                        True),
343
                    {
344
                        'k _k2': {'k_': 'v_1', 'k_2': {'k_3': 'v_3'}},
345
                        'k': {'k2': 'v2', 'k2_k3': 'v3'}})
346
            ):
347
            initial = dict(args[0])
348
            self.assert_dicts_are_equal(pretty_keys(*args), exp)
349
            self.assert_dicts_are_equal(initial, args[0])
350

  
351

  
282 352
#  TestCase auxiliary methods
283 353

  
284 354
def runTestCase(cls, test_name, args=[], failure_collector=[]):

Also available in: Unified diff