Protocol Buffers - Google's data interchange format (grpc依赖)
https://developers.google.com/protocol-buffers/
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.
154 lines
6.4 KiB
154 lines
6.4 KiB
/* |
|
* Copyright (c) 2009-2021, Google LLC |
|
* All rights reserved. |
|
* |
|
* 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 LLC 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 Google LLC 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. |
|
*/ |
|
|
|
#include "gtest/gtest.h" |
|
#include "src/google/protobuf/test_messages_proto3.upb.h" |
|
#include "upb/def.hpp" |
|
#include "upb/json_decode.h" |
|
#include "upb/json_encode.h" |
|
#include "upb/msg_test.upb.h" |
|
#include "upb/msg_test.upbdefs.h" |
|
#include "upb/upb.hpp" |
|
|
|
void VerifyMessage(const upb_test_TestExtensions *ext_msg) { |
|
EXPECT_TRUE(upb_test_TestExtensions_has_optional_int32_ext(ext_msg)); |
|
// EXPECT_FALSE(upb_test_TestExtensions_Nested_has_optional_int32_ext(ext_msg)); |
|
EXPECT_TRUE(upb_test_has_optional_msg_ext(ext_msg)); |
|
|
|
EXPECT_EQ(123, upb_test_TestExtensions_optional_int32_ext(ext_msg)); |
|
const protobuf_test_messages_proto3_TestAllTypesProto3 *ext_submsg = |
|
upb_test_optional_msg_ext(ext_msg); |
|
EXPECT_TRUE(ext_submsg != nullptr); |
|
EXPECT_EQ(456, |
|
protobuf_test_messages_proto3_TestAllTypesProto3_optional_int32( |
|
ext_submsg)); |
|
} |
|
|
|
TEST(MessageTest, Extensions) { |
|
upb::Arena arena; |
|
upb_test_TestExtensions *ext_msg = upb_test_TestExtensions_new(arena.ptr()); |
|
|
|
EXPECT_FALSE(upb_test_TestExtensions_has_optional_int32_ext(ext_msg)); |
|
// EXPECT_FALSE(upb_test_TestExtensions_Nested_has_optional_int32_ext(ext_msg)); |
|
EXPECT_FALSE(upb_test_has_optional_msg_ext(ext_msg)); |
|
|
|
upb::SymbolTable symtab; |
|
upb::MessageDefPtr m(upb_test_TestExtensions_getmsgdef(symtab.ptr())); |
|
EXPECT_TRUE(m.ptr() != nullptr); |
|
|
|
std::string json = R"json( |
|
{ |
|
"[upb_test.TestExtensions.optional_int32_ext]": 123, |
|
"[upb_test.TestExtensions.Nested.repeated_int32_ext]": [2, 4, 6], |
|
"[upb_test.optional_msg_ext]": {"optional_int32": 456} |
|
} |
|
)json"; |
|
upb::Status status; |
|
EXPECT_TRUE(upb_json_decode(json.data(), json.size(), ext_msg, m.ptr(), |
|
symtab.ptr(), 0, arena.ptr(), status.ptr())) |
|
<< status.error_message(); |
|
|
|
VerifyMessage(ext_msg); |
|
|
|
// Test round-trip through binary format. |
|
size_t size; |
|
char *serialized = upb_test_TestExtensions_serialize(ext_msg, arena.ptr(), &size); |
|
ASSERT_TRUE(serialized != nullptr); |
|
ASSERT_GE(size, 0); |
|
|
|
upb_test_TestExtensions *ext_msg2 = upb_test_TestExtensions_parse_ex( |
|
serialized, size, upb_symtab_extreg(symtab.ptr()), 0, arena.ptr()); |
|
VerifyMessage(ext_msg2); |
|
|
|
// Test round-trip through JSON format. |
|
size_t json_size = |
|
upb_json_encode(ext_msg, m.ptr(), symtab.ptr(), 0, NULL, 0, status.ptr()); |
|
char *json_buf = |
|
static_cast<char *>(upb_arena_malloc(arena.ptr(), json_size + 1)); |
|
upb_json_encode(ext_msg, m.ptr(), symtab.ptr(), 0, json_buf, json_size + 1, |
|
status.ptr()); |
|
upb_test_TestExtensions *ext_msg3 = upb_test_TestExtensions_new(arena.ptr()); |
|
EXPECT_TRUE(upb_json_decode(json_buf, json_size, ext_msg3, m.ptr(), |
|
symtab.ptr(), 0, arena.ptr(), status.ptr())) |
|
<< status.error_message(); |
|
VerifyMessage(ext_msg3); |
|
} |
|
|
|
void VerifyMessageSet(const upb_test_TestMessageSet *mset_msg) { |
|
EXPECT_TRUE(upb_test_MessageSetMember_has_message_set_extension(mset_msg)); |
|
const upb_test_MessageSetMember *member = |
|
upb_test_MessageSetMember_message_set_extension(mset_msg); |
|
EXPECT_TRUE(member != nullptr); |
|
EXPECT_TRUE(upb_test_MessageSetMember_has_optional_int32(member)); |
|
EXPECT_EQ(234, upb_test_MessageSetMember_optional_int32(member)); |
|
} |
|
|
|
TEST(MessageTest, MessageSet) { |
|
upb::Arena arena; |
|
upb_test_TestMessageSet *ext_msg = upb_test_TestMessageSet_new(arena.ptr()); |
|
|
|
EXPECT_FALSE(upb_test_MessageSetMember_has_message_set_extension(ext_msg)); |
|
|
|
upb::SymbolTable symtab; |
|
upb::MessageDefPtr m(upb_test_TestMessageSet_getmsgdef(symtab.ptr())); |
|
EXPECT_TRUE(m.ptr() != nullptr); |
|
|
|
std::string json = R"json( |
|
{ |
|
"[upb_test.MessageSetMember]": {"optional_int32": 234} |
|
} |
|
)json"; |
|
upb::Status status; |
|
EXPECT_TRUE(upb_json_decode(json.data(), json.size(), ext_msg, m.ptr(), |
|
symtab.ptr(), 0, arena.ptr(), status.ptr())) |
|
<< status.error_message(); |
|
|
|
VerifyMessageSet(ext_msg); |
|
|
|
// Test round-trip through binary format. |
|
size_t size; |
|
char *serialized = upb_test_TestMessageSet_serialize(ext_msg, arena.ptr(), &size); |
|
ASSERT_TRUE(serialized != nullptr); |
|
ASSERT_GE(size, 0); |
|
|
|
upb_test_TestMessageSet *ext_msg2 = upb_test_TestMessageSet_parse_ex( |
|
serialized, size, upb_symtab_extreg(symtab.ptr()), 0, arena.ptr()); |
|
VerifyMessageSet(ext_msg2); |
|
|
|
// Test round-trip through JSON format. |
|
size_t json_size = |
|
upb_json_encode(ext_msg, m.ptr(), symtab.ptr(), 0, NULL, 0, status.ptr()); |
|
char *json_buf = |
|
static_cast<char *>(upb_arena_malloc(arena.ptr(), json_size + 1)); |
|
upb_json_encode(ext_msg, m.ptr(), symtab.ptr(), 0, json_buf, json_size + 1, |
|
status.ptr()); |
|
upb_test_TestMessageSet *ext_msg3 = upb_test_TestMessageSet_new(arena.ptr()); |
|
EXPECT_TRUE(upb_json_decode(json_buf, json_size, ext_msg3, m.ptr(), |
|
symtab.ptr(), 0, arena.ptr(), status.ptr())) |
|
<< status.error_message(); |
|
VerifyMessageSet(ext_msg3); |
|
}
|
|
|