| 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 unittest | 8 import unittest |
| 9 | 9 |
| 10 from fetcher_cache import FetcherCache | 10 from fetcher_cache import FetcherCache |
| 11 from local_fetcher import LocalFetcher | 11 from local_fetcher import LocalFetcher |
| 12 from template_data_source import TemplateDataSource | 12 from template_data_source import TemplateDataSource |
| 13 from third_party.handlebar import Handlebar | 13 from third_party.handlebar import Handlebar |
| 14 | 14 |
| 15 class TemplateDataSourceTest(unittest.TestCase): | 15 class TemplateDataSourceTest(unittest.TestCase): |
| 16 def setUp(self): | 16 def setUp(self): |
| 17 self._base_path = os.path.join('test_data', 'template_data_source') | 17 self._base_path = os.path.join('test_data', 'template_data_source') |
| 18 self._fake_api_data_source = {} | 18 self._fake_api_data_source = {} |
| 19 self._fake_intro_data_source = {} | 19 self._fake_intro_data_source = {} |
| 20 self._fake_samples_data_source = {} |
| 20 | 21 |
| 21 def _ReadLocalFile(self, filename): | 22 def _ReadLocalFile(self, filename): |
| 22 with open(os.path.join(self._base_path, filename), 'r') as f: | 23 with open(os.path.join(self._base_path, filename), 'r') as f: |
| 23 return f.read() | 24 return f.read() |
| 24 | 25 |
| 25 def _RenderTest(self, name, data_source): | 26 def _RenderTest(self, name, data_source): |
| 26 template_name = name + '_tmpl.html' | 27 template_name = name + '_tmpl.html' |
| 27 template = Handlebar(self._ReadLocalFile(template_name)) | 28 template = Handlebar(self._ReadLocalFile(template_name)) |
| 28 self.assertEquals( | 29 self.assertEquals( |
| 29 self._ReadLocalFile(name + '_expected.html'), | 30 self._ReadLocalFile(name + '_expected.html'), |
| 30 data_source.Render(template_name)) | 31 data_source.Render(template_name)) |
| 31 | 32 |
| 32 def _CreateTemplateDataSource(self, input_dict, cache_builder): | 33 def _CreateTemplateDataSource(self, input_dict, cache_builder): |
| 33 return TemplateDataSource('fake_branch', | 34 return TemplateDataSource('fake_branch', |
| 34 input_dict, | 35 input_dict, |
| 35 self._fake_intro_data_source, | 36 self._fake_intro_data_source, |
| 37 self._fake_samples_data_source, |
| 36 cache_builder, | 38 cache_builder, |
| 37 ['./']) | 39 ['./']) |
| 38 | 40 |
| 39 def testSimple(self): | 41 def testSimple(self): |
| 40 self._base_path = os.path.join(self._base_path, 'simple') | 42 self._base_path = os.path.join(self._base_path, 'simple') |
| 41 fetcher = LocalFetcher(self._base_path) | 43 fetcher = LocalFetcher(self._base_path) |
| 42 cache_builder = FetcherCache.Builder(fetcher, 0) | 44 cache_builder = FetcherCache.Builder(fetcher, 0) |
| 43 t_data_source = self._CreateTemplateDataSource(self._fake_api_data_source, | 45 t_data_source = self._CreateTemplateDataSource(self._fake_api_data_source, |
| 44 cache_builder) | 46 cache_builder) |
| 45 template_a1 = Handlebar(self._ReadLocalFile('test1.html')) | 47 template_a1 = Handlebar(self._ReadLocalFile('test1.html')) |
| (...skipping 27 matching lines...) Expand all Loading... |
| 73 json.loads(self._ReadLocalFile('test1.json')), | 75 json.loads(self._ReadLocalFile('test1.json')), |
| 74 cache_builder)) | 76 cache_builder)) |
| 75 self._RenderTest( | 77 self._RenderTest( |
| 76 'test2', | 78 'test2', |
| 77 self._CreateTemplateDataSource( | 79 self._CreateTemplateDataSource( |
| 78 json.loads(self._ReadLocalFile('test2.json')), | 80 json.loads(self._ReadLocalFile('test2.json')), |
| 79 cache_builder)) | 81 cache_builder)) |
| 80 | 82 |
| 81 if __name__ == '__main__': | 83 if __name__ == '__main__': |
| 82 unittest.main() | 84 unittest.main() |
| OLD | NEW |