| #!/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 os |
| import unittest |
| |
| from future import Future |
| from reference_resolver import ReferenceResolver |
| from test_object_store import TestObjectStore |
| from test_util import Server2Path |
| from third_party.json_schema_compiler.model import Namespace |
| |
| |
| _TEST_DATA = { |
| 'baz': { |
| 'namespace': 'baz', |
| 'description': '', |
| 'types': [ |
| { |
| 'id': 'baz_t1', |
| 'type': 'any', |
| }, |
| { |
| 'id': 'baz_t2', |
| 'type': 'any', |
| }, |
| { |
| 'id': 'baz_t3', |
| 'type': 'any', |
| } |
| ], |
| 'functions': [ |
| { |
| 'name': 'baz_f1', |
| 'type': 'function' |
| }, |
| { |
| 'name': 'baz_f2', |
| 'type': 'function' |
| }, |
| { |
| 'name': 'baz_f3', |
| 'type': 'function' |
| } |
| ], |
| 'events': [ |
| { |
| 'name': 'baz_e1', |
| 'type': 'function' |
| }, |
| { |
| 'name': 'baz_e2', |
| 'type': 'function' |
| }, |
| { |
| 'name': 'baz_e3', |
| 'type': 'function' |
| } |
| ], |
| 'properties': { |
| 'baz_p1': {'type': 'any'}, |
| 'baz_p2': {'type': 'any'}, |
| 'baz_p3': {'type': 'any'} |
| } |
| }, |
| 'bar.bon': { |
| 'namespace': 'bar.bon', |
| 'description': '', |
| 'types': [ |
| { |
| 'id': 'bar_bon_t1', |
| 'type': 'any', |
| }, |
| { |
| 'id': 'bar_bon_t2', |
| 'type': 'any', |
| }, |
| { |
| 'id': 'bar_bon_t3', |
| 'type': 'any', |
| } |
| ], |
| 'functions': [ |
| { |
| 'name': 'bar_bon_f1', |
| 'type': 'function' |
| }, |
| { |
| 'name': 'bar_bon_f2', |
| 'type': 'function' |
| }, |
| { |
| 'name': 'bar_bon_f3', |
| 'type': 'function' |
| } |
| ], |
| 'events': [ |
| { |
| 'name': 'bar_bon_e1', |
| 'type': 'function' |
| }, |
| { |
| 'name': 'bar_bon_e2', |
| 'type': 'function' |
| }, |
| { |
| 'name': 'bar_bon_e3', |
| 'type': 'function' |
| } |
| ], |
| 'properties': { |
| 'bar_bon_p1': {'type': 'any'}, |
| 'bar_bon_p2': {'type': 'any'}, |
| 'bar_bon_p3': {'type': 'any'} |
| } |
| }, |
| 'bar': { |
| 'namespace': 'bar', |
| 'description': '', |
| 'types': [ |
| { |
| 'id': 'bar_t1', |
| 'type': 'any', |
| 'properties': { |
| 'bar_t1_p1': { |
| 'type': 'any' |
| } |
| } |
| }, |
| { |
| 'id': 'bar_t2', |
| 'type': 'any', |
| 'properties': { |
| 'bar_t2_p1': { |
| 'type': 'any' |
| } |
| } |
| }, |
| { |
| 'id': 'bar_t3', |
| 'type': 'any', |
| }, |
| { |
| 'id': 'bon', |
| 'type': 'any' |
| } |
| ], |
| 'functions': [ |
| { |
| 'name': 'bar_f1', |
| 'type': 'function' |
| }, |
| { |
| 'name': 'bar_f2', |
| 'type': 'function' |
| }, |
| { |
| 'name': 'bar_f3', |
| 'type': 'function' |
| } |
| ], |
| 'events': [ |
| { |
| 'name': 'bar_e1', |
| 'type': 'function' |
| }, |
| { |
| 'name': 'bar_e2', |
| 'type': 'function' |
| }, |
| { |
| 'name': 'bar_e3', |
| 'type': 'function' |
| } |
| ], |
| 'properties': { |
| 'bar_p1': {'type': 'any'}, |
| 'bar_p2': {'type': 'any'}, |
| 'bar_p3': {'$ref': 'bar_t1'} |
| } |
| }, |
| 'foo': { |
| 'namespace': 'foo', |
| 'description': '', |
| 'types': [ |
| { |
| 'id': 'foo_t1', |
| 'type': 'any', |
| }, |
| { |
| 'id': 'foo_t2', |
| 'type': 'any', |
| }, |
| { |
| 'id': 'foo_t3', |
| 'type': 'any', |
| 'events': [ |
| { |
| 'name': 'foo_t3_e1', |
| 'type': 'function' |
| } |
| ] |
| } |
| ], |
| 'functions': [ |
| { |
| 'name': 'foo_f1', |
| 'type': 'function' |
| }, |
| { |
| 'name': 'foo_f2', |
| 'type': 'function' |
| }, |
| { |
| 'name': 'foo_f3', |
| 'type': 'function' |
| } |
| ], |
| 'events': [ |
| { |
| 'name': 'foo_e1', |
| 'type': 'function' |
| }, |
| { |
| 'name': 'foo_e2', |
| 'type': 'function' |
| }, |
| { |
| 'name': 'foo_e3', |
| 'type': 'function' |
| } |
| ], |
| 'properties': { |
| 'foo_p1': {'$ref': 'foo_t3'}, |
| 'foo_p2': {'type': 'any'}, |
| 'foo_p3': {'type': 'any'} |
| } |
| } |
| } |
| |
| |
| class _FakePlatformBundle(object): |
| def __init__(self): |
| self.platforms = ('apps', 'extensions') |
| |
| def GetAPIModels(self, platform): |
| if platform == 'apps': |
| return _FakeAPIModels(_TEST_DATA) |
| # Only includes some of the data in the 'extensions' APIModels. |
| # ReferenceResolver will have to look at other platforms to resolve 'foo'. |
| return _FakeAPIModels({ |
| 'bar': _TEST_DATA['bar'], |
| 'bar.bon': _TEST_DATA['bar.bon'], |
| 'baz': _TEST_DATA['baz'] |
| }) |
| |
| |
| class _FakeAPIModels(object): |
| def __init__(self, apis): |
| self._apis = apis |
| |
| def GetNames(self): |
| return self._apis.keys() |
| |
| def GetModel(self, name): |
| return Future(value=Namespace(self._apis[name], 'fake/path.json')) |
| |
| |
| class ReferenceResolverTest(unittest.TestCase): |
| def setUp(self): |
| self._base_path = Server2Path('test_data', 'test_json') |
| |
| def _ReadLocalFile(self, filename): |
| with open(os.path.join(self._base_path, filename), 'r') as f: |
| return f.read() |
| |
| def testGetLink(self): |
| apps_resolver = ReferenceResolver( |
| _FakePlatformBundle().GetAPIModels('apps'), |
| TestObjectStore('apps/test')) |
| extensions_resolver = ReferenceResolver( |
| _FakePlatformBundle().GetAPIModels('extensions'), |
| TestObjectStore('extensions/test')) |
| |
| self.assertEqual({ |
| 'href': 'foo', |
| 'text': 'foo', |
| 'name': 'foo' |
| }, apps_resolver.GetLink('foo', namespace='baz')) |
| self.assertEqual({ |
| 'href': 'foo#type-foo_t1', |
| 'text': 'foo.foo_t1', |
| 'name': 'foo_t1' |
| }, apps_resolver.GetLink('foo.foo_t1', namespace='baz')) |
| self.assertEqual({ |
| 'href': 'baz#event-baz_e1', |
| 'text': 'baz_e1', |
| 'name': 'baz_e1' |
| }, apps_resolver.GetLink('baz.baz_e1', namespace='baz')) |
| self.assertEqual({ |
| 'href': 'baz#event-baz_e1', |
| 'text': 'baz_e1', |
| 'name': 'baz_e1' |
| }, apps_resolver.GetLink('baz_e1', namespace='baz')) |
| self.assertEqual({ |
| 'href': 'foo#method-foo_f1', |
| 'text': 'foo.foo_f1', |
| 'name': 'foo_f1' |
| }, apps_resolver.GetLink('foo.foo_f1', namespace='baz')) |
| self.assertEqual({ |
| 'href': 'foo#property-foo_p3', |
| 'text': 'foo.foo_p3', |
| 'name': 'foo_p3' |
| }, apps_resolver.GetLink('foo.foo_p3', namespace='baz')) |
| self.assertEqual({ |
| 'href': 'bar.bon#type-bar_bon_t3', |
| 'text': 'bar.bon.bar_bon_t3', |
| 'name': 'bar_bon_t3' |
| }, apps_resolver.GetLink('bar.bon.bar_bon_t3', namespace='baz')) |
| self.assertEqual({ |
| 'href': 'bar.bon#property-bar_bon_p3', |
| 'text': 'bar_bon_p3', |
| 'name': 'bar_bon_p3' |
| }, apps_resolver.GetLink('bar_bon_p3', namespace='bar.bon')) |
| self.assertEqual({ |
| 'href': 'bar.bon#property-bar_bon_p3', |
| 'text': 'bar_bon_p3', |
| 'name': 'bar_bon_p3' |
| }, apps_resolver.GetLink('bar.bon.bar_bon_p3', namespace='bar.bon')) |
| self.assertEqual({ |
| 'href': 'bar#event-bar_e2', |
| 'text': 'bar_e2', |
| 'name': 'bar_e2' |
| }, apps_resolver.GetLink('bar.bar_e2', namespace='bar')) |
| self.assertEqual({ |
| 'href': 'bar#type-bon', |
| 'text': 'bon', |
| 'name': 'bon' |
| }, apps_resolver.GetLink('bar.bon', namespace='bar')) |
| self.assertEqual({ |
| 'href': 'foo#event-foo_t3-foo_t3_e1', |
| 'text': 'foo_t3.foo_t3_e1', |
| 'name': 'foo_t3_e1' |
| }, apps_resolver.GetLink('foo_t3.foo_t3_e1', namespace='foo')) |
| self.assertEqual({ |
| 'href': 'foo#event-foo_t3-foo_t3_e1', |
| 'text': 'foo_t3.foo_t3_e1', |
| 'name': 'foo_t3_e1' |
| }, apps_resolver.GetLink('foo.foo_t3.foo_t3_e1', namespace='foo')) |
| self.assertEqual({ |
| 'href': 'foo#event-foo_t3-foo_t3_e1', |
| 'text': 'foo_t3.foo_t3_e1', |
| 'name': 'foo_t3_e1' |
| }, apps_resolver.GetLink('foo.foo_p1.foo_t3_e1', namespace='foo')) |
| self.assertEqual({ |
| 'href': 'bar#property-bar_t1-bar_t1_p1', |
| 'text': 'bar.bar_t1.bar_t1_p1', |
| 'name': 'bar_t1_p1' |
| }, apps_resolver.GetLink('bar.bar_p3.bar_t1_p1', namespace='foo')) |
| # Test extensions_resolver. |
| self.assertEqual({ |
| 'href': 'bar#property-bar_t1-bar_t1_p1', |
| 'text': 'bar.bar_t1.bar_t1_p1', |
| 'name': 'bar_t1_p1' |
| }, extensions_resolver.GetLink('bar.bar_p3.bar_t1_p1', namespace='foo')) |
| self.assertEqual({ |
| 'href': 'bar#property-bar_t1-bar_t1_p1', |
| 'text': 'bar_t1.bar_t1_p1', |
| 'name': 'bar_t1_p1' |
| }, apps_resolver.GetLink('bar_p3.bar_t1_p1', namespace='bar')) |
| self.assertEqual( |
| None, |
| apps_resolver.GetLink('bar.bar_p3.bar_t2_p1', namespace='bar')) |
| self.assertEqual( |
| None, |
| apps_resolver.GetLink('bar.bon.bar_e3', namespace='bar')) |
| self.assertEqual( |
| None, |
| apps_resolver.GetLink('bar_p3', namespace='baz.bon')) |
| self.assertEqual( |
| None, |
| apps_resolver.GetLink('falafel.faf', namespace='a')) |
| self.assertEqual( |
| None, |
| apps_resolver.GetLink('bar_p3', namespace='foo')) |
| # Exists in apps but not extensions. |
| self.assertEqual( |
| None, |
| extensions_resolver.GetLink('foo.foo_p3', namespace='baz')) |
| |
| if __name__ == '__main__': |
| unittest.main() |
