| 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 file_system_cache import FileSystemCache | 10 from file_system_cache import FileSystemCache |
| (...skipping 22 matching lines...) Expand all Loading... |
| 33 self.assertEquals( | 33 self.assertEquals( |
| 34 self._ReadLocalFile(name + '_expected.html'), | 34 self._ReadLocalFile(name + '_expected.html'), |
| 35 data_source.Render(template_name)) | 35 data_source.Render(template_name)) |
| 36 | 36 |
| 37 def _CreateTemplateDataSource(self, input_dict, cache_builder): | 37 def _CreateTemplateDataSource(self, input_dict, cache_builder): |
| 38 return (TemplateDataSource.Factory('fake_branch', | 38 return (TemplateDataSource.Factory('fake_branch', |
| 39 input_dict, | 39 input_dict, |
| 40 self._fake_intro_data_source, | 40 self._fake_intro_data_source, |
| 41 self._fake_samples_data_source, | 41 self._fake_samples_data_source, |
| 42 cache_builder, | 42 cache_builder, |
| 43 ['./', './']) | 43 './', |
| 44 './') |
| 44 .Create(_FakeRequest())) | 45 .Create(_FakeRequest())) |
| 45 | 46 |
| 46 def testSimple(self): | 47 def testSimple(self): |
| 47 self._base_path = os.path.join(self._base_path, 'simple') | 48 self._base_path = os.path.join(self._base_path, 'simple') |
| 48 fetcher = LocalFileSystem(self._base_path) | 49 fetcher = LocalFileSystem(self._base_path) |
| 49 cache_builder = FileSystemCache.Builder(fetcher) | 50 cache_builder = FileSystemCache.Builder(fetcher) |
| 50 t_data_source = self._CreateTemplateDataSource(self._fake_api_data_source, | 51 t_data_source = self._CreateTemplateDataSource(self._fake_api_data_source, |
| 51 cache_builder) | 52 cache_builder) |
| 52 template_a1 = Handlebar(self._ReadLocalFile('test1.html')) | 53 template_a1 = Handlebar(self._ReadLocalFile('test1.html')) |
| 53 self.assertEqual(template_a1.render({}, {'templates': {}}).text, | 54 self.assertEqual(template_a1.render({}, {'templates': {}}).text, |
| 54 t_data_source['test1'].render({}, {'templates': {}}).text) | 55 t_data_source['test1'].render({}, {'templates': {}}).text) |
| 55 | 56 |
| 56 template_a2 = Handlebar(self._ReadLocalFile('test2.html')) | 57 template_a2 = Handlebar(self._ReadLocalFile('test2.html')) |
| 57 self.assertEqual(template_a2.render({}, {'templates': {}}).text, | 58 self.assertEqual(template_a2.render({}, {'templates': {}}).text, |
| 58 t_data_source['test2'].render({}, {'templates': {}}).text) | 59 t_data_source['test2'].render({}, {'templates': {}}).text) |
| 59 | 60 |
| 60 self.assertEqual(None, t_data_source['junk.html']) | 61 self.assertEqual(None, t_data_source['junk.html']) |
| 61 | 62 |
| 62 def testPartials(self): | 63 def testPartials(self): |
| 63 self._base_path = os.path.join(self._base_path, 'partials') | 64 self._base_path = os.path.join(self._base_path, 'partials') |
| 64 fetcher = LocalFileSystem(self._base_path) | 65 fetcher = LocalFileSystem(self._base_path) |
| 65 cache_builder = FileSystemCache.Builder(fetcher) | 66 cache_builder = FileSystemCache.Builder(fetcher) |
| 66 t_data_source = self._CreateTemplateDataSource(self._fake_api_data_source, | 67 t_data_source = self._CreateTemplateDataSource(self._fake_api_data_source, |
| 67 cache_builder) | 68 cache_builder) |
| 68 self.assertEqual(self._ReadLocalFile('test_expected.html'), | 69 self.assertEqual( |
| 70 self._ReadLocalFile('test_expected.html'), |
| 69 t_data_source['test_tmpl'].render( | 71 t_data_source['test_tmpl'].render( |
| 70 json.loads(self._ReadLocalFile('input.json')), t_data_source).text) | 72 json.loads(self._ReadLocalFile('input.json')), t_data_source).text) |
| 71 | 73 |
| 72 def testRender(self): | 74 def testRender(self): |
| 73 self._base_path = os.path.join(self._base_path, 'render') | 75 self._base_path = os.path.join(self._base_path, 'render') |
| 74 fetcher = LocalFileSystem(self._base_path) | 76 fetcher = LocalFileSystem(self._base_path) |
| 75 context = json.loads(self._ReadLocalFile('test1.json')) | 77 context = json.loads(self._ReadLocalFile('test1.json')) |
| 76 cache_builder = FileSystemCache.Builder(fetcher) | 78 cache_builder = FileSystemCache.Builder(fetcher) |
| 77 self._RenderTest( | 79 self._RenderTest( |
| 78 'test1', | 80 'test1', |
| 79 self._CreateTemplateDataSource( | 81 self._CreateTemplateDataSource( |
| 80 json.loads(self._ReadLocalFile('test1.json')), | 82 json.loads(self._ReadLocalFile('test1.json')), |
| 81 cache_builder)) | 83 cache_builder)) |
| 82 self._RenderTest( | 84 self._RenderTest( |
| 83 'test2', | 85 'test2', |
| 84 self._CreateTemplateDataSource( | 86 self._CreateTemplateDataSource( |
| 85 json.loads(self._ReadLocalFile('test2.json')), | 87 json.loads(self._ReadLocalFile('test2.json')), |
| 86 cache_builder)) | 88 cache_builder)) |
| 87 | 89 |
| 88 if __name__ == '__main__': | 90 if __name__ == '__main__': |
| 89 unittest.main() | 91 unittest.main() |
| OLD | NEW |