Docserver: separate models for apps and extensions

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

Index: chrome/common/extensions/docs/server2/api_categorizer_test.py
diff --git a/chrome/common/extensions/docs/server2/api_categorizer_test.py b/chrome/common/extensions/docs/server2/api_categorizer_test.py
index bdbbc51c69c269a3678e8da68a3a9b16b03da7c3..69e3478e0350c3aa34b23d95990b28f238ec6bdc 100755
--- a/chrome/common/extensions/docs/server2/api_categorizer_test.py
+++ b/chrome/common/extensions/docs/server2/api_categorizer_test.py
@@ -57,21 +57,26 @@ _TEST_DATA = {
 
 class APICategorizerTest(unittest.TestCase):
   def setUp(self):
-    self._api_categorizer = APICategorizer(
-        TestFileSystem(_TEST_DATA, relative_to=CHROME_EXTENSIONS),
-        CompiledFileSystem.Factory(ObjectStoreCreator.ForTest()))
+    self._test_file_system = TestFileSystem(
+        _TEST_DATA, relative_to=CHROME_EXTENSIONS)
+    self._compiled_file_system = CompiledFileSystem.Factory(
+        ObjectStoreCreator.ForTest())
 
   def testGetAPICategory(self):
-    get_category = self._api_categorizer.GetCategory
-    self.assertEqual('chrome', get_category('apps', 'alarms'))
-    self.assertEqual('chrome', get_category('extensions', 'alarms'))
-    self.assertEqual('private', get_category('apps', 'musicManagerPrivate'))
-    self.assertEqual('private', get_category('extensions', 'notDocumentedApi'))
-    self.assertEqual('experimental',
-                     get_category('apps', 'experimental.bluetooth'))
-    self.assertEqual('experimental',
-                     get_category('extensions', 'experimental.history'))
+    def assertGetEqual(expected, category, only_on=None):
+      for platform in ('apps', 'extensions'):
+        get_category = APICategorizer(
+            self._test_file_system,
+            self._compiled_file_system,
+            platform if only_on is None else only_on).GetCategory(category)
+        self.assertEqual(expected, get_category)
+
+    assertGetEqual('chrome', 'alarms')
+    assertGetEqual('private', 'musicManagerPrivate')
+    assertGetEqual('private', 'notDocumentedApi')
+    assertGetEqual('experimental', 'experimental.bluetooth', only_on='apps')
+    assertGetEqual('experimental', 'experimental.history', only_on='extensions')
 
 
 if __name__ == '__main__':
-  unittest.main()
+  unittest.main()