| 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 import json |
| 6 import os | 7 import os |
| 7 import sys | 8 import sys |
| 8 import unittest | 9 import unittest |
| 9 | 10 |
| 10 from compiled_file_system import CompiledFileSystem | 11 from compiled_file_system import CompiledFileSystem |
| 11 from local_file_system import LocalFileSystem | 12 from local_file_system import LocalFileSystem |
| 12 from object_store_creator import ObjectStoreCreator | 13 from object_store_creator import ObjectStoreCreator |
| 13 from sidenav_data_source import SidenavDataSource | 14 from sidenav_data_source import SidenavDataSource |
| 14 | 15 |
| 15 class SamplesDataSourceTest(unittest.TestCase): | 16 class SamplesDataSourceTest(unittest.TestCase): |
| 16 def setUp(self): | 17 def setUp(self): |
| 17 self._json_path = 'docs/server2/test_data/sidenav_data_source' | 18 self._json_path = 'docs/server2/test_data/sidenav_data_source' |
| 18 self._compiled_fs_factory = CompiledFileSystem.Factory( | 19 self._compiled_fs_factory = CompiledFileSystem.Factory( |
| 19 LocalFileSystem.Create(), | 20 LocalFileSystem.Create(), |
| 20 ObjectStoreCreator.ForTest()) | 21 ObjectStoreCreator.ForTest()) |
| 21 | 22 |
| 22 def _CheckLevels(self, items, level=2): | 23 def _CheckLevels(self, items, level=2): |
| 23 for item in items: | 24 for item in items: |
| 24 self.assertEqual(level, item['level']) | 25 self.assertEqual(level, item['level']) |
| 25 if 'items' in item: | 26 if 'items' in item: |
| 26 self._CheckLevels(item['items'], level=level + 1) | 27 self._CheckLevels(item['items'], level=level + 1) |
| 27 | 28 |
| 29 def _ReadLocalFile(self, filename): |
| 30 with open(os.path.join(sys.path[0], |
| 31 'test_data', |
| 32 'sidenav_data_source', |
| 33 filename), 'r') as f: |
| 34 return f.read() |
| 35 |
| 28 def testLevels(self): | 36 def testLevels(self): |
| 29 sidenav_data_source = SidenavDataSource.Factory(self._compiled_fs_factory, | 37 sidenav_data_source = SidenavDataSource.Factory(self._compiled_fs_factory, |
| 30 self._json_path).Create('') | 38 self._json_path, |
| 39 '').Create('') |
| 31 sidenav_json = sidenav_data_source.get('test') | 40 sidenav_json = sidenav_data_source.get('test') |
| 32 self._CheckLevels(sidenav_json) | 41 self._CheckLevels(sidenav_json) |
| 33 | 42 |
| 34 def testSelected(self): | 43 def testSelected(self): |
| 35 sidenav_data_source = SidenavDataSource.Factory( | 44 sidenav_data_source = SidenavDataSource.Factory( |
| 36 self._compiled_fs_factory, | 45 self._compiled_fs_factory, |
| 37 self._json_path).Create('www.b.com') | 46 self._json_path, |
| 47 '').Create('www.b.com') |
| 38 sidenav_json = sidenav_data_source.get('test') | 48 sidenav_json = sidenav_data_source.get('test') |
| 39 # This will be prettier once JSON is loaded with an OrderedDict. | 49 # This will be prettier once JSON is loaded with an OrderedDict. |
| 40 for item in sidenav_json: | 50 for item in sidenav_json: |
| 41 if item['title'] == 'Jim': | 51 if item['title'] == 'Jim': |
| 42 self.assertTrue(item.get('child_selected', False)) | 52 self.assertTrue(item.get('child_selected', False)) |
| 43 for next_item in item['items']: | 53 for next_item in item['items']: |
| 44 if next_item['title'] == 'B': | 54 if next_item['title'] == 'B': |
| 45 self.assertTrue(next_item.get('selected', False)) | 55 self.assertTrue(next_item.get('selected', False)) |
| 46 return | 56 return |
| 47 # If we didn't return already, we should fail. | 57 # If we didn't return already, we should fail. |
| 48 self.fail() | 58 self.fail() |
| 49 | 59 |
| 60 def testAbsolutePath(self): |
| 61 sidenav_data_source = SidenavDataSource.Factory( |
| 62 self._compiled_fs_factory, |
| 63 self._json_path, |
| 64 '/trunk').Create('absolute_path/test.html') |
| 65 sidenav_json = sidenav_data_source.get('absolute_path') |
| 66 self.assertEqual( |
| 67 sidenav_json, |
| 68 json.loads(self._ReadLocalFile('absolute_path_sidenav_expected.json'))) |
| 69 |
| 50 if __name__ == '__main__': | 70 if __name__ == '__main__': |
| 51 unittest.main() | 71 unittest.main() |
| OLD | NEW |