OLD | NEW |
(Empty) | |
| 1 #!/usr/bin/env python |
| 2 # Copyright 2017 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 tests for the gen_predetermined_ids module.''' |
| 7 |
| 8 import os |
| 9 import sys |
| 10 if __name__ == '__main__': |
| 11 sys.path.append(os.path.join(os.path.dirname(__file__), '../..')) |
| 12 |
| 13 import StringIO |
| 14 import unittest |
| 15 |
| 16 from grit.format import gen_predetermined_ids |
| 17 |
| 18 class GenPredeterminedIdsUnittest(unittest.TestCase): |
| 19 def testGenerateResourceMapping(self): |
| 20 original_resources = {200: 'A', 201: 'B', 300: 'C', 350: 'D', 370: 'E'} |
| 21 ordered_resource_ids = [300, 201, 370] |
| 22 mapping = gen_predetermined_ids.GenerateResourceMapping( |
| 23 original_resources, ordered_resource_ids) |
| 24 self.assertEqual({101: 'C', 102: 'B', 103: 'E'}, mapping) |
| 25 |
| 26 def testReadResourceIdsFromFile(self): |
| 27 f = StringIO.StringIO(''' |
| 28 // This file is automatically generated by GRIT. Do not edit. |
| 29 |
| 30 #pragma once |
| 31 |
| 32 #define IDS_BOOKMARKS_NO_ITEMS 12500 |
| 33 #define IDS_BOOKMARK_BAR_IMPORT_LINK _Pragma("whitelisted_resource_12501") 12501 |
| 34 #define IDS_BOOKMARK_X __pragma(message("whitelisted_resource_12502")) 12502 |
| 35 ''') |
| 36 resources = {} |
| 37 gen_predetermined_ids.ReadResourceIdsFromFile(f, resources) |
| 38 self.assertEqual({12500: 'IDS_BOOKMARKS_NO_ITEMS', |
| 39 12501: 'IDS_BOOKMARK_BAR_IMPORT_LINK', |
| 40 12502: 'IDS_BOOKMARK_X'}, resources) |
| 41 |
| 42 if __name__ == '__main__': |
| 43 unittest.main() |
OLD | NEW |