| Index: tools/grit/grit/util_unittest.py
|
| diff --git a/tools/grit/grit/util_unittest.py b/tools/grit/grit/util_unittest.py
|
| new file mode 100755
|
| index 0000000000000000000000000000000000000000..03f8cfef9931bb56880fccd16fd3fea507dfe05c
|
| --- /dev/null
|
| +++ b/tools/grit/grit/util_unittest.py
|
| @@ -0,0 +1,119 @@
|
| +#!/usr/bin/env python
|
| +# Copyright (c) 2012 The Chromium Authors. All rights reserved.
|
| +# Use of this source code is governed by a BSD-style license that can be
|
| +# found in the LICENSE file.
|
| +
|
| +'''Unit test that checks some of util functions.
|
| +'''
|
| +
|
| +import os
|
| +import sys
|
| +if __name__ == '__main__':
|
| + sys.path.append(os.path.join(os.path.dirname(__file__), '..'))
|
| +
|
| +import unittest
|
| +
|
| +from grit import util
|
| +
|
| +
|
| +class UtilUnittest(unittest.TestCase):
|
| + ''' Tests functions from util
|
| + '''
|
| +
|
| + def testNewClassInstance(self):
|
| + # Test short class name with no fully qualified package name
|
| + # Should fail, it is not supported by the function now (as documented)
|
| + cls = util.NewClassInstance('grit.util.TestClassToLoad',
|
| + TestBaseClassToLoad)
|
| + self.failUnless(cls == None)
|
| +
|
| + # Test non existent class name
|
| + cls = util.NewClassInstance('grit.util_unittest.NotExistingClass',
|
| + TestBaseClassToLoad)
|
| + self.failUnless(cls == None)
|
| +
|
| + # Test valid class name and valid base class
|
| + cls = util.NewClassInstance('grit.util_unittest.TestClassToLoad',
|
| + TestBaseClassToLoad)
|
| + self.failUnless(isinstance(cls, TestBaseClassToLoad))
|
| +
|
| + # Test valid class name with wrong hierarchy
|
| + cls = util.NewClassInstance('grit.util_unittest.TestClassNoBase',
|
| + TestBaseClassToLoad)
|
| + self.failUnless(cls == None)
|
| +
|
| + def testCanonicalLanguage(self):
|
| + self.failUnless(util.CanonicalLanguage('en') == 'en')
|
| + self.failUnless(util.CanonicalLanguage('pt_br') == 'pt-BR')
|
| + self.failUnless(util.CanonicalLanguage('pt-br') == 'pt-BR')
|
| + self.failUnless(util.CanonicalLanguage('pt-BR') == 'pt-BR')
|
| + self.failUnless(util.CanonicalLanguage('pt/br') == 'pt-BR')
|
| + self.failUnless(util.CanonicalLanguage('pt/BR') == 'pt-BR')
|
| + self.failUnless(util.CanonicalLanguage('no_no_bokmal') == 'no-NO-BOKMAL')
|
| +
|
| + def testUnescapeHtml(self):
|
| + self.failUnless(util.UnescapeHtml('ϲ') == unichr(1010))
|
| + self.failUnless(util.UnescapeHtml('ꯍ') == unichr(43981))
|
| +
|
| + def testRelativePath(self):
|
| + """ Verify that MakeRelativePath works in some tricky cases."""
|
| +
|
| + def TestRelativePathCombinations(base_path, other_path, expected_result):
|
| + """ Verify that the relative path function works for
|
| + the given paths regardless of whether or not they end with
|
| + a trailing slash."""
|
| + for path1 in [base_path, base_path + os.path.sep]:
|
| + for path2 in [other_path, other_path + os.path.sep]:
|
| + result = util.MakeRelativePath(path1, path2)
|
| + self.failUnless(result == expected_result)
|
| +
|
| + # set-up variables
|
| + root_dir = 'c:%sa' % os.path.sep
|
| + result1 = '..%sabc' % os.path.sep
|
| + path1 = root_dir + 'bc'
|
| + result2 = 'bc'
|
| + path2 = '%s%s%s' % (root_dir, os.path.sep, result2)
|
| + # run the tests
|
| + TestRelativePathCombinations(root_dir, path1, result1)
|
| + TestRelativePathCombinations(root_dir, path2, result2)
|
| +
|
| + def testReadFile(self):
|
| + def Test(data, encoding, expected_result):
|
| + with open('testfile', 'wb') as f:
|
| + f.write(data)
|
| + if util.ReadFile('testfile', encoding) != expected_result:
|
| + print (util.ReadFile('testfile', encoding), expected_result)
|
| + self.failUnless(util.ReadFile('testfile', encoding) == expected_result)
|
| +
|
| + test_std_newline = '\xEF\xBB\xBFabc\ndef' # EF BB BF is UTF-8 BOM
|
| + newlines = ['\n', '\r\n', '\r']
|
| +
|
| + with util.TempDir({}) as tmp_dir:
|
| + with tmp_dir.AsCurrentDir():
|
| + for newline in newlines:
|
| + test = test_std_newline.replace('\n', newline)
|
| + Test(test, util.BINARY, test)
|
| + # RAW_TEXT uses universal newline mode
|
| + Test(test, util.RAW_TEXT, test_std_newline)
|
| + # utf-8 doesn't strip BOM
|
| + Test(test, 'utf-8', test_std_newline.decode('utf-8'))
|
| + # utf-8-sig strips BOM
|
| + Test(test, 'utf-8-sig', test_std_newline.decode('utf-8')[1:])
|
| + # test another encoding
|
| + Test(test, 'cp1252', test_std_newline.decode('cp1252'))
|
| + self.assertRaises(UnicodeDecodeError, Test, '\x80', 'utf-8', None)
|
| +
|
| +
|
| +class TestBaseClassToLoad(object):
|
| + pass
|
| +
|
| +class TestClassToLoad(TestBaseClassToLoad):
|
| + pass
|
| +
|
| +class TestClassNoBase(object):
|
| + pass
|
| +
|
| +
|
| +if __name__ == '__main__':
|
| + unittest.main()
|
| +
|
|
|