| 1 | #!/usr/bin/env python2
|
| 2 | from __future__ import print_function
|
| 3 | """
|
| 4 | fastfunc_test.py: Tests for fastfunc.c
|
| 5 | """
|
| 6 | import unittest
|
| 7 |
|
| 8 | from mycpp.mylib import log
|
| 9 |
|
| 10 | import fastfunc # module under test
|
| 11 |
|
| 12 | _ = log
|
| 13 |
|
| 14 |
|
| 15 | class FastfuncTest(unittest.TestCase):
|
| 16 |
|
| 17 | def testJ8Encode(self):
|
| 18 | s = 'hello \xff \x01 ' + u'\u03bc " \''.encode('utf-8')
|
| 19 | #print(s)
|
| 20 |
|
| 21 | x = fastfunc.J8EncodeString(s, 0)
|
| 22 | print(x)
|
| 23 |
|
| 24 | x = fastfunc.J8EncodeString(s, 1)
|
| 25 | print(x)
|
| 26 |
|
| 27 | def testShellEncode(self):
|
| 28 | s = 'hello \xff \x01 ' + u'\u03bc " \''.encode('utf-8')
|
| 29 | #print(s)
|
| 30 |
|
| 31 | x = fastfunc.ShellEncodeString(s, 0)
|
| 32 | print(x)
|
| 33 |
|
| 34 | x = fastfunc.ShellEncodeString(s, 1)
|
| 35 | print(x)
|
| 36 |
|
| 37 | def testUtf8(self):
|
| 38 | s = 'hi \xff'
|
| 39 | self.assertEqual(True, fastfunc.PartIsUtf8(s, 0, 3))
|
| 40 | self.assertEqual(False, fastfunc.PartIsUtf8(s, 3, 4))
|
| 41 |
|
| 42 | def testUtf8Decode(self):
|
| 43 | # interface is:
|
| 44 | # def Utf8DecodeOne(s: str, start: int) -> (codepoint_or_error: int, bytes_read: int)
|
| 45 | # If codepoint_or_error < 0, then there was a decoding error whose absolute
|
| 46 | # integer value is listed in the Utf8Error enum.
|
| 47 |
|
| 48 | s = 'h\xE2\xA1\x80\xC5\x81'
|
| 49 | self.assertEqual((ord('h'), 1), fastfunc.Utf8DecodeOne(s, 0))
|
| 50 | self.assertEqual((0x2840, 3), fastfunc.Utf8DecodeOne(s, 1))
|
| 51 | self.assertEqual((0x141, 2), fastfunc.Utf8DecodeOne(s, 4))
|
| 52 |
|
| 53 | # UTF8_ERR_OVERLONG = 1
|
| 54 | self.assertEqual((-1, 2), fastfunc.Utf8DecodeOne("\xC1\x81", 0))
|
| 55 |
|
| 56 | # UTF8_ERR_SURROGATE = 2
|
| 57 | self.assertEqual((-2, 3), fastfunc.Utf8DecodeOne("\xED\xBF\x80", 0))
|
| 58 |
|
| 59 | # UTF8_ERR_TOO_LARGE = 3
|
| 60 | self.assertEqual((-3, 4), fastfunc.Utf8DecodeOne("\xF4\xA0\x80\x80", 0))
|
| 61 |
|
| 62 | # UTF8_ERR_BAD_ENCODING = 4
|
| 63 | self.assertEqual((-4, 2), fastfunc.Utf8DecodeOne("\xC2\xFF", 0))
|
| 64 |
|
| 65 | # UTF8_ERR_TRUNCATED_BYTES = 5
|
| 66 | self.assertEqual((-5, 1), fastfunc.Utf8DecodeOne("\xC2", 0))
|
| 67 |
|
| 68 | def testCanOmit(self):
|
| 69 | self.assertEqual(True, fastfunc.CanOmitQuotes('foo'))
|
| 70 | self.assertEqual(False, fastfunc.CanOmitQuotes('foo bar'))
|
| 71 | self.assertEqual(True, fastfunc.CanOmitQuotes('my-dir/my_file.cc'))
|
| 72 | self.assertEqual(False, fastfunc.CanOmitQuotes(''))
|
| 73 |
|
| 74 | self.assertEqual(False, fastfunc.CanOmitQuotes('true'))
|
| 75 | self.assertEqual(False, fastfunc.CanOmitQuotes('false'))
|
| 76 | self.assertEqual(False, fastfunc.CanOmitQuotes('null'))
|
| 77 |
|
| 78 | self.assertEqual(True, fastfunc.CanOmitQuotes('truez'))
|
| 79 | self.assertEqual(True, fastfunc.CanOmitQuotes('nul'))
|
| 80 |
|
| 81 |
|
| 82 | if __name__ == '__main__':
|
| 83 | unittest.main()
|