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 |