Home | History | Annotate | Download | only in grit 
      1 #!/usr/bin/env python
      2 # Copyright (c) 2012 The Chromium Authors. All rights reserved.
      3 # Use of this source code is governed by a BSD-style license that can be
      4 # found in the LICENSE file.
      5 
      6 '''Unit test that checks some of util functions.
      7 '''
      8 
      9 import os
     10 import sys
     11 if __name__ == '__main__':
     12   sys.path.append(os.path.join(os.path.dirname(__file__), '..'))
     13 
     14 import unittest
     15 
     16 from grit import util
     17 
     18 
     19 class UtilUnittest(unittest.TestCase):
     20   ''' Tests functions from util
     21   '''
     22 
     23   def testNewClassInstance(self):
     24     # Test short class name with no fully qualified package name
     25     # Should fail, it is not supported by the function now (as documented)
     26     cls = util.NewClassInstance('grit.util.TestClassToLoad',
     27                                 TestBaseClassToLoad)
     28     self.failUnless(cls == None)
     29 
     30     # Test non existent class name
     31     cls = util.NewClassInstance('grit.util_unittest.NotExistingClass',
     32                                 TestBaseClassToLoad)
     33     self.failUnless(cls == None)
     34 
     35     # Test valid class name and valid base class
     36     cls = util.NewClassInstance('grit.util_unittest.TestClassToLoad',
     37                                 TestBaseClassToLoad)
     38     self.failUnless(isinstance(cls, TestBaseClassToLoad))
     39 
     40     # Test valid class name with wrong hierarchy
     41     cls = util.NewClassInstance('grit.util_unittest.TestClassNoBase',
     42                                 TestBaseClassToLoad)
     43     self.failUnless(cls == None)
     44 
     45   def testCanonicalLanguage(self):
     46     self.failUnless(util.CanonicalLanguage('en') == 'en')
     47     self.failUnless(util.CanonicalLanguage('pt_br') == 'pt-BR')
     48     self.failUnless(util.CanonicalLanguage('pt-br') == 'pt-BR')
     49     self.failUnless(util.CanonicalLanguage('pt-BR') == 'pt-BR')
     50     self.failUnless(util.CanonicalLanguage('pt/br') == 'pt-BR')
     51     self.failUnless(util.CanonicalLanguage('pt/BR') == 'pt-BR')
     52     self.failUnless(util.CanonicalLanguage('no_no_bokmal') == 'no-NO-BOKMAL')
     53 
     54   def testUnescapeHtml(self):
     55     self.failUnless(util.UnescapeHtml('ϲ') == unichr(1010))
     56     self.failUnless(util.UnescapeHtml('ꯍ') == unichr(43981))
     57 
     58   def testRelativePath(self):
     59     """ Verify that MakeRelativePath works in some tricky cases."""
     60 
     61     def TestRelativePathCombinations(base_path, other_path, expected_result):
     62       """ Verify that the relative path function works for
     63       the given paths regardless of whether or not they end with
     64       a trailing slash."""
     65       for path1 in [base_path, base_path + os.path.sep]:
     66         for path2 in [other_path, other_path + os.path.sep]:
     67           result = util.MakeRelativePath(path1, path2)
     68           self.failUnless(result == expected_result)
     69 
     70     # set-up variables
     71     root_dir = 'c:%sa' % os.path.sep
     72     result1 = '..%sabc' % os.path.sep
     73     path1 = root_dir + 'bc'
     74     result2 = 'bc'
     75     path2 = '%s%s%s' % (root_dir, os.path.sep, result2)
     76     # run the tests
     77     TestRelativePathCombinations(root_dir, path1, result1)
     78     TestRelativePathCombinations(root_dir, path2, result2)
     79 
     80   def testReadFile(self):
     81     def Test(data, encoding, expected_result):
     82       with open('testfile', 'wb') as f:
     83         f.write(data)
     84       if util.ReadFile('testfile', encoding) != expected_result:
     85         print (util.ReadFile('testfile', encoding), expected_result)
     86       self.failUnless(util.ReadFile('testfile', encoding) == expected_result)
     87 
     88     test_std_newline = '\xEF\xBB\xBFabc\ndef'  # EF BB BF is UTF-8 BOM
     89     newlines = ['\n', '\r\n', '\r']
     90 
     91     with util.TempDir({}) as tmp_dir:
     92       with tmp_dir.AsCurrentDir():
     93         for newline in newlines:
     94           test = test_std_newline.replace('\n', newline)
     95           Test(test, util.BINARY, test)
     96           # RAW_TEXT uses universal newline mode
     97           Test(test, util.RAW_TEXT, test_std_newline)
     98           # utf-8 doesn't strip BOM
     99           Test(test, 'utf-8', test_std_newline.decode('utf-8'))
    100           # utf-8-sig strips BOM
    101           Test(test, 'utf-8-sig', test_std_newline.decode('utf-8')[1:])
    102           # test another encoding
    103           Test(test, 'cp1252', test_std_newline.decode('cp1252'))
    104         self.assertRaises(UnicodeDecodeError, Test, '\x80', 'utf-8', None)
    105 
    106 
    107 class TestBaseClassToLoad(object):
    108   pass
    109 
    110 class TestClassToLoad(TestBaseClassToLoad):
    111   pass
    112 
    113 class TestClassNoBase(object):
    114   pass
    115 
    116 
    117 if __name__ == '__main__':
    118   unittest.main()
    119 
    120