Contents

# frozen_string_literal: true

require 'grpc_mock/request_pattern'
require 'grpc_mock/response'
require 'grpc_mock/response_sequence'
require 'grpc_mock/errors'

module GrpcMock
  class RequestStub
    # @param path [String] gRPC path like /${service_name}/${method_name}
    def initialize(path)
      @request_pattern = RequestPattern.new(path)
      @response_sequence = []
    end

    def with(request = nil, &block)
      @request_pattern.with(request, &block)
      self
    end

    def to_return(*values)
      responses = [*values].flatten.map { |v| Response::Value.new(v) }
      @response_sequence << GrpcMock::ResponsesSequence.new(responses)
      self
    end

    def to_raise(*exceptions)
      responses = [*exceptions].flatten.map { |e| Response::ExceptionValue.new(e) }
      @response_sequence << GrpcMock::ResponsesSequence.new(responses)
      self
    end

    def response
      if @response_sequence.empty?
        raise GrpcMock::NoResponseError, 'Must be set some values by using #GrpMock::RequestStub#to_run'
      elsif @response_sequence.size == 1
        @response_sequence.first.next
      else
        if @response_sequence.first.end?
          @response_sequence.shift
        end

        @response_sequence.first.next
      end
    end

    # @param path [String]
    # @param request [Object]
    # @return [Bool]
    def match?(path, request)
      @request_pattern.match?(path, request)
    end
  end
end

Version data entries

3 entries across 3 versions & 2 rubygems

Version	Path
gruffish-0.5.0.pre1	lib/grpc_mock/request_stub.rb
grpc_mock-0.4.0	lib/grpc_mock/request_stub.rb
grpc_mock-0.3.1	lib/grpc_mock/request_stub.rb