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 22 matching lines...) Expand all Loading... |
33 def setUp(self): | 33 def setUp(self): |
34 self._base_path = os.path.join(sys.path[0], | 34 self._base_path = os.path.join(sys.path[0], |
35 'test_data', | 35 'test_data', |
36 'template_data_source') | 36 'template_data_source') |
37 self._fake_api_list_data_source_factory = _FakeFactory() | 37 self._fake_api_list_data_source_factory = _FakeFactory() |
38 self._fake_intro_data_source_factory = _FakeFactory() | 38 self._fake_intro_data_source_factory = _FakeFactory() |
39 self._fake_samples_data_source_factory = _FakeFactory() | 39 self._fake_samples_data_source_factory = _FakeFactory() |
40 self._fake_sidenav_data_source_factory = _FakeFactory() | 40 self._fake_sidenav_data_source_factory = _FakeFactory() |
41 self._fake_known_issues_data_source = {} | 41 self._fake_known_issues_data_source = {} |
42 self._object_store = InMemoryObjectStore('fake_branch') | 42 self._object_store = InMemoryObjectStore('fake_branch') |
| 43 self._fake_channel_data = [ |
| 44 { 'name': 'a', 'branch': '11', 'version': '1' }, |
| 45 { 'name': 'b', 'branch': '22', 'version': '2' }, |
| 46 { 'name': 'c', 'branch': '33', 'version': '3' }, |
| 47 { 'name': 'd', 'branch': '44', 'version': '4' } |
| 48 ] |
43 | 49 |
44 def _ReadLocalFile(self, filename): | 50 def _ReadLocalFile(self, filename): |
45 with open(os.path.join(self._base_path, filename), 'r') as f: | 51 with open(os.path.join(self._base_path, filename), 'r') as f: |
46 return f.read() | 52 return f.read() |
47 | 53 |
48 def _RenderTest(self, name, data_source): | 54 def _RenderTest(self, name, data_source): |
49 template_name = name + '_tmpl.html' | 55 template_name = name + '_tmpl.html' |
50 template = Handlebar(self._ReadLocalFile(template_name)) | 56 template = Handlebar(self._ReadLocalFile(template_name)) |
51 self.assertEquals( | 57 self.assertEquals( |
52 self._ReadLocalFile(name + '_expected.html'), | 58 self._ReadLocalFile(name + '_expected.html'), |
(...skipping 11 matching lines...) Expand all Loading... |
64 return (TemplateDataSource.Factory('fake_channel', | 70 return (TemplateDataSource.Factory('fake_channel', |
65 api_data_factory, | 71 api_data_factory, |
66 self._fake_api_list_data_source_factory, | 72 self._fake_api_list_data_source_factory, |
67 self._fake_intro_data_source_factory, | 73 self._fake_intro_data_source_factory, |
68 self._fake_samples_data_source_factory, | 74 self._fake_samples_data_source_factory, |
69 self._fake_known_issues_data_source, | 75 self._fake_known_issues_data_source, |
70 self._fake_sidenav_data_source_factory, | 76 self._fake_sidenav_data_source_factory, |
71 cache_factory, | 77 cache_factory, |
72 reference_resolver_factory, | 78 reference_resolver_factory, |
73 '.', | 79 '.', |
74 '.') | 80 '.', |
| 81 self._fake_channel_data) |
75 .Create(_FakeRequest(), 'extensions/foo')) | 82 .Create(_FakeRequest(), 'extensions/foo')) |
76 | 83 |
77 def testSimple(self): | 84 def testSimple(self): |
78 self._base_path = os.path.join(self._base_path, 'simple') | 85 self._base_path = os.path.join(self._base_path, 'simple') |
79 fetcher = LocalFileSystem(self._base_path) | 86 fetcher = LocalFileSystem(self._base_path) |
80 cache_factory = CompiledFileSystem.Factory(fetcher, self._object_store) | 87 cache_factory = CompiledFileSystem.Factory(fetcher, self._object_store) |
81 t_data_source = self._CreateTemplateDataSource(cache_factory) | 88 t_data_source = self._CreateTemplateDataSource(cache_factory) |
82 template_a1 = Handlebar(self._ReadLocalFile('test1.html')) | 89 template_a1 = Handlebar(self._ReadLocalFile('test1.html')) |
83 self.assertEqual(template_a1.render({}, {'templates': {}}).text, | 90 self.assertEqual(template_a1.render({}, {'templates': {}}).text, |
84 t_data_source.get('test1').render({}, {'templates': {}}).text) | 91 t_data_source.get('test1').render({}, {'templates': {}}).text) |
(...skipping 25 matching lines...) Expand all Loading... |
110 cache_factory, | 117 cache_factory, |
111 api_data=json.loads(self._ReadLocalFile('test1.json')))) | 118 api_data=json.loads(self._ReadLocalFile('test1.json')))) |
112 self._RenderTest( | 119 self._RenderTest( |
113 'test2', | 120 'test2', |
114 self._CreateTemplateDataSource( | 121 self._CreateTemplateDataSource( |
115 cache_factory, | 122 cache_factory, |
116 api_data=json.loads(self._ReadLocalFile('test2.json')))) | 123 api_data=json.loads(self._ReadLocalFile('test2.json')))) |
117 | 124 |
118 if __name__ == '__main__': | 125 if __name__ == '__main__': |
119 unittest.main() | 126 unittest.main() |
OLD | NEW |