| # Copyright 2015 gRPC authors. |
| # |
| # Licensed under the Apache License, Version 2.0 (the "License"); |
| # you may not use this file except in compliance with the License. |
| # You may obtain a copy of the License at |
| # |
| # http://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, software |
| # distributed under the License is distributed on an "AS IS" BASIS, |
| # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| # See the License for the specific language governing permissions and |
| # limitations under the License. |
| |
| require 'grpc' |
| |
| def load_test_certs |
| test_root = File.join(File.dirname(__FILE__), 'testdata') |
| files = ['ca.pem', 'server1.key', 'server1.pem'] |
| contents = files.map { |f| File.open(File.join(test_root, f)).read } |
| [contents[0], [{ private_key: contents[1], cert_chain: contents[2] }], false] |
| end |
| |
| Server = GRPC::Core::Server |
| |
| describe Server do |
| def create_test_cert |
| GRPC::Core::ServerCredentials.new(*load_test_certs) |
| end |
| |
| describe '#start' do |
| it 'runs without failing' do |
| blk = proc { new_core_server_for_testing(nil).start } |
| expect(&blk).to_not raise_error |
| end |
| |
| it 'fails if the server is closed' do |
| s = new_core_server_for_testing(nil) |
| s.shutdown_and_notify(nil) |
| s.close |
| expect { s.start }.to raise_error(RuntimeError) |
| end |
| end |
| |
| describe '#shutdown_and_notify and #destroy' do |
| it 'destroys a server ok' do |
| s = start_a_server |
| blk = proc do |
| s.shutdown_and_notify(nil) |
| s.destroy |
| end |
| expect(&blk).to_not raise_error |
| end |
| |
| it 'can be called more than once without error' do |
| s = start_a_server |
| begin |
| blk = proc do |
| s.shutdown_and_notify(nil) |
| s.destroy |
| end |
| expect(&blk).to_not raise_error |
| blk.call |
| expect(&blk).to_not raise_error |
| ensure |
| s.shutdown_and_notify(nil) |
| s.close |
| end |
| end |
| end |
| |
| describe '#shutdown_and_notify and #close' do |
| it 'closes a server ok' do |
| s = start_a_server |
| begin |
| blk = proc do |
| s.shutdown_and_notify(nil) |
| s.close |
| end |
| expect(&blk).to_not raise_error |
| ensure |
| s.shutdown_and_notify(nil) |
| s.close |
| end |
| end |
| |
| it 'can be called more than once without error' do |
| s = start_a_server |
| blk = proc do |
| s.shutdown_and_notify(nil) |
| s.close |
| end |
| expect(&blk).to_not raise_error |
| blk.call |
| expect(&blk).to_not raise_error |
| end |
| end |
| |
| describe '#add_http_port' do |
| describe 'for insecure servers' do |
| it 'runs without failing' do |
| blk = proc do |
| s = new_core_server_for_testing(nil) |
| s.add_http2_port('localhost:0', :this_port_is_insecure) |
| s.shutdown_and_notify(nil) |
| s.close |
| end |
| expect(&blk).to_not raise_error |
| end |
| |
| it 'fails if the server is closed' do |
| s = new_core_server_for_testing(nil) |
| s.shutdown_and_notify(nil) |
| s.close |
| blk = proc do |
| s.add_http2_port('localhost:0', :this_port_is_insecure) |
| end |
| expect(&blk).to raise_error(RuntimeError) |
| end |
| end |
| |
| describe 'for secure servers' do |
| let(:cert) { create_test_cert } |
| it 'runs without failing' do |
| blk = proc do |
| s = new_core_server_for_testing(nil) |
| s.add_http2_port('localhost:0', cert) |
| s.shutdown_and_notify(nil) |
| s.close |
| end |
| expect(&blk).to_not raise_error |
| end |
| |
| it 'fails if the server is closed' do |
| s = new_core_server_for_testing(nil) |
| s.shutdown_and_notify(nil) |
| s.close |
| blk = proc { s.add_http2_port('localhost:0', cert) } |
| expect(&blk).to raise_error(RuntimeError) |
| end |
| end |
| end |
| |
| shared_examples '#new' do |
| it 'takes nil channel args' do |
| expect { new_core_server_for_testing(nil) }.to_not raise_error |
| end |
| |
| it 'does not take a hash with bad keys as channel args' do |
| blk = construct_with_args(Object.new => 1) |
| expect(&blk).to raise_error TypeError |
| blk = construct_with_args(1 => 1) |
| expect(&blk).to raise_error TypeError |
| end |
| |
| it 'does not take a hash with bad values as channel args' do |
| blk = construct_with_args(symbol: Object.new) |
| expect(&blk).to raise_error TypeError |
| blk = construct_with_args('1' => {}) |
| expect(&blk).to raise_error TypeError |
| end |
| |
| it 'can take a hash with a symbol key as channel args' do |
| blk = construct_with_args(a_symbol: 1) |
| expect(&blk).to_not raise_error |
| end |
| |
| it 'can take a hash with a string key as channel args' do |
| blk = construct_with_args('a_symbol' => 1) |
| expect(&blk).to_not raise_error |
| end |
| |
| it 'can take a hash with a string value as channel args' do |
| blk = construct_with_args(a_symbol: '1') |
| expect(&blk).to_not raise_error |
| end |
| |
| it 'can take a hash with a symbol value as channel args' do |
| blk = construct_with_args(a_symbol: :another_symbol) |
| expect(&blk).to_not raise_error |
| end |
| |
| it 'can take a hash with a numeric value as channel args' do |
| blk = construct_with_args(a_symbol: 1) |
| expect(&blk).to_not raise_error |
| end |
| |
| it 'can take a hash with many args as channel args' do |
| args = Hash[127.times.collect { |x| [x.to_s, x] }] |
| blk = construct_with_args(args) |
| expect(&blk).to_not raise_error |
| end |
| end |
| |
| describe '#new with an insecure channel' do |
| def construct_with_args(a) |
| proc { new_core_server_for_testing(a) } |
| end |
| |
| it_behaves_like '#new' |
| end |
| |
| def start_a_server |
| s = new_core_server_for_testing(nil) |
| s.add_http2_port('0.0.0.0:0', :this_port_is_insecure) |
| s.start |
| s |
| end |
| end |