Docserver: Display API features that are available to content scripts

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

Index: chrome/common/extensions/docs/server2/api_list_data_source_test.py
diff --git a/chrome/common/extensions/docs/server2/api_list_data_source_test.py b/chrome/common/extensions/docs/server2/api_list_data_source_test.py
index 77c8ee9b141c07b4c41c3a4029ca005eff6fdbd7..1e2b1fe0d8e888e56096dd5e3d38383b483ce728 100755
--- a/chrome/common/extensions/docs/server2/api_list_data_source_test.py
+++ b/chrome/common/extensions/docs/server2/api_list_data_source_test.py
@@ -7,6 +7,7 @@ import unittest
 import json
 
 from api_list_data_source import APIListDataSource
+from api_models import ContentScriptAPI
 from extensions_paths import CHROME_EXTENSIONS
 from server_instance import ServerInstance
 from test_file_system import TestFileSystem
@@ -28,7 +29,8 @@ def _ToTestFeatures(names):
   features = dict((name, {
                      'name': name,
                      'extension_types': platforms_to_extension_types(platforms),
-                   }) for name, platforms in names)
+                     'contexts': context
+                   }) for name, platforms, context in names)
   features['sockets.udp']['channel'] = 'dev'
   return features
 
@@ -46,17 +48,21 @@ def _ToTestAPISchema(names, apis):
 
 
 _TEST_API_FEATURES = _ToTestFeatures([
-  ('alarms', ['apps', 'extensions']),
-  ('app.window', ['apps']),
-  ('browserAction', ['extensions']),
-  ('experimental.bluetooth', ['apps']),
-  ('experimental.history', ['extensions'],),
-  ('experimental.power', ['apps', 'extensions']),
-  ('infobars', ['extensions']),
-  ('something_internal', ['apps']),
-  ('something_else_internal', ['extensions']),
-  ('storage', ['apps', 'extensions']),
-  ('sockets.udp', ['apps', 'extensions'])
+  ('alarms', ['apps', 'extensions'], ['content_script']),
+  ('app.window', ['apps'], []),
+  ('browserAction', ['extensions'], []),
+  ('experimental.bluetooth', ['apps'], []),
+  ('experimental.history', ['extensions'], []),
+  ('experimental.power', ['apps', 'extensions'], []),
+  ('extension', ['extensions'], ['content_script']),
+  ('extension.onRequest', ['extensions'], ['content_script']),
+  ('extension.sendNativeMessage', ['extensions'], []),
+  ('extension.sendRequest', ['extensions'], ['content_script']),
+  ('infobars', ['extensions'], []),
+  ('something_internal', ['apps'], []),
+  ('something_else_internal', ['extensions'], []),
+  ('storage', ['apps', 'extensions'], []),
+  ('sockets.udp', ['apps', 'extensions'], [])
 ])
 
 
@@ -67,6 +73,7 @@ _TEST_API_DATA = _ToTestAPIData([
   ('experimental.bluetooth', u'<code>experimental.bluetooth</code>'),
   ('experimental.history', u'<code>experimental.history</code>'),
   ('experimental.power', u'<code>experimental.power</code>'),
+  ('extension', u'<code>extension</code>'),
   ('infobars', u'<code>infobars</code>'),
   ('something_internal', u'<code>something_internal</code>'),
   ('something_else_internal', u'<code>something_else_internal</code>'),
@@ -82,6 +89,7 @@ _TEST_API_SCHEMA = [
   ('experimental.bluetooth', 'experimental_bluetooth.json'),
   ('experimental.history', 'experimental_history.json'),
   ('experimental.power', 'experimental_power.json'),
+  ('extension', 'extension.json'),
   ('infobars', 'infobars.json'),
   ('something_internal', 'something_internal.json'),
   ('something_else_internal', 'something_else_internal.json'),
@@ -117,6 +125,7 @@ _TEST_DATA = _ToTestAPISchema(_TEST_API_SCHEMA, {
           'browserAction.html',
           'experimental_history.html',
           'experimental_power.html',
+          'extension.html',
           'infobars.html',
           'storage.html',
           'sockets_udp.html'
@@ -195,6 +204,11 @@ class APIListDataSourceTest(unittest.TestCase):
             'description': u'<code>browserAction</code>'
           },
           {
+            'name': 'extension',
+            'version': 5,
+            'description': u'<code>extension</code>'
+          },
+          {
             'name': 'infobars',
             'version': 5,
             'description': u'<code>infobars</code>'
@@ -227,5 +241,21 @@ class APIListDataSourceTest(unittest.TestCase):
           'last': True
         }], self._api_list.get('extensions').get('experimental'))
 
+  def testContentScripts(self):
+    expected_alarms_api = ContentScriptAPI('alarms')
+    expected_extension_api = ContentScriptAPI('extension')
+    expected_extension_api.restrictedTo.append({
+      'node': 'onRequest',
+      'parent': 'extension',
+      'first': True
+    })
+    expected_extension_api.restrictedTo.append({
+      'node': 'sendRequest',
+      'last': True,
+      'parent': 'extension'
+    })
+    self.assertEqual([expected_alarms_api, expected_extension_api],
+                     self._api_list.get('content_scripts'))
+
 if __name__ == '__main__':
   unittest.main()