Extensions Docs Server: Generalize $ref's to work for any schema node

chrome/common/extensions/docs/server2/template_data_source_test.py

 #!/usr/bin/env python
 # Copyright (c) 2012 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 import json
 import os
 import sys
 import unittest
 
@@ skipping 52 matching lines @@
             .Create(_FakeRequest(), 'extensions/foo'))
 
   def testSimple(self):
     self._base_path = os.path.join(self._base_path, 'simple')
     fetcher = LocalFileSystem(self._base_path)
     cache_factory = CompiledFileSystem.Factory(fetcher, self._object_store)
     t_data_source = self._CreateTemplateDataSource(
         self._fake_api_data_source_factory, cache_factory)
     template_a1 = Handlebar(self._ReadLocalFile('test1.html'))
     self.assertEqual(template_a1.render({}, {'templates': {}}).text,
-        t_data_source['test1'].render({}, {'templates': {}}).text)
+        t_data_source.get('test1').render({}, {'templates': {}}).text)
 
     template_a2 = Handlebar(self._ReadLocalFile('test2.html'))
     self.assertEqual(template_a2.render({}, {'templates': {}}).text,
-        t_data_source['test2'].render({}, {'templates': {}}).text)
+        t_data_source.get('test2').render({}, {'templates': {}}).text)
 
-    self.assertEqual(None, t_data_source['junk.html'])
+    self.assertEqual(None, t_data_source.get('junk.html'))
 
   def testPartials(self):
     self._base_path = os.path.join(self._base_path, 'partials')
     fetcher = LocalFileSystem(self._base_path)
     cache_factory = CompiledFileSystem.Factory(fetcher, self._object_store)
     t_data_source = self._CreateTemplateDataSource(
         self._fake_api_data_source_factory, cache_factory)
     self.assertEqual(
         self._ReadLocalFile('test_expected.html'),
-        t_data_source['test_tmpl'].render(
+        t_data_source.get('test_tmpl').render(
             json.loads(self._ReadLocalFile('input.json')), t_data_source).text)
 
   def testRender(self):
     self._base_path = os.path.join(self._base_path, 'render')
     fetcher = LocalFileSystem(self._base_path)
     context = json.loads(self._ReadLocalFile('test1.json'))
     cache_factory = CompiledFileSystem.Factory(fetcher, self._object_store)
     self._RenderTest(
         'test1',
         self._CreateTemplateDataSource(
             json.loads(self._ReadLocalFile('test1.json')),
                            cache_factory))
     self._RenderTest(
         'test2',
         self._CreateTemplateDataSource(
             json.loads(self._ReadLocalFile('test2.json')),
                            cache_factory))
 
 if __name__ == '__main__':
   unittest.main()