You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
120 lines
4.9 KiB
120 lines
4.9 KiB
#! /usr/bin/python |
|
# |
|
# Protocol Buffers - Google's data interchange format |
|
# Copyright 2008 Google Inc. All rights reserved. |
|
# https://developers.google.com/protocol-buffers/ |
|
# |
|
# Redistribution and use in source and binary forms, with or without |
|
# modification, are permitted provided that the following conditions are |
|
# met: |
|
# |
|
# * Redistributions of source code must retain the above copyright |
|
# notice, this list of conditions and the following disclaimer. |
|
# * Redistributions in binary form must reproduce the above |
|
# copyright notice, this list of conditions and the following disclaimer |
|
# in the documentation and/or other materials provided with the |
|
# distribution. |
|
# * Neither the name of Google Inc. nor the names of its |
|
# contributors may be used to endorse or promote products derived from |
|
# this software without specific prior written permission. |
|
# |
|
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
|
# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
|
# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
|
# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
|
# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
|
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
|
# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
|
# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
|
# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
|
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
|
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
|
|
|
"""Tests for google.protobuf.symbol_database.""" |
|
|
|
from google.apputils import basetest |
|
from google.protobuf import unittest_pb2 |
|
from google.protobuf import symbol_database |
|
|
|
|
|
class SymbolDatabaseTest(basetest.TestCase): |
|
|
|
def _Database(self): |
|
db = symbol_database.SymbolDatabase() |
|
# Register representative types from unittest_pb2. |
|
db.RegisterFileDescriptor(unittest_pb2.DESCRIPTOR) |
|
db.RegisterMessage(unittest_pb2.TestAllTypes) |
|
db.RegisterMessage(unittest_pb2.TestAllTypes.NestedMessage) |
|
db.RegisterMessage(unittest_pb2.TestAllTypes.OptionalGroup) |
|
db.RegisterMessage(unittest_pb2.TestAllTypes.RepeatedGroup) |
|
db.RegisterEnumDescriptor(unittest_pb2.ForeignEnum.DESCRIPTOR) |
|
db.RegisterEnumDescriptor(unittest_pb2.TestAllTypes.NestedEnum.DESCRIPTOR) |
|
return db |
|
|
|
def testGetPrototype(self): |
|
instance = self._Database().GetPrototype( |
|
unittest_pb2.TestAllTypes.DESCRIPTOR) |
|
self.assertTrue(instance is unittest_pb2.TestAllTypes) |
|
|
|
def testGetMessages(self): |
|
messages = self._Database().GetMessages( |
|
['google/protobuf/unittest.proto']) |
|
self.assertTrue( |
|
unittest_pb2.TestAllTypes is |
|
messages['protobuf_unittest.TestAllTypes']) |
|
|
|
def testGetSymbol(self): |
|
self.assertEquals( |
|
unittest_pb2.TestAllTypes, self._Database().GetSymbol( |
|
'protobuf_unittest.TestAllTypes')) |
|
self.assertEquals( |
|
unittest_pb2.TestAllTypes.NestedMessage, self._Database().GetSymbol( |
|
'protobuf_unittest.TestAllTypes.NestedMessage')) |
|
self.assertEquals( |
|
unittest_pb2.TestAllTypes.OptionalGroup, self._Database().GetSymbol( |
|
'protobuf_unittest.TestAllTypes.OptionalGroup')) |
|
self.assertEquals( |
|
unittest_pb2.TestAllTypes.RepeatedGroup, self._Database().GetSymbol( |
|
'protobuf_unittest.TestAllTypes.RepeatedGroup')) |
|
|
|
def testEnums(self): |
|
# Check registration of types in the pool. |
|
self.assertEquals( |
|
'protobuf_unittest.ForeignEnum', |
|
self._Database().pool.FindEnumTypeByName( |
|
'protobuf_unittest.ForeignEnum').full_name) |
|
self.assertEquals( |
|
'protobuf_unittest.TestAllTypes.NestedEnum', |
|
self._Database().pool.FindEnumTypeByName( |
|
'protobuf_unittest.TestAllTypes.NestedEnum').full_name) |
|
|
|
def testFindMessageTypeByName(self): |
|
self.assertEquals( |
|
'protobuf_unittest.TestAllTypes', |
|
self._Database().pool.FindMessageTypeByName( |
|
'protobuf_unittest.TestAllTypes').full_name) |
|
self.assertEquals( |
|
'protobuf_unittest.TestAllTypes.NestedMessage', |
|
self._Database().pool.FindMessageTypeByName( |
|
'protobuf_unittest.TestAllTypes.NestedMessage').full_name) |
|
|
|
def testFindFindContainingSymbol(self): |
|
# Lookup based on either enum or message. |
|
self.assertEquals( |
|
'google/protobuf/unittest.proto', |
|
self._Database().pool.FindFileContainingSymbol( |
|
'protobuf_unittest.TestAllTypes.NestedEnum').name) |
|
self.assertEquals( |
|
'google/protobuf/unittest.proto', |
|
self._Database().pool.FindFileContainingSymbol( |
|
'protobuf_unittest.TestAllTypes').name) |
|
|
|
def testFindFileByName(self): |
|
self.assertEquals( |
|
'google/protobuf/unittest.proto', |
|
self._Database().pool.FindFileByName( |
|
'google/protobuf/unittest.proto').name) |
|
|
|
|
|
if __name__ == '__main__': |
|
basetest.main()
|
|
|