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.
110 lines
3.6 KiB
110 lines
3.6 KiB
#!/usr/bin/ruby |
|
# |
|
# generated_code.rb is in the same directory as this test. |
|
$LOAD_PATH.unshift(File.expand_path(File.dirname(__FILE__))) |
|
|
|
old_gc = GC.stress |
|
# Ruby 2.7.0 - 2.7.1 has a GC bug in its parser, so turn off stress for now |
|
# See https://bugs.ruby-lang.org/issues/16807 |
|
GC.stress = 0x01 | 0x04 unless RUBY_VERSION.match?(/^2\.7\./) |
|
require 'generated_code_pb' |
|
require 'generated_code_proto2_pb' |
|
GC.stress = old_gc |
|
|
|
require 'test/unit' |
|
|
|
class GCTest < Test::Unit::TestCase |
|
def get_msg_proto3 |
|
A::B::C::TestMessage.new( |
|
:optional_int32 => 1, |
|
:optional_int64 => 1, |
|
:optional_uint32 => 1, |
|
:optional_uint64 => 1, |
|
:optional_bool => true, |
|
:optional_double => 1.0, |
|
:optional_float => 1.0, |
|
:optional_string => "a", |
|
:optional_bytes => "b", |
|
:optional_enum => A::B::C::TestEnum::A, |
|
:optional_msg => A::B::C::TestMessage.new(), |
|
:repeated_int32 => [1], |
|
:repeated_int64 => [1], |
|
:repeated_uint32 => [1], |
|
:repeated_uint64 => [1], |
|
:repeated_bool => [true], |
|
:repeated_double => [1.0], |
|
:repeated_float => [1.0], |
|
:repeated_string => ["a"], |
|
:repeated_bytes => ["b"], |
|
:repeated_enum => [A::B::C::TestEnum::A], |
|
:repeated_msg => [A::B::C::TestMessage.new()], |
|
:map_int32_string => {1 => "a"}, |
|
:map_int64_string => {1 => "a"}, |
|
:map_uint32_string => {1 => "a"}, |
|
:map_uint64_string => {1 => "a"}, |
|
:map_bool_string => {true => "a"}, |
|
:map_string_string => {"a" => "a"}, |
|
:map_string_msg => {"a" => A::B::C::TestMessage.new()}, |
|
:map_string_int32 => {"a" => 1}, |
|
:map_string_bool => {"a" => true}, |
|
) |
|
end |
|
|
|
def get_msg_proto2 |
|
A::B::Proto2::TestMessage.new( |
|
:optional_int32 => 1, |
|
:optional_int64 => 1, |
|
:optional_uint32 => 1, |
|
:optional_uint64 => 1, |
|
:optional_bool => true, |
|
:optional_double => 1.0, |
|
:optional_float => 1.0, |
|
:optional_string => "a", |
|
:optional_bytes => "b", |
|
:optional_enum => A::B::Proto2::TestEnum::A, |
|
:optional_msg => A::B::Proto2::TestMessage.new(), |
|
:repeated_int32 => [1], |
|
:repeated_int64 => [1], |
|
:repeated_uint32 => [1], |
|
:repeated_uint64 => [1], |
|
:repeated_bool => [true], |
|
:repeated_double => [1.0], |
|
:repeated_float => [1.0], |
|
:repeated_string => ["a"], |
|
:repeated_bytes => ["b"], |
|
:repeated_enum => [A::B::Proto2::TestEnum::A], |
|
:repeated_msg => [A::B::Proto2::TestMessage.new()], |
|
:required_int32 => 1, |
|
:required_int64 => 1, |
|
:required_uint32 => 1, |
|
:required_uint64 => 1, |
|
:required_bool => true, |
|
:required_double => 1.0, |
|
:required_float => 1.0, |
|
:required_string => "a", |
|
:required_bytes => "b", |
|
:required_enum => A::B::Proto2::TestEnum::A, |
|
:required_msg => A::B::Proto2::TestMessage.new(), |
|
) |
|
end |
|
|
|
def test_generated_msg |
|
old_gc = GC.stress |
|
GC.stress = 0x01 | 0x04 |
|
from = get_msg_proto3 |
|
data = A::B::C::TestMessage.encode(from) |
|
to = A::B::C::TestMessage.decode(data) |
|
|
|
# This doesn't work for proto2 on JRuby because there is a nested required message. |
|
# A::B::Proto2::TestMessage has :required_msg which is of type: |
|
# A::B::Proto2::TestMessage so there is no way to generate a valid |
|
# message that doesn't exceed the depth limit |
|
if !defined? JRUBY_VERSION |
|
from = get_msg_proto2 |
|
data = A::B::Proto2::TestMessage.encode(from) |
|
to = A::B::Proto2::TestMessage.decode(data) |
|
end |
|
GC.stress = old_gc |
|
puts "passed" |
|
end |
|
end
|
|
|