OLD | NEW |
1 #!/usr/bin/env python | 1 #!/usr/bin/env python |
2 # Copyright 2013 The Chromium Authors. All rights reserved. | 2 # Copyright 2013 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 import unittest | 5 import unittest |
6 | 6 |
7 from api_categorizer import APICategorizer | 7 from api_categorizer import APICategorizer |
8 from compiled_file_system import CompiledFileSystem | 8 from compiled_file_system import CompiledFileSystem |
9 from extensions_paths import EXTENSIONS | 9 from extensions_paths import CHROME_EXTENSIONS |
10 from object_store_creator import ObjectStoreCreator | 10 from object_store_creator import ObjectStoreCreator |
11 from test_file_system import TestFileSystem | 11 from test_file_system import TestFileSystem |
12 | 12 |
13 | 13 |
14 def _ToTestData(obj): | 14 def _ToTestData(obj): |
15 '''Transforms |obj| into test data by turning a list of files into an object | 15 '''Transforms |obj| into test data by turning a list of files into an object |
16 mapping that file to its contents (derived from its name). | 16 mapping that file to its contents (derived from its name). |
17 ''' | 17 ''' |
18 return dict((name, name) for name in obj) | 18 return dict((name, name) for name in obj) |
19 | 19 |
(...skipping 31 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
51 ]), | 51 ]), |
52 }, | 52 }, |
53 }, | 53 }, |
54 } | 54 } |
55 } | 55 } |
56 | 56 |
57 | 57 |
58 class APICategorizerTest(unittest.TestCase): | 58 class APICategorizerTest(unittest.TestCase): |
59 def setUp(self): | 59 def setUp(self): |
60 self._api_categorizer = APICategorizer( | 60 self._api_categorizer = APICategorizer( |
61 TestFileSystem(_TEST_DATA, relative_to=EXTENSIONS), | 61 TestFileSystem(_TEST_DATA, relative_to=CHROME_EXTENSIONS), |
62 CompiledFileSystem.Factory(ObjectStoreCreator.ForTest())) | 62 CompiledFileSystem.Factory(ObjectStoreCreator.ForTest())) |
63 | 63 |
64 def testGetAPICategory(self): | 64 def testGetAPICategory(self): |
65 get_category = self._api_categorizer.GetCategory | 65 get_category = self._api_categorizer.GetCategory |
66 self.assertEqual('chrome', get_category('apps', 'alarms')) | 66 self.assertEqual('chrome', get_category('apps', 'alarms')) |
67 self.assertEqual('chrome', get_category('extensions', 'alarms')) | 67 self.assertEqual('chrome', get_category('extensions', 'alarms')) |
68 self.assertEqual('private', get_category('apps', 'musicManagerPrivate')) | 68 self.assertEqual('private', get_category('apps', 'musicManagerPrivate')) |
69 self.assertEqual('private', get_category('extensions', 'notDocumentedApi')) | 69 self.assertEqual('private', get_category('extensions', 'notDocumentedApi')) |
70 self.assertEqual('experimental', | 70 self.assertEqual('experimental', |
71 get_category('apps', 'experimental.bluetooth')) | 71 get_category('apps', 'experimental.bluetooth')) |
72 self.assertEqual('experimental', | 72 self.assertEqual('experimental', |
73 get_category('extensions', 'experimental.history')) | 73 get_category('extensions', 'experimental.history')) |
74 | 74 |
75 | 75 |
76 if __name__ == '__main__': | 76 if __name__ == '__main__': |
77 unittest.main() | 77 unittest.main() |
OLD | NEW |