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 json |
7 import os | 7 import os |
8 import sys | 8 import sys |
9 import unittest | 9 import unittest |
10 | 10 |
(...skipping 10 matching lines...) Expand all Loading... |
21 class _FakeFactory(object): | 21 class _FakeFactory(object): |
22 def __init__(self, input_dict=None): | 22 def __init__(self, input_dict=None): |
23 if input_dict is None: | 23 if input_dict is None: |
24 self._input_dict = {} | 24 self._input_dict = {} |
25 else: | 25 else: |
26 self._input_dict = input_dict | 26 self._input_dict = input_dict |
27 | 27 |
28 def Create(self, *args, **optargs): | 28 def Create(self, *args, **optargs): |
29 return self._input_dict | 29 return self._input_dict |
30 | 30 |
| 31 class _FakeSidenavDataSource(object): |
| 32 def __init__(self, input_dict): |
| 33 self._input_dict = input_dict |
| 34 |
| 35 def get(self, key): |
| 36 return self._input_dict.get(key) |
| 37 |
| 38 def SetPath(self, path): |
| 39 pass |
| 40 |
| 41 class Factory(object): |
| 42 def __init__(self, input_dict={}): |
| 43 self._input_dict = input_dict |
| 44 |
| 45 def Create(self, *args, **optargs): |
| 46 return _FakeSidenavDataSource(self._input_dict) |
| 47 |
31 class TemplateDataSourceTest(unittest.TestCase): | 48 class TemplateDataSourceTest(unittest.TestCase): |
32 def setUp(self): | 49 def setUp(self): |
33 self._base_path = os.path.join(sys.path[0], | 50 self._base_path = os.path.join(sys.path[0], |
34 'test_data', | 51 'test_data', |
35 'template_data_source') | 52 'template_data_source') |
36 self._fake_api_list_data_source_factory = _FakeFactory() | 53 self._fake_api_list_data_source_factory = _FakeFactory() |
37 self._fake_intro_data_source_factory = _FakeFactory() | 54 self._fake_intro_data_source_factory = _FakeFactory() |
38 self._fake_samples_data_source_factory = _FakeFactory() | 55 self._fake_samples_data_source_factory = _FakeFactory() |
39 self._fake_sidenav_data_source_factory = _FakeFactory() | 56 self._fake_sidenav_data_source_factory = _FakeSidenavDataSource.Factory() |
40 | 57 |
41 def _ReadLocalFile(self, filename): | 58 def _ReadLocalFile(self, filename): |
42 with open(os.path.join(self._base_path, filename), 'r') as f: | 59 with open(os.path.join(self._base_path, filename), 'r') as f: |
43 return f.read() | 60 return f.read() |
44 | 61 |
45 def _RenderTest(self, name, data_source): | 62 def _RenderTest(self, name, data_source): |
46 template_name = name + '_tmpl.html' | 63 template_name = name + '_tmpl.html' |
47 template = Handlebar(self._ReadLocalFile(template_name)) | 64 template = Handlebar(self._ReadLocalFile(template_name)) |
48 self.assertEquals( | 65 self.assertEquals( |
49 self._ReadLocalFile(name + '_expected.html'), | 66 self._ReadLocalFile(name + '_expected.html'), |
(...skipping 16 matching lines...) Expand all Loading... |
66 'fake_channel', | 83 'fake_channel', |
67 api_data_factory, | 84 api_data_factory, |
68 self._fake_api_list_data_source_factory, | 85 self._fake_api_list_data_source_factory, |
69 self._fake_intro_data_source_factory, | 86 self._fake_intro_data_source_factory, |
70 self._fake_samples_data_source_factory, | 87 self._fake_samples_data_source_factory, |
71 self._fake_sidenav_data_source_factory, | 88 self._fake_sidenav_data_source_factory, |
72 compiled_fs_factory, | 89 compiled_fs_factory, |
73 reference_resolver_factory, | 90 reference_resolver_factory, |
74 '.', | 91 '.', |
75 '.', | 92 '.', |
76 '/static')) | 93 '')) |
77 | 94 |
78 def testSimple(self): | 95 def testSimple(self): |
79 self._base_path = os.path.join(self._base_path, 'simple') | 96 self._base_path = os.path.join(self._base_path, 'simple') |
80 fetcher = LocalFileSystem(self._base_path) | 97 fetcher = LocalFileSystem(self._base_path) |
81 compiled_fs_factory = CompiledFileSystem.Factory( | 98 compiled_fs_factory = CompiledFileSystem.Factory( |
82 fetcher, | 99 fetcher, |
83 ObjectStoreCreator.ForTest()) | 100 ObjectStoreCreator.ForTest()) |
84 t_data_source = self._CreateTemplateDataSource( | 101 t_data_source = self._CreateTemplateDataSource( |
85 compiled_fs_factory, | 102 compiled_fs_factory, |
86 ObjectStoreCreator.ForTest()) | 103 ObjectStoreCreator.ForTest()) |
(...skipping 42 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
129 compiled_fs_factory, | 146 compiled_fs_factory, |
130 api_data=json.loads(self._ReadLocalFile('test1.json')))) | 147 api_data=json.loads(self._ReadLocalFile('test1.json')))) |
131 self._RenderTest( | 148 self._RenderTest( |
132 'test2', | 149 'test2', |
133 self._CreateTemplateDataSource( | 150 self._CreateTemplateDataSource( |
134 compiled_fs_factory, | 151 compiled_fs_factory, |
135 api_data=json.loads(self._ReadLocalFile('test2.json')))) | 152 api_data=json.loads(self._ReadLocalFile('test2.json')))) |
136 | 153 |
137 if __name__ == '__main__': | 154 if __name__ == '__main__': |
138 unittest.main() | 155 unittest.main() |
OLD | NEW |