OLD | NEW |
| (Empty) |
1 #!/usr/bin/env python | |
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 | |
4 # found in the LICENSE file. | |
5 | |
6 import os | |
7 import unittest | |
8 import test_urlfetch | |
9 | |
10 from local_fetcher import LocalFetcher | |
11 from template_fetcher import TemplateFetcher | |
12 from third_party.handlebar import Handlebar | |
13 | |
14 def _ReadFile(filename): | |
15 with open(filename, 'r') as f: | |
16 return f.read() | |
17 | |
18 class TestFetcher(LocalFetcher): | |
19 def FetchResource(self, branch, path): | |
20 result = self._Resource() | |
21 result.content = _ReadFile(os.path.join(self.base_path, branch, path)) | |
22 return result | |
23 | |
24 class TemplateFetcherTest(unittest.TestCase): | |
25 def testFetcher(self): | |
26 path = os.path.join('test_data', 'template_fetcher') | |
27 fetcher = TestFetcher(path) | |
28 t_fetcher_a = TemplateFetcher('a', fetcher) | |
29 t_fetcher_b = TemplateFetcher('b', fetcher) | |
30 | |
31 template_a1 = Handlebar(_ReadFile(os.path.join(path, 'a', 'test1.html'))) | |
32 self.assertEqual(template_a1.render('{}', {'templates': {}}).text, | |
33 t_fetcher_a['test1.html'].render('{}', {'templates': {}}).text) | |
34 | |
35 template_a2 = Handlebar(_ReadFile(os.path.join(path, 'a', 'test2.html'))) | |
36 self.assertEqual(template_a2.render('{}', {'templates': {}}).text, | |
37 t_fetcher_a['test2.html'].render('{}', {'templates': {}}).text) | |
38 | |
39 template_b1 = Handlebar(_ReadFile(os.path.join(path, 'b', 'test1.html'))) | |
40 self.assertEqual(template_b1.render('{}', {'templates': {}}).text, | |
41 t_fetcher_b['test1.html'].render('{}', {'templates': {}}).text) | |
42 | |
43 self.assertEqual('', t_fetcher_b['junk.html']) | |
44 | |
45 if __name__ == '__main__': | |
46 unittest.main() | |
OLD | NEW |