Index: tools/grit/grit/format/data_pack_unittest.py |
diff --git a/tools/grit/grit/format/data_pack_unittest.py b/tools/grit/grit/format/data_pack_unittest.py |
index 28a7b298fb5c2bf7fd3d6234609a963e090d1fcb..f6e9edcfcb6c99799e76d8c4230c3fddb66793aa 100755 |
--- a/tools/grit/grit/format/data_pack_unittest.py |
+++ b/tools/grit/grit/format/data_pack_unittest.py |
@@ -17,8 +17,8 @@ |
class FormatDataPackUnittest(unittest.TestCase): |
- def testReadDataPackV4(self): |
- expected_data = ( |
+ def testWriteDataPack(self): |
+ expected = ( |
'\x04\x00\x00\x00' # header(version |
'\x04\x00\x00\x00' # no. entries, |
'\x01' # encoding) |
@@ -28,42 +28,9 @@ |
'\x0a\x00\x3f\x00\x00\x00' # index entry 10 |
'\x00\x00\x3f\x00\x00\x00' # extra entry for the size of last |
'this is id 4this is id 6') # data |
- expected_resources = { |
- 1: '', |
- 4: 'this is id 4', |
- 6: 'this is id 6', |
- 10: '', |
- } |
- expected_data_pack = data_pack.DataPackContents( |
- expected_resources, data_pack.UTF8) |
- loaded = data_pack.ReadDataPackFromString(expected_data) |
- self.assertEquals(loaded, expected_data_pack) |
- |
- def testReadWriteDataPackV5(self): |
- expected_data = ( |
- '\x05\x00\x00\x00' # version |
- '\x01\x00\x00\x00' # encoding & padding |
- '\x03\x00' # resource_count |
- '\x01\x00' # alias_count |
- '\x01\x00\x28\x00\x00\x00' # index entry 1 |
- '\x04\x00\x28\x00\x00\x00' # index entry 4 |
- '\x06\x00\x34\x00\x00\x00' # index entry 6 |
- '\x00\x00\x40\x00\x00\x00' # extra entry for the size of last |
- '\x0a\x00\x01\x00' # alias table |
- 'this is id 4this is id 6') # data |
- expected_resources = { |
- 1: '', |
- 4: 'this is id 4', |
- 6: 'this is id 6', |
- 10: 'this is id 4', |
- } |
- data = data_pack.WriteDataPackToString(expected_resources, data_pack.UTF8) |
- self.assertEquals(data, expected_data) |
- |
- expected_data_pack = data_pack.DataPackContents( |
- expected_resources, data_pack.UTF8) |
- loaded = data_pack.ReadDataPackFromString(expected_data) |
- self.assertEquals(loaded, expected_data_pack) |
+ input = {1: '', 4: 'this is id 4', 6: 'this is id 6', 10: ''} |
+ output = data_pack.WriteDataPackToString(input, data_pack.UTF8) |
+ self.failUnless(output == expected) |
def testRePackUnittest(self): |
expected_with_whitelist = { |
@@ -83,14 +50,12 @@ |
in inputs] |
# RePack using whitelist |
- output, _ = data_pack.RePackFromDataPackStrings( |
- inputs, whitelist, suppress_removed_key_output=True) |
+ output, _ = data_pack.RePackFromDataPackStrings(inputs, whitelist) |
self.assertDictEqual(expected_with_whitelist, output, |
'Incorrect resource output') |
# RePack a None whitelist |
- output, _ = data_pack.RePackFromDataPackStrings( |
- inputs, None, suppress_removed_key_output=True) |
+ output, _ = data_pack.RePackFromDataPackStrings(inputs, None) |
self.assertDictEqual(expected_without_whitelist, output, |
'Incorrect resource output') |