| 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 |