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

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

Index: chrome/common/extensions/docs/server2/docs_server_utils_test.py
diff --git a/chrome/common/extensions/docs/server2/docs_server_utils_test.py b/chrome/common/extensions/docs/server2/docs_server_utils_test.py
new file mode 100755
index 0000000000000000000000000000000000000000..1b958e4a1a3d62029ff16561d3c27e5d4821d9ff
--- /dev/null
+++ b/chrome/common/extensions/docs/server2/docs_server_utils_test.py
@@ -0,0 +1,92 @@
+#!/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
+
+from docs_server_utils import GetLinkToRefType
+from file_system import FileNotFoundError
+
+class FakeAPIDataSource(object):
+  def __init__(self, json_data):
+    self._json = json_data
+
+  def __getitem__(self, key):
+    if key not in self._json:
+      raise FileNotFoundError(key)
+    return self._json[key]
+
+  def keys(self):
+    return self._json.keys()
+
+class APIDataSourceTest(unittest.TestCase):
+  def setUp(self):
+    self._base_path = os.path.join(sys.path[0], 'test_data', 'test_json')
+
+  def _ReadLocalFile(self, filename):
+    with open(os.path.join(self._base_path, filename), 'r') as f:
+      return f.read()
+
+  def testGetLinkToRefType(self):
+    data_source = FakeAPIDataSource(
+        json.loads(self._ReadLocalFile('fake_data_source.json')))
+    self.assertEqual({
+      'href': 'foo.html#type-foo_t1',
+      'text': 'foo.foo_t1'
+    }, GetLinkToRefType('baz', 'foo.foo_t1', data_source, data_source.keys()))
+    self.assertEqual({
+      'href': '#event-baz_e1',
+      'text': 'baz_e1'
+    }, GetLinkToRefType('baz', 'baz.baz_e1', data_source, data_source.keys()))
+    self.assertEqual({
+      'href': '#event-baz_e1',
+      'text': 'baz_e1'
+    }, GetLinkToRefType('baz', 'baz_e1', data_source, data_source.keys()))
+    self.assertEqual({
+      'href': 'foo.html#method-foo_f1',
+      'text': 'foo.foo_f1'
+    }, GetLinkToRefType('baz', 'foo.foo_f1', data_source, data_source.keys()))
+    self.assertEqual({
+      'href': 'foo.html#property-foo_p3',
+      'text': 'foo.foo_p3'
+    }, GetLinkToRefType('baz', 'foo.foo_p3', data_source, data_source.keys()))
+    self.assertEqual({
+      'href': 'bar.bon.html#type-bar_t3',
+      'text': 'bar.bon.bar_t3'
+    }, GetLinkToRefType('baz',
+                        'bar.bon.bar_t3',
+                        data_source,
+                        data_source.keys()))
+    self.assertEqual({
+      'href': '#property-bar_p3',
+      'text': 'bar_p3'
+    }, GetLinkToRefType('bar.bon', 'bar_p3', data_source, data_source.keys()))
+    self.assertEqual({
+      'href': '#property-bar_p3',
+      'text': 'bar_p3'
+    }, GetLinkToRefType('bar.bon',
+                        'bar.bon.bar_p3',
+                        data_source,
+                        data_source.keys()))
+    self.assertEqual(
+        None,
+        GetLinkToRefType('bar',
+                         'bar.bon.bar_e3',
+                         data_source,
+                         data_source.keys()))
+    self.assertEqual(
+        None,
+        GetLinkToRefType('baz.bon', 'bar_p3', data_source, data_source.keys()))
+    self.assertEqual(
+        None,
+        GetLinkToRefType('a', 'falafel.faf', data_source, data_source.keys()))
+    self.assertEqual(
+        None,
+        GetLinkToRefType('foo', 'bar_p3', data_source, data_source.keys()))
+
+if __name__ == '__main__':
+  unittest.main()