| Index: chrome/common/extensions/docs/server2/features_model_test.py
|
| diff --git a/chrome/common/extensions/docs/server2/features_model_test.py b/chrome/common/extensions/docs/server2/features_model_test.py
|
| index 80fb6620bb111694837991803005ec3c51780875..8a7335888f5fa7fcb45426d49194c69646768f64 100755
|
| --- a/chrome/common/extensions/docs/server2/features_model_test.py
|
| +++ b/chrome/common/extensions/docs/server2/features_model_test.py
|
| @@ -6,7 +6,7 @@
|
| from copy import deepcopy
|
| import unittest
|
|
|
| -from features_model import FeaturesModel
|
| +from features_model import FeaturesModel, Feature
|
|
|
| raw_features_json = {
|
| 'doc1': {
|
| @@ -37,26 +37,26 @@ raw_features_json = {
|
| class FeaturesUtilityTest(unittest.TestCase):
|
| def testFeatureFile(self):
|
| init_json = {
|
| - 'doc1': {
|
| + 'doc1': Feature({
|
| 'platform': ['extension', 'app'],
|
| 'name': 'doc1'
|
| - },
|
| - 'doc2': {
|
| + }),
|
| + 'doc2': Feature({
|
| 'platform': [],
|
| 'name': 'doc2'
|
| - },
|
| - 'doc4': {
|
| + }),
|
| + 'doc4': Feature({
|
| 'platform': ['extension', 'app'],
|
| 'name': 'doc4'
|
| - },
|
| - 'doc5': {
|
| + }),
|
| + 'doc5': Feature({
|
| 'platform': ['extension'],
|
| 'name': 'doc5'
|
| - },
|
| - 'doc1.sub1': {
|
| + }),
|
| + 'doc1.sub1': Feature({
|
| 'platform': ['app'],
|
| 'name': 'doc1.sub1'
|
| - }
|
| + })
|
| }
|
|
|
| features_model = FeaturesModel().FromJson(raw_features_json)
|
| @@ -79,12 +79,12 @@ class FeaturesUtilityTest(unittest.TestCase):
|
| }
|
|
|
| expected_merge_result = {
|
| - 'doc1': {
|
| + 'doc1': Feature({
|
| 'documentation': 'http://documentation',
|
| 'platform': ['extension', 'app'],
|
| 'name': 'doc1'
|
| - },
|
| - 'doc2': {
|
| + }),
|
| + 'doc2': Feature({
|
| 'platform': [],
|
| 'name': 'doc2',
|
| 'children': {
|
| @@ -92,40 +92,40 @@ class FeaturesUtilityTest(unittest.TestCase):
|
| 'name': 'sub'
|
| }
|
| }
|
| - },
|
| - 'doc4': {
|
| + }),
|
| + 'doc4': Feature({
|
| 'platform': ['extension', 'app'],
|
| 'name': 'doc4'
|
| - },
|
| - 'doc5': {
|
| + }),
|
| + 'doc5': Feature({
|
| 'platform': ['extension'],
|
| 'name': 'doc5',
|
| 'example': {}
|
| - },
|
| - 'doc1.sub1': {
|
| + }),
|
| + 'doc1.sub1': Feature({
|
| 'platform': ['app'],
|
| 'name': 'doc1.sub1'
|
| - }
|
| + })
|
| }
|
|
|
| features_model = features_model.MergeWith(additional_dict)
|
| self.assertEqual(expected_merge_result, features_model.Get())
|
|
|
| expected_filter = {
|
| - 'doc1': {
|
| + 'doc1': Feature({
|
| 'documentation': 'http://documentation',
|
| 'platform': ['extension', 'app'],
|
| 'name': 'doc1'
|
| - },
|
| - 'doc4': {
|
| + }),
|
| + 'doc4': Feature({
|
| 'platform': ['extension', 'app'],
|
| 'name': 'doc4'
|
| - },
|
| - 'doc5': {
|
| + }),
|
| + 'doc5': Feature({
|
| 'platform': ['extension'],
|
| 'name': 'doc5',
|
| 'example': {}
|
| - }
|
| + })
|
| }
|
|
|
| self.assertEqual(
|
|
|