Docserver: separate models for apps and extensions

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

 #!/usr/bin/env python
 # Copyright 2013 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.
 
 from copy import deepcopy
 import json
 import unittest
 
 from future import Future
@@ skipping 231 matching lines @@
         'name': 'doc1',
         'platforms': ['apps', 'extensions'],
         'example': {},
         'level': 'required'
       },
       'doc1.sub1': {
         'name': 'doc1.sub1',
         'platforms': ['apps'],
         'annotations': ['important!'],
         'level': 'recommended'
-      },
-      'doc2': {
-        'name': 'doc2',
-        'platforms': ['extensions']
       }
     }
 
     expected_app = [
       {
         'example': '{...}',
         'has_example': True,
         'level': 'required',
         'name': 'doc1',
         'platforms': ['apps', 'extensions'],
         'children': [
           {
             'annotations': [
               'Recommended',
               'important!'
             ],
             'level': 'recommended',
             'name': 'sub1',
             'platforms': ['apps'],
             'is_last': True
           }
         ],
         'is_last': True
       }
     ]
 
+    class FakePlatformBundle(object):
+      def GetFeaturesBundle(self, platform):
+        return FakeFeaturesBundle()
+
     class FakeFeaturesBundle(object):
       def GetManifestFeatures(self):
         return Future(value=manifest_features)
 
     class FakeServerInstance(object):
       def __init__(self):
-        self.features_bundle = FakeFeaturesBundle()
+        self.platform_bundle = FakePlatformBundle()
         self.object_store_creator = ObjectStoreCreator.ForTest()
 
     mds = manifest_data_source.ManifestDataSource(FakeServerInstance(), None)
     self.assertEqual(expected_app, mds.get('apps'))
 
 if __name__ == '__main__':
   unittest.main()