mirror of https://github.com/grpc/grpc.git
Merge pull request #2954 from tbetbetbe/grpc-ruby-mv-interop-test-to-pb
Grpc ruby mv interop test to pbpull/2956/head
Michael Lumish
10 years ago
commit
b7eefcf757
17 changed files with 855 additions and 715 deletions
@ -1,8 +0,0 @@ |
|||||||
Interop test protos |
|
||||||
=================== |
|
||||||
|
|
||||||
These ruby classes were generated with protoc v3, using grpc's ruby compiler |
|
||||||
plugin. |
|
||||||
|
|
||||||
- As of 2015/01 protoc v3 is available in the |
|
||||||
[google-protobuf](https://github.com/google/protobuf) repo |
|
||||||
@ -1,44 +0,0 @@ |
|||||||
# Copyright 2015, Google Inc. |
|
||||||
# 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 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. |
|
||||||
|
|
||||||
# Generated by the protocol buffer compiler. DO NOT EDIT! |
|
||||||
# source: test/cpp/interop/empty.proto |
|
||||||
|
|
||||||
require 'google/protobuf' |
|
||||||
|
|
||||||
Google::Protobuf::DescriptorPool.generated_pool.build do |
|
||||||
add_message "grpc.testing.Empty" do |
|
||||||
end |
|
||||||
end |
|
||||||
|
|
||||||
module Grpc |
|
||||||
module Testing |
|
||||||
Empty = Google::Protobuf::DescriptorPool.generated_pool.lookup("grpc.testing.Empty").msgclass |
|
||||||
end |
|
||||||
end |
|
||||||
@ -1,60 +0,0 @@ |
|||||||
# Copyright 2015, Google Inc. |
|
||||||
# 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 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. |
|
||||||
|
|
||||||
# Generated by the protocol buffer compiler. DO NOT EDIT! |
|
||||||
# Source: test/cpp/interop/test.proto for package 'grpc.testing' |
|
||||||
|
|
||||||
require 'grpc' |
|
||||||
require 'test/cpp/interop/test' |
|
||||||
|
|
||||||
module Grpc |
|
||||||
module Testing |
|
||||||
module TestService |
|
||||||
|
|
||||||
# TODO: add proto service documentation here |
|
||||||
class Service |
|
||||||
|
|
||||||
include GRPC::GenericService |
|
||||||
|
|
||||||
self.marshal_class_method = :encode |
|
||||||
self.unmarshal_class_method = :decode |
|
||||||
self.service_name = 'grpc.testing.TestService' |
|
||||||
|
|
||||||
rpc :EmptyCall, Empty, Empty |
|
||||||
rpc :UnaryCall, SimpleRequest, SimpleResponse |
|
||||||
rpc :StreamingOutputCall, StreamingOutputCallRequest, stream(StreamingOutputCallResponse) |
|
||||||
rpc :StreamingInputCall, stream(StreamingInputCallRequest), StreamingInputCallResponse |
|
||||||
rpc :FullDuplexCall, stream(StreamingOutputCallRequest), stream(StreamingOutputCallResponse) |
|
||||||
rpc :HalfDuplexCall, stream(StreamingOutputCallRequest), stream(StreamingOutputCallResponse) |
|
||||||
end |
|
||||||
|
|
||||||
Stub = Service.rpc_stub_class |
|
||||||
end |
|
||||||
end |
|
||||||
end |
|
||||||
@ -0,0 +1,453 @@ |
|||||||
|
#!/usr/bin/env ruby |
||||||
|
|
||||||
|
# Copyright 2015, Google Inc. |
||||||
|
# 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 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. |
||||||
|
|
||||||
|
# client is a testing tool that accesses a gRPC interop testing server and runs |
||||||
|
# a test on it. |
||||||
|
# |
||||||
|
# Helps validate interoperation b/w different gRPC implementations. |
||||||
|
# |
||||||
|
# Usage: $ path/to/client.rb --server_host=<hostname> \ |
||||||
|
# --server_port=<port> \ |
||||||
|
# --test_case=<testcase_name> |
||||||
|
|
||||||
|
this_dir = File.expand_path(File.dirname(__FILE__)) |
||||||
|
lib_dir = File.join(File.dirname(File.dirname(this_dir)), 'lib') |
||||||
|
pb_dir = File.dirname(File.dirname(this_dir)) |
||||||
|
$LOAD_PATH.unshift(lib_dir) unless $LOAD_PATH.include?(lib_dir) |
||||||
|
$LOAD_PATH.unshift(pb_dir) unless $LOAD_PATH.include?(pb_dir) |
||||||
|
$LOAD_PATH.unshift(this_dir) unless $LOAD_PATH.include?(this_dir) |
||||||
|
|
||||||
|
require 'optparse' |
||||||
|
|
||||||
|
require 'grpc' |
||||||
|
require 'googleauth' |
||||||
|
require 'google/protobuf' |
||||||
|
|
||||||
|
require 'test/proto/empty' |
||||||
|
require 'test/proto/messages' |
||||||
|
require 'test/proto/test_services' |
||||||
|
|
||||||
|
require 'signet/ssl_config' |
||||||
|
|
||||||
|
AUTH_ENV = Google::Auth::CredentialsLoader::ENV_VAR |
||||||
|
|
||||||
|
# AssertionError is use to indicate interop test failures. |
||||||
|
class AssertionError < RuntimeError; end |
||||||
|
|
||||||
|
# Fails with AssertionError if the block does evaluate to true |
||||||
|
def assert(msg = 'unknown cause') |
||||||
|
fail 'No assertion block provided' unless block_given? |
||||||
|
fail AssertionError, msg unless yield |
||||||
|
end |
||||||
|
|
||||||
|
# loads the certificates used to access the test server securely. |
||||||
|
def load_test_certs |
||||||
|
this_dir = File.expand_path(File.dirname(__FILE__)) |
||||||
|
data_dir = File.join(File.dirname(File.dirname(this_dir)), 'spec/testdata') |
||||||
|
files = ['ca.pem', 'server1.key', 'server1.pem'] |
||||||
|
files.map { |f| File.open(File.join(data_dir, f)).read } |
||||||
|
end |
||||||
|
|
||||||
|
# loads the certificates used to access the test server securely. |
||||||
|
def load_prod_cert |
||||||
|
fail 'could not find a production cert' if ENV['SSL_CERT_FILE'].nil? |
||||||
|
GRPC.logger.info("loading prod certs from #{ENV['SSL_CERT_FILE']}") |
||||||
|
File.open(ENV['SSL_CERT_FILE']).read |
||||||
|
end |
||||||
|
|
||||||
|
# creates SSL Credentials from the test certificates. |
||||||
|
def test_creds |
||||||
|
certs = load_test_certs |
||||||
|
GRPC::Core::Credentials.new(certs[0]) |
||||||
|
end |
||||||
|
|
||||||
|
# creates SSL Credentials from the production certificates. |
||||||
|
def prod_creds |
||||||
|
cert_text = load_prod_cert |
||||||
|
GRPC::Core::Credentials.new(cert_text) |
||||||
|
end |
||||||
|
|
||||||
|
# creates the SSL Credentials. |
||||||
|
def ssl_creds(use_test_ca) |
||||||
|
return test_creds if use_test_ca |
||||||
|
prod_creds |
||||||
|
end |
||||||
|
|
||||||
|
# creates a test stub that accesses host:port securely. |
||||||
|
def create_stub(opts) |
||||||
|
address = "#{opts.host}:#{opts.port}" |
||||||
|
if opts.secure |
||||||
|
stub_opts = { |
||||||
|
:creds => ssl_creds(opts.use_test_ca), |
||||||
|
GRPC::Core::Channel::SSL_TARGET => opts.host_override |
||||||
|
} |
||||||
|
|
||||||
|
# Add service account creds if specified |
||||||
|
wants_creds = %w(all compute_engine_creds service_account_creds) |
||||||
|
if wants_creds.include?(opts.test_case) |
||||||
|
unless opts.oauth_scope.nil? |
||||||
|
auth_creds = Google::Auth.get_application_default(opts.oauth_scope) |
||||||
|
stub_opts[:update_metadata] = auth_creds.updater_proc |
||||||
|
end |
||||||
|
end |
||||||
|
|
||||||
|
if opts.test_case == 'oauth2_auth_token' |
||||||
|
auth_creds = Google::Auth.get_application_default(opts.oauth_scope) |
||||||
|
kw = auth_creds.updater_proc.call({}) # gives as an auth token |
||||||
|
|
||||||
|
# use a metadata update proc that just adds the auth token. |
||||||
|
stub_opts[:update_metadata] = proc { |md| md.merge(kw) } |
||||||
|
end |
||||||
|
|
||||||
|
if opts.test_case == 'jwt_token_creds' # don't use a scope |
||||||
|
auth_creds = Google::Auth.get_application_default |
||||||
|
stub_opts[:update_metadata] = auth_creds.updater_proc |
||||||
|
end |
||||||
|
|
||||||
|
GRPC.logger.info("... connecting securely to #{address}") |
||||||
|
Grpc::Testing::TestService::Stub.new(address, **stub_opts) |
||||||
|
else |
||||||
|
GRPC.logger.info("... connecting insecurely to #{address}") |
||||||
|
Grpc::Testing::TestService::Stub.new(address) |
||||||
|
end |
||||||
|
end |
||||||
|
|
||||||
|
# produces a string of null chars (\0) of length l. |
||||||
|
def nulls(l) |
||||||
|
fail 'requires #{l} to be +ve' if l < 0 |
||||||
|
[].pack('x' * l).force_encoding('utf-8') |
||||||
|
end |
||||||
|
|
||||||
|
# a PingPongPlayer implements the ping pong bidi test. |
||||||
|
class PingPongPlayer |
||||||
|
include Grpc::Testing |
||||||
|
include Grpc::Testing::PayloadType |
||||||
|
attr_accessor :queue |
||||||
|
attr_accessor :canceller_op |
||||||
|
|
||||||
|
# reqs is the enumerator over the requests |
||||||
|
def initialize(msg_sizes) |
||||||
|
@queue = Queue.new |
||||||
|
@msg_sizes = msg_sizes |
||||||
|
@canceller_op = nil # used to cancel after the first response |
||||||
|
end |
||||||
|
|
||||||
|
def each_item |
||||||
|
return enum_for(:each_item) unless block_given? |
||||||
|
req_cls, p_cls = StreamingOutputCallRequest, ResponseParameters # short |
||||||
|
count = 0 |
||||||
|
@msg_sizes.each do |m| |
||||||
|
req_size, resp_size = m |
||||||
|
req = req_cls.new(payload: Payload.new(body: nulls(req_size)), |
||||||
|
response_type: :COMPRESSABLE, |
||||||
|
response_parameters: [p_cls.new(size: resp_size)]) |
||||||
|
yield req |
||||||
|
resp = @queue.pop |
||||||
|
assert('payload type is wrong') { :COMPRESSABLE == resp.payload.type } |
||||||
|
assert("payload body #{count} has the wrong length") do |
||||||
|
resp_size == resp.payload.body.length |
||||||
|
end |
||||||
|
p "OK: ping_pong #{count}" |
||||||
|
count += 1 |
||||||
|
unless @canceller_op.nil? |
||||||
|
canceller_op.cancel |
||||||
|
break |
||||||
|
end |
||||||
|
end |
||||||
|
end |
||||||
|
end |
||||||
|
|
||||||
|
# defines methods corresponding to each interop test case. |
||||||
|
class NamedTests |
||||||
|
include Grpc::Testing |
||||||
|
include Grpc::Testing::PayloadType |
||||||
|
|
||||||
|
def initialize(stub, args) |
||||||
|
@stub = stub |
||||||
|
@args = args |
||||||
|
end |
||||||
|
|
||||||
|
def empty_unary |
||||||
|
resp = @stub.empty_call(Empty.new) |
||||||
|
assert('empty_unary: invalid response') { resp.is_a?(Empty) } |
||||||
|
p 'OK: empty_unary' |
||||||
|
end |
||||||
|
|
||||||
|
def large_unary |
||||||
|
perform_large_unary |
||||||
|
p 'OK: large_unary' |
||||||
|
end |
||||||
|
|
||||||
|
def service_account_creds |
||||||
|
# ignore this test if the oauth options are not set |
||||||
|
if @args.oauth_scope.nil? |
||||||
|
p 'NOT RUN: service_account_creds; no service_account settings' |
||||||
|
return |
||||||
|
end |
||||||
|
json_key = File.read(ENV[AUTH_ENV]) |
||||||
|
wanted_email = MultiJson.load(json_key)['client_email'] |
||||||
|
resp = perform_large_unary(fill_username: true, |
||||||
|
fill_oauth_scope: true) |
||||||
|
assert("#{__callee__}: bad username") { wanted_email == resp.username } |
||||||
|
assert("#{__callee__}: bad oauth scope") do |
||||||
|
@args.oauth_scope.include?(resp.oauth_scope) |
||||||
|
end |
||||||
|
p "OK: #{__callee__}" |
||||||
|
end |
||||||
|
|
||||||
|
def jwt_token_creds |
||||||
|
json_key = File.read(ENV[AUTH_ENV]) |
||||||
|
wanted_email = MultiJson.load(json_key)['client_email'] |
||||||
|
resp = perform_large_unary(fill_username: true) |
||||||
|
assert("#{__callee__}: bad username") { wanted_email == resp.username } |
||||||
|
p "OK: #{__callee__}" |
||||||
|
end |
||||||
|
|
||||||
|
def compute_engine_creds |
||||||
|
resp = perform_large_unary(fill_username: true, |
||||||
|
fill_oauth_scope: true) |
||||||
|
assert("#{__callee__}: bad username") do |
||||||
|
@args.default_service_account == resp.username |
||||||
|
end |
||||||
|
p "OK: #{__callee__}" |
||||||
|
end |
||||||
|
|
||||||
|
def oauth2_auth_token |
||||||
|
resp = perform_large_unary(fill_username: true, |
||||||
|
fill_oauth_scope: true) |
||||||
|
json_key = File.read(ENV[AUTH_ENV]) |
||||||
|
wanted_email = MultiJson.load(json_key)['client_email'] |
||||||
|
assert("#{__callee__}: bad username") { wanted_email == resp.username } |
||||||
|
assert("#{__callee__}: bad oauth scope") do |
||||||
|
@args.oauth_scope.include?(resp.oauth_scope) |
||||||
|
end |
||||||
|
p "OK: #{__callee__}" |
||||||
|
end |
||||||
|
|
||||||
|
def per_rpc_creds |
||||||
|
auth_creds = Google::Auth.get_application_default(@args.oauth_scope) |
||||||
|
kw = auth_creds.updater_proc.call({}) |
||||||
|
resp = perform_large_unary(fill_username: true, |
||||||
|
fill_oauth_scope: true, |
||||||
|
**kw) |
||||||
|
json_key = File.read(ENV[AUTH_ENV]) |
||||||
|
wanted_email = MultiJson.load(json_key)['client_email'] |
||||||
|
assert("#{__callee__}: bad username") { wanted_email == resp.username } |
||||||
|
assert("#{__callee__}: bad oauth scope") do |
||||||
|
@args.oauth_scope.include?(resp.oauth_scope) |
||||||
|
end |
||||||
|
p "OK: #{__callee__}" |
||||||
|
end |
||||||
|
|
||||||
|
def client_streaming |
||||||
|
msg_sizes = [27_182, 8, 1828, 45_904] |
||||||
|
wanted_aggregate_size = 74_922 |
||||||
|
reqs = msg_sizes.map do |x| |
||||||
|
req = Payload.new(body: nulls(x)) |
||||||
|
StreamingInputCallRequest.new(payload: req) |
||||||
|
end |
||||||
|
resp = @stub.streaming_input_call(reqs) |
||||||
|
assert("#{__callee__}: aggregate payload size is incorrect") do |
||||||
|
wanted_aggregate_size == resp.aggregated_payload_size |
||||||
|
end |
||||||
|
p "OK: #{__callee__}" |
||||||
|
end |
||||||
|
|
||||||
|
def server_streaming |
||||||
|
msg_sizes = [31_415, 9, 2653, 58_979] |
||||||
|
response_spec = msg_sizes.map { |s| ResponseParameters.new(size: s) } |
||||||
|
req = StreamingOutputCallRequest.new(response_type: :COMPRESSABLE, |
||||||
|
response_parameters: response_spec) |
||||||
|
resps = @stub.streaming_output_call(req) |
||||||
|
resps.each_with_index do |r, i| |
||||||
|
assert("#{__callee__}: too many responses") { i < msg_sizes.length } |
||||||
|
assert("#{__callee__}: payload body #{i} has the wrong length") do |
||||||
|
msg_sizes[i] == r.payload.body.length |
||||||
|
end |
||||||
|
assert("#{__callee__}: payload type is wrong") do |
||||||
|
:COMPRESSABLE == r.payload.type |
||||||
|
end |
||||||
|
end |
||||||
|
p "OK: #{__callee__}" |
||||||
|
end |
||||||
|
|
||||||
|
def ping_pong |
||||||
|
msg_sizes = [[27_182, 31_415], [8, 9], [1828, 2653], [45_904, 58_979]] |
||||||
|
ppp = PingPongPlayer.new(msg_sizes) |
||||||
|
resps = @stub.full_duplex_call(ppp.each_item) |
||||||
|
resps.each { |r| ppp.queue.push(r) } |
||||||
|
p "OK: #{__callee__}" |
||||||
|
end |
||||||
|
|
||||||
|
def timeout_on_sleeping_server |
||||||
|
msg_sizes = [[27_182, 31_415]] |
||||||
|
ppp = PingPongPlayer.new(msg_sizes) |
||||||
|
resps = @stub.full_duplex_call(ppp.each_item, timeout: 0.001) |
||||||
|
resps.each { |r| ppp.queue.push(r) } |
||||||
|
fail 'Should have raised GRPC::BadStatus(DEADLINE_EXCEEDED)' |
||||||
|
rescue GRPC::BadStatus => e |
||||||
|
assert("#{__callee__}: status was wrong") do |
||||||
|
e.code == GRPC::Core::StatusCodes::DEADLINE_EXCEEDED |
||||||
|
end |
||||||
|
p "OK: #{__callee__}" |
||||||
|
end |
||||||
|
|
||||||
|
def empty_stream |
||||||
|
ppp = PingPongPlayer.new([]) |
||||||
|
resps = @stub.full_duplex_call(ppp.each_item) |
||||||
|
count = 0 |
||||||
|
resps.each do |r| |
||||||
|
ppp.queue.push(r) |
||||||
|
count += 1 |
||||||
|
end |
||||||
|
assert("#{__callee__}: too many responses expected 0") do |
||||||
|
count == 0 |
||||||
|
end |
||||||
|
p "OK: #{__callee__}" |
||||||
|
end |
||||||
|
|
||||||
|
def cancel_after_begin |
||||||
|
msg_sizes = [27_182, 8, 1828, 45_904] |
||||||
|
reqs = msg_sizes.map do |x| |
||||||
|
req = Payload.new(body: nulls(x)) |
||||||
|
StreamingInputCallRequest.new(payload: req) |
||||||
|
end |
||||||
|
op = @stub.streaming_input_call(reqs, return_op: true) |
||||||
|
op.cancel |
||||||
|
op.execute |
||||||
|
fail 'Should have raised GRPC:Cancelled' |
||||||
|
rescue GRPC::Cancelled |
||||||
|
assert("#{__callee__}: call operation should be CANCELLED") { op.cancelled } |
||||||
|
p "OK: #{__callee__}" |
||||||
|
end |
||||||
|
|
||||||
|
def cancel_after_first_response |
||||||
|
msg_sizes = [[27_182, 31_415], [8, 9], [1828, 2653], [45_904, 58_979]] |
||||||
|
ppp = PingPongPlayer.new(msg_sizes) |
||||||
|
op = @stub.full_duplex_call(ppp.each_item, return_op: true) |
||||||
|
ppp.canceller_op = op # causes ppp to cancel after the 1st message |
||||||
|
op.execute.each { |r| ppp.queue.push(r) } |
||||||
|
fail 'Should have raised GRPC:Cancelled' |
||||||
|
rescue GRPC::Cancelled |
||||||
|
assert("#{__callee__}: call operation should be CANCELLED") { op.cancelled } |
||||||
|
op.wait |
||||||
|
p "OK: #{__callee__}" |
||||||
|
end |
||||||
|
|
||||||
|
def all |
||||||
|
all_methods = NamedTests.instance_methods(false).map(&:to_s) |
||||||
|
all_methods.each do |m| |
||||||
|
next if m == 'all' || m.start_with?('assert') |
||||||
|
p "TESTCASE: #{m}" |
||||||
|
method(m).call |
||||||
|
end |
||||||
|
end |
||||||
|
|
||||||
|
private |
||||||
|
|
||||||
|
def perform_large_unary(fill_username: false, fill_oauth_scope: false, **kw) |
||||||
|
req_size, wanted_response_size = 271_828, 314_159 |
||||||
|
payload = Payload.new(type: :COMPRESSABLE, body: nulls(req_size)) |
||||||
|
req = SimpleRequest.new(response_type: :COMPRESSABLE, |
||||||
|
response_size: wanted_response_size, |
||||||
|
payload: payload) |
||||||
|
req.fill_username = fill_username |
||||||
|
req.fill_oauth_scope = fill_oauth_scope |
||||||
|
resp = @stub.unary_call(req, **kw) |
||||||
|
assert('payload type is wrong') do |
||||||
|
:COMPRESSABLE == resp.payload.type |
||||||
|
end |
||||||
|
assert('payload body has the wrong length') do |
||||||
|
wanted_response_size == resp.payload.body.length |
||||||
|
end |
||||||
|
assert('payload body is invalid') do |
||||||
|
nulls(wanted_response_size) == resp.payload.body |
||||||
|
end |
||||||
|
resp |
||||||
|
end |
||||||
|
end |
||||||
|
|
||||||
|
# Args is used to hold the command line info. |
||||||
|
Args = Struct.new(:default_service_account, :host, :host_override, |
||||||
|
:oauth_scope, :port, :secure, :test_case, |
||||||
|
:use_test_ca) |
||||||
|
|
||||||
|
# validates the the command line options, returning them as a Hash. |
||||||
|
def parse_args |
||||||
|
args = Args.new |
||||||
|
args.host_override = 'foo.test.google.fr' |
||||||
|
OptionParser.new do |opts| |
||||||
|
opts.on('--oauth_scope scope', |
||||||
|
'Scope for OAuth tokens') { |v| args['oauth_scope'] = v } |
||||||
|
opts.on('--server_host SERVER_HOST', 'server hostname') do |v| |
||||||
|
args['host'] = v |
||||||
|
end |
||||||
|
opts.on('--default_service_account email_address', |
||||||
|
'email address of the default service account') do |v| |
||||||
|
args['default_service_account'] = v |
||||||
|
end |
||||||
|
opts.on('--server_host_override HOST_OVERRIDE', |
||||||
|
'override host via a HTTP header') do |v| |
||||||
|
args['host_override'] = v |
||||||
|
end |
||||||
|
opts.on('--server_port SERVER_PORT', 'server port') { |v| args['port'] = v } |
||||||
|
# instance_methods(false) gives only the methods defined in that class |
||||||
|
test_cases = NamedTests.instance_methods(false).map(&:to_s) |
||||||
|
test_case_list = test_cases.join(',') |
||||||
|
opts.on('--test_case CODE', test_cases, {}, 'select a test_case', |
||||||
|
" (#{test_case_list})") { |v| args['test_case'] = v } |
||||||
|
opts.on('-s', '--use_tls', 'require a secure connection?') do |v| |
||||||
|
args['secure'] = v |
||||||
|
end |
||||||
|
opts.on('-t', '--use_test_ca', |
||||||
|
'if secure, use the test certificate?') do |v| |
||||||
|
args['use_test_ca'] = v |
||||||
|
end |
||||||
|
end.parse! |
||||||
|
_check_args(args) |
||||||
|
end |
||||||
|
|
||||||
|
def _check_args(args) |
||||||
|
%w(host port test_case).each do |a| |
||||||
|
if args[a].nil? |
||||||
|
fail(OptionParser::MissingArgument, "please specify --#{a}") |
||||||
|
end |
||||||
|
end |
||||||
|
args |
||||||
|
end |
||||||
|
|
||||||
|
def main |
||||||
|
opts = parse_args |
||||||
|
stub = create_stub(opts) |
||||||
|
NamedTests.new(stub, opts).method(opts['test_case']).call |
||||||
|
end |
||||||
|
|
||||||
|
main |
||||||
@ -0,0 +1,15 @@ |
|||||||
|
# Generated by the protocol buffer compiler. DO NOT EDIT! |
||||||
|
# source: test/proto/empty.proto |
||||||
|
|
||||||
|
require 'google/protobuf' |
||||||
|
|
||||||
|
Google::Protobuf::DescriptorPool.generated_pool.build do |
||||||
|
add_message "grpc.testing.Empty" do |
||||||
|
end |
||||||
|
end |
||||||
|
|
||||||
|
module Grpc |
||||||
|
module Testing |
||||||
|
Empty = Google::Protobuf::DescriptorPool.generated_pool.lookup("grpc.testing.Empty").msgclass |
||||||
|
end |
||||||
|
end |
||||||
@ -0,0 +1,14 @@ |
|||||||
|
# Generated by the protocol buffer compiler. DO NOT EDIT! |
||||||
|
# source: test/proto/test.proto |
||||||
|
|
||||||
|
require 'google/protobuf' |
||||||
|
|
||||||
|
require 'test/proto/empty' |
||||||
|
require 'test/proto/messages' |
||||||
|
Google::Protobuf::DescriptorPool.generated_pool.build do |
||||||
|
end |
||||||
|
|
||||||
|
module Grpc |
||||||
|
module Testing |
||||||
|
end |
||||||
|
end |
||||||
@ -0,0 +1,64 @@ |
|||||||
|
# Generated by the protocol buffer compiler. DO NOT EDIT! |
||||||
|
# Source: test/proto/test.proto for package 'grpc.testing' |
||||||
|
|
||||||
|
require 'grpc' |
||||||
|
require 'test/proto/test' |
||||||
|
|
||||||
|
module Grpc |
||||||
|
module Testing |
||||||
|
module TestService |
||||||
|
|
||||||
|
# TODO: add proto service documentation here |
||||||
|
class Service |
||||||
|
|
||||||
|
include GRPC::GenericService |
||||||
|
|
||||||
|
self.marshal_class_method = :encode |
||||||
|
self.unmarshal_class_method = :decode |
||||||
|
self.service_name = 'grpc.testing.TestService' |
||||||
|
|
||||||
|
rpc :EmptyCall, Empty, Empty |
||||||
|
rpc :UnaryCall, SimpleRequest, SimpleResponse |
||||||
|
rpc :StreamingOutputCall, StreamingOutputCallRequest, stream(StreamingOutputCallResponse) |
||||||
|
rpc :StreamingInputCall, stream(StreamingInputCallRequest), StreamingInputCallResponse |
||||||
|
rpc :FullDuplexCall, stream(StreamingOutputCallRequest), stream(StreamingOutputCallResponse) |
||||||
|
rpc :HalfDuplexCall, stream(StreamingOutputCallRequest), stream(StreamingOutputCallResponse) |
||||||
|
end |
||||||
|
|
||||||
|
Stub = Service.rpc_stub_class |
||||||
|
end |
||||||
|
module UnimplementedService |
||||||
|
|
||||||
|
# TODO: add proto service documentation here |
||||||
|
class Service |
||||||
|
|
||||||
|
include GRPC::GenericService |
||||||
|
|
||||||
|
self.marshal_class_method = :encode |
||||||
|
self.unmarshal_class_method = :decode |
||||||
|
self.service_name = 'grpc.testing.UnimplementedService' |
||||||
|
|
||||||
|
rpc :UnimplementedCall, Empty, Empty |
||||||
|
end |
||||||
|
|
||||||
|
Stub = Service.rpc_stub_class |
||||||
|
end |
||||||
|
module ReconnectService |
||||||
|
|
||||||
|
# TODO: add proto service documentation here |
||||||
|
class Service |
||||||
|
|
||||||
|
include GRPC::GenericService |
||||||
|
|
||||||
|
self.marshal_class_method = :encode |
||||||
|
self.unmarshal_class_method = :decode |
||||||
|
self.service_name = 'grpc.testing.ReconnectService' |
||||||
|
|
||||||
|
rpc :Start, Empty, Empty |
||||||
|
rpc :Stop, Empty, ReconnectInfo |
||||||
|
end |
||||||
|
|
||||||
|
Stub = Service.rpc_stub_class |
||||||
|
end |
||||||
|
end |
||||||
|
end |
||||||
@ -0,0 +1,196 @@ |
|||||||
|
#!/usr/bin/env ruby |
||||||
|
|
||||||
|
# Copyright 2015, Google Inc. |
||||||
|
# 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 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. |
||||||
|
|
||||||
|
# interop_server is a Testing app that runs a gRPC interop testing server. |
||||||
|
# |
||||||
|
# It helps validate interoperation b/w gRPC in different environments |
||||||
|
# |
||||||
|
# Helps validate interoperation b/w different gRPC implementations. |
||||||
|
# |
||||||
|
# Usage: $ path/to/interop_server.rb --port |
||||||
|
|
||||||
|
this_dir = File.expand_path(File.dirname(__FILE__)) |
||||||
|
lib_dir = File.join(File.dirname(File.dirname(this_dir)), 'lib') |
||||||
|
pb_dir = File.dirname(File.dirname(this_dir)) |
||||||
|
$LOAD_PATH.unshift(lib_dir) unless $LOAD_PATH.include?(lib_dir) |
||||||
|
$LOAD_PATH.unshift(pb_dir) unless $LOAD_PATH.include?(pb_dir) |
||||||
|
$LOAD_PATH.unshift(this_dir) unless $LOAD_PATH.include?(this_dir) |
||||||
|
|
||||||
|
require 'forwardable' |
||||||
|
require 'optparse' |
||||||
|
|
||||||
|
require 'grpc' |
||||||
|
|
||||||
|
require 'test/proto/empty' |
||||||
|
require 'test/proto/messages' |
||||||
|
require 'test/proto/test_services' |
||||||
|
|
||||||
|
# loads the certificates by the test server. |
||||||
|
def load_test_certs |
||||||
|
this_dir = File.expand_path(File.dirname(__FILE__)) |
||||||
|
data_dir = File.join(File.dirname(File.dirname(this_dir)), 'spec/testdata') |
||||||
|
files = ['ca.pem', 'server1.key', 'server1.pem'] |
||||||
|
files.map { |f| File.open(File.join(data_dir, f)).read } |
||||||
|
end |
||||||
|
|
||||||
|
# creates a ServerCredentials from the test certificates. |
||||||
|
def test_server_creds |
||||||
|
certs = load_test_certs |
||||||
|
GRPC::Core::ServerCredentials.new(nil, certs[1], certs[2]) |
||||||
|
end |
||||||
|
|
||||||
|
# produces a string of null chars (\0) of length l. |
||||||
|
def nulls(l) |
||||||
|
fail 'requires #{l} to be +ve' if l < 0 |
||||||
|
[].pack('x' * l).force_encoding('utf-8') |
||||||
|
end |
||||||
|
|
||||||
|
# A EnumeratorQueue wraps a Queue yielding the items added to it via each_item. |
||||||
|
class EnumeratorQueue |
||||||
|
extend Forwardable |
||||||
|
def_delegators :@q, :push |
||||||
|
|
||||||
|
def initialize(sentinel) |
||||||
|
@q = Queue.new |
||||||
|
@sentinel = sentinel |
||||||
|
end |
||||||
|
|
||||||
|
def each_item |
||||||
|
return enum_for(:each_item) unless block_given? |
||||||
|
loop do |
||||||
|
r = @q.pop |
||||||
|
break if r.equal?(@sentinel) |
||||||
|
fail r if r.is_a? Exception |
||||||
|
yield r |
||||||
|
end |
||||||
|
end |
||||||
|
end |
||||||
|
|
||||||
|
# A runnable implementation of the schema-specified testing service, with each |
||||||
|
# service method implemented as required by the interop testing spec. |
||||||
|
class TestTarget < Grpc::Testing::TestService::Service |
||||||
|
include Grpc::Testing |
||||||
|
include Grpc::Testing::PayloadType |
||||||
|
|
||||||
|
def empty_call(_empty, _call) |
||||||
|
Empty.new |
||||||
|
end |
||||||
|
|
||||||
|
def unary_call(simple_req, _call) |
||||||
|
req_size = simple_req.response_size |
||||||
|
SimpleResponse.new(payload: Payload.new(type: :COMPRESSABLE, |
||||||
|
body: nulls(req_size))) |
||||||
|
end |
||||||
|
|
||||||
|
def streaming_input_call(call) |
||||||
|
sizes = call.each_remote_read.map { |x| x.payload.body.length } |
||||||
|
sum = sizes.inject { |s, x| s + x } |
||||||
|
StreamingInputCallResponse.new(aggregated_payload_size: sum) |
||||||
|
end |
||||||
|
|
||||||
|
def streaming_output_call(req, _call) |
||||||
|
cls = StreamingOutputCallResponse |
||||||
|
req.response_parameters.map do |p| |
||||||
|
cls.new(payload: Payload.new(type: req.response_type, |
||||||
|
body: nulls(p.size))) |
||||||
|
end |
||||||
|
end |
||||||
|
|
||||||
|
def full_duplex_call(reqs) |
||||||
|
# reqs is a lazy Enumerator of the requests sent by the client. |
||||||
|
q = EnumeratorQueue.new(self) |
||||||
|
cls = StreamingOutputCallResponse |
||||||
|
Thread.new do |
||||||
|
begin |
||||||
|
GRPC.logger.info('interop-server: started receiving') |
||||||
|
reqs.each do |req| |
||||||
|
resp_size = req.response_parameters[0].size |
||||||
|
GRPC.logger.info("read a req, response size is #{resp_size}") |
||||||
|
resp = cls.new(payload: Payload.new(type: req.response_type, |
||||||
|
body: nulls(resp_size))) |
||||||
|
q.push(resp) |
||||||
|
end |
||||||
|
GRPC.logger.info('interop-server: finished receiving') |
||||||
|
q.push(self) |
||||||
|
rescue StandardError => e |
||||||
|
GRPC.logger.info('interop-server: failed') |
||||||
|
GRPC.logger.warn(e) |
||||||
|
q.push(e) # share the exception with the enumerator |
||||||
|
end |
||||||
|
end |
||||||
|
q.each_item |
||||||
|
end |
||||||
|
|
||||||
|
def half_duplex_call(reqs) |
||||||
|
# TODO: update with unique behaviour of the half_duplex_call if that's |
||||||
|
# ever required by any of the tests. |
||||||
|
full_duplex_call(reqs) |
||||||
|
end |
||||||
|
end |
||||||
|
|
||||||
|
# validates the the command line options, returning them as a Hash. |
||||||
|
def parse_options |
||||||
|
options = { |
||||||
|
'port' => nil, |
||||||
|
'secure' => false |
||||||
|
} |
||||||
|
OptionParser.new do |opts| |
||||||
|
opts.banner = 'Usage: --port port' |
||||||
|
opts.on('--port PORT', 'server port') do |v| |
||||||
|
options['port'] = v |
||||||
|
end |
||||||
|
opts.on('-s', '--use_tls', 'require a secure connection?') do |v| |
||||||
|
options['secure'] = v |
||||||
|
end |
||||||
|
end.parse! |
||||||
|
|
||||||
|
if options['port'].nil? |
||||||
|
fail(OptionParser::MissingArgument, 'please specify --port') |
||||||
|
end |
||||||
|
options |
||||||
|
end |
||||||
|
|
||||||
|
def main |
||||||
|
opts = parse_options |
||||||
|
host = "0.0.0.0:#{opts['port']}" |
||||||
|
s = GRPC::RpcServer.new |
||||||
|
if opts['secure'] |
||||||
|
s.add_http2_port(host, test_server_creds) |
||||||
|
GRPC.logger.info("... running securely on #{host}") |
||||||
|
else |
||||||
|
s.add_http2_port(host) |
||||||
|
GRPC.logger.info("... running insecurely on #{host}") |
||||||
|
end |
||||||
|
s.handle(TestTarget) |
||||||
|
s.run_till_terminated |
||||||
|
end |
||||||
|
|
||||||
|
main |
||||||
Loading…
Reference in new issue