OLD | NEW |
1 #!/usr/bin/env python | 1 #!/usr/bin/env python |
2 # Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 3 # Use of this source code is governed by a BSD-style license that can be |
4 # found in the LICENSE file. | 4 # found in the LICENSE file. |
5 | 5 |
6 '''Unit tests for grit.format.data_pack''' | 6 '''Unit tests for grit.format.data_pack''' |
7 | 7 |
8 | 8 |
9 import os | 9 import os |
10 import sys | 10 import sys |
(...skipping 10 matching lines...) Expand all Loading... |
21 expected = ( | 21 expected = ( |
22 '\x04\x00\x00\x00' # header(version | 22 '\x04\x00\x00\x00' # header(version |
23 '\x04\x00\x00\x00' # no. entries, | 23 '\x04\x00\x00\x00' # no. entries, |
24 '\x01' # encoding) | 24 '\x01' # encoding) |
25 '\x01\x00\x27\x00\x00\x00' # index entry 1 | 25 '\x01\x00\x27\x00\x00\x00' # index entry 1 |
26 '\x04\x00\x27\x00\x00\x00' # index entry 4 | 26 '\x04\x00\x27\x00\x00\x00' # index entry 4 |
27 '\x06\x00\x33\x00\x00\x00' # index entry 6 | 27 '\x06\x00\x33\x00\x00\x00' # index entry 6 |
28 '\x0a\x00\x3f\x00\x00\x00' # index entry 10 | 28 '\x0a\x00\x3f\x00\x00\x00' # index entry 10 |
29 '\x00\x00\x3f\x00\x00\x00' # extra entry for the size of last | 29 '\x00\x00\x3f\x00\x00\x00' # extra entry for the size of last |
30 'this is id 4this is id 6') # data | 30 'this is id 4this is id 6') # data |
31 input = { 1: "", 4: "this is id 4", 6: "this is id 6", 10: "" } | 31 input = {1: '', 4: 'this is id 4', 6: 'this is id 6', 10: ''} |
32 output = data_pack.WriteDataPackToString(input, data_pack.UTF8) | 32 output = data_pack.WriteDataPackToString(input, data_pack.UTF8) |
33 self.failUnless(output == expected) | 33 self.failUnless(output == expected) |
34 | 34 |
| 35 def testRePackUnittest(self): |
| 36 expected_with_whitelist = { |
| 37 1: 'Never gonna', 10: 'give you up', 20: 'Never gonna let', |
| 38 30: 'you down', 40: 'Never', 50: 'gonna run around and', |
| 39 60: 'desert you'} |
| 40 expected_without_whitelist = { |
| 41 1: 'Never gonna', 10: 'give you up', 20: 'Never gonna let', 65: 'Close', |
| 42 30: 'you down', 40: 'Never', 50: 'gonna run around and', 4: 'click', |
| 43 60: 'desert you', 6: 'chirr', 32: 'oops, try again', 70: 'Awww, snap!'} |
| 44 inputs = [{1: 'Never gonna', 4: 'click', 6: 'chirr', 10: 'give you up'}, |
| 45 {20: 'Never gonna let', 30: 'you down', 32: 'oops, try again'}, |
| 46 {40: 'Never', 50: 'gonna run around and', 60: 'desert you'}, |
| 47 {65: 'Close', 70: 'Awww, snap!'}] |
| 48 whitelist = [1, 10, 20, 30, 40, 50, 60] |
| 49 inputs = [data_pack.DataPackContents(input, data_pack.UTF8) for input |
| 50 in inputs] |
| 51 |
| 52 # RePack using whitelist |
| 53 output, _ = data_pack.RePackFromDataPackStrings(inputs, whitelist) |
| 54 self.assertDictEqual(expected_with_whitelist, output, |
| 55 'Incorrect resource output') |
| 56 |
| 57 # RePack a None whitelist |
| 58 output, _ = data_pack.RePackFromDataPackStrings(inputs, None) |
| 59 self.assertDictEqual(expected_without_whitelist, output, |
| 60 'Incorrect resource output') |
| 61 |
35 | 62 |
36 if __name__ == '__main__': | 63 if __name__ == '__main__': |
37 unittest.main() | 64 unittest.main() |
OLD | NEW |