Contents

shared_examples_for "a method with a GRPC timeout" do |stub_class, method_name, expectation_target, *args|
  include GRPC::Core::TimeConsts

  context "#{stub_class} timeouts for #{method_name}" do
    let(:handler) { local_stub(stub_class, 5) }
    let(:client_stub) { handler.instance_variable_get "@stub"}
    it 'uses the timeout value' do
      deadline = from_relative_time(5)
      allow(handler).to receive(:deadline).with(nil).and_return(deadline)
      allow(handler).to receive(:deadline).with(5).and_return(deadline)

      expect(client_stub).to receive(expectation_target).with(anything, hash_including(deadline: deadline)).and_call_original

      handler.public_send(method_name, *args)
    end

    it "can have a seperate timeout passed in" do
      deadline = from_relative_time(1)
      allow(handler).to receive(:deadline).with(1).and_return(deadline)
      expect(client_stub).to receive(expectation_target).with(anything, hash_including(deadline: deadline)).and_call_original
      handler.public_send(method_name, *args, timeout: 1)
    end

    it 'raises a GRPC:DeadlineExceeded if the request takes too long' do
      handler = local_stub(stub_class, 0)
      expect {handler.public_send(method_name, *args)}.to raise_error(GRPC::DeadlineExceeded)
    end
  end
end

shared_examples_for "Etcdv3 instance using a timeout" do |command, *args|
  it "raises a GRPC::DeadlineExceeded exception when it takes too long"  do
    expect do
      test_args = args.dup
      test_args.push({timeout: 0})
      conn.public_send(command, *test_args)
    end.to raise_exception(GRPC::DeadlineExceeded)
  end
  it "accepts a timeout" do
    test_args = args.dup
    test_args.push({timeout: 10})
    expect{ conn.public_send(command, *test_args) }.to_not raise_exception
  end
end

Version data entries

4 entries across 4 versions & 1 rubygems

Version	Path
etcdv3-0.11.3	spec/helpers/shared_examples_for_timeout.rb
etcdv3-0.10.2	spec/helpers/shared_examples_for_timeout.rb
etcdv3-0.10.1	spec/helpers/shared_examples_for_timeout.rb
etcdv3-0.9.0	spec/helpers/shared_examples_for_timeout.rb