Extensions Docs Server: HandlebarDictGenerator

chrome/common/extensions/docs/server2/api_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 unittest
 
 from fetcher_cache import FetcherCache
 from local_fetcher import LocalFetcher
 from api_data_source import APIDataSource
 
+class TestAPIDataSource(APIDataSource):
+  def _LoadAPI(self, api):
+    return json.loads(api)[0]
+
 class APIDataSourceTest(unittest.TestCase):
   def setUp(self):
     self._base_path = os.path.join('test_data', 'api_data_source')
 
   def _ReadLocalFile(self, filename):
     with open(os.path.join(self._base_path, filename), 'r') as f:
       return f.read()
 
   def testSimple(self):
     self._base_path = os.path.join(self._base_path, 'simple')
     fetcher = LocalFetcher(self._base_path)
     cache_builder = FetcherCache.Builder(fetcher, 0)
-    data_source = APIDataSource(cache_builder, ['./'])
+    data_source = TestAPIDataSource(cache_builder, ['./'])
 
     # Take the dict out of the list.
     expected = json.loads(self._ReadLocalFile('test_file.json'))[0]

[not at google - send to devlin, 2012/06/19 20:33:10]

Could you modify "expected" here rather than overriding _LoadAPI for
TestAPIDataSource? It's kind of defeating the purpose of testing APIDataSource
if you're going to override part of its behaviour.

[cduvall, 2012/06/19 22:21:02]

Done.

     self.assertEqual(expected, data_source['test_file'])
     self.assertEqual(expected, data_source['testFile'])
     self.assertEqual(expected, data_source['testFile.html'])
 
     self.assertEqual(None, data_source['junk'])
 
 if __name__ == '__main__':
   unittest.main()