| 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 local_fetcher import LocalFetcher | 10 from local_fetcher import LocalFetcher |
| 11 from template_data_source import TemplateDataSource | 11 from template_data_source import TemplateDataSource |
| 12 from third_party.handlebar import Handlebar | 12 from third_party.handlebar import Handlebar |
| 13 | 13 |
| 14 class TemplateDataSourceTest(unittest.TestCase): | 14 class TemplateDataSourceTest(unittest.TestCase): |
| 15 def setUp(self): | 15 def setUp(self): |
| 16 self._base_path = os.path.join('test_data', 'template_data_source') | 16 self._base_path = os.path.join('test_data', 'template_data_source') |
| 17 | 17 |
| 18 def _ReadLocalFile(self, filename): | 18 def _ReadLocalFile(self, filename): |
| 19 with open(os.path.join(self._base_path, filename), 'r') as f: | 19 with open(os.path.join(self._base_path, filename), 'r') as f: |
| 20 return f.read() | 20 return f.read() |
| 21 | 21 |
| 22 def _RenderTest(self, name, data_source): | 22 def _RenderTest(self, name, data_source): |
| 23 template_name = name + '_tmpl.html' | 23 template_name = name + '_tmpl.html' |
| 24 template = Handlebar(self._ReadLocalFile(template_name)) | 24 template = Handlebar(self._ReadLocalFile(template_name)) |
| 25 self.assertEquals(self._ReadLocalFile(name + '_expected.html'), | 25 context = json.loads(self._ReadLocalFile(name + '.json')) |
| 26 data_source.Render(template_name, | 26 self.assertEquals( |
| 27 self._ReadLocalFile(name + '.json'))) | 27 self._ReadLocalFile(name + '_expected.html'), |
| 28 data_source.Render(template_name, context)) |
| 28 | 29 |
| 29 def testSimple(self): | 30 def testSimple(self): |
| 30 self._base_path = os.path.join(self._base_path, 'simple') | 31 self._base_path = os.path.join(self._base_path, 'simple') |
| 31 fetcher = LocalFetcher(self._base_path) | 32 fetcher = LocalFetcher(self._base_path) |
| 32 t_data_source = TemplateDataSource(fetcher, ['./'], 0) | 33 t_data_source = TemplateDataSource(fetcher, ['./'], 0) |
| 33 | 34 |
| 34 template_a1 = Handlebar(self._ReadLocalFile('test1.html')) | 35 template_a1 = Handlebar(self._ReadLocalFile('test1.html')) |
| 35 self.assertEqual(template_a1.render({}, {'templates': {}}).text, | 36 self.assertEqual(template_a1.render({}, {'templates': {}}).text, |
| 36 t_data_source['test1'].render({}, {'templates': {}}).text) | 37 t_data_source['test1'].render({}, {'templates': {}}).text) |
| 37 | 38 |
| (...skipping 16 matching lines...) Expand all Loading... |
| 54 self._base_path = os.path.join(self._base_path, 'render') | 55 self._base_path = os.path.join(self._base_path, 'render') |
| 55 fetcher = LocalFetcher(self._base_path) | 56 fetcher = LocalFetcher(self._base_path) |
| 56 t_data_source = TemplateDataSource(fetcher, ['./'], 0) | 57 t_data_source = TemplateDataSource(fetcher, ['./'], 0) |
| 57 self._RenderTest('test1', t_data_source) | 58 self._RenderTest('test1', t_data_source) |
| 58 self._RenderTest('test2', t_data_source) | 59 self._RenderTest('test2', t_data_source) |
| 59 | 60 |
| 60 | 61 |
| 61 | 62 |
| 62 if __name__ == '__main__': | 63 if __name__ == '__main__': |
| 63 unittest.main() | 64 unittest.main() |
| OLD | NEW |