Contents

# frozen_string_literal: true

require "proto_pharm/request_pattern"
require "proto_pharm/response"
require "proto_pharm/response_sequence"
require "proto_pharm/errors"

module ProtoPharm
  class RequestStub
    attr_reader :received_requests, :request_pattern, :response_sequence

    delegate :path, to: :request_pattern, allow_nil: true

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

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

    def to_return(*values, &block)
      if block_given?
        raise ArgumentError, "Cannot stub with static response if stubbing with a block" if values.any?

        responses = [ Response::Value.new(block) ]
      else
        responses = [*values].flatten.map { |v| Response::Value.new(v) }
      end

      @response_sequence << ProtoPharm::ResponsesSequence.new(responses)

      self
    end

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

    def response
      if @response_sequence.empty?
        raise ProtoPharm::NoResponseError, "Must be set some values by using ProtoPharm::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

    def received!(request)
      @received_requests << request
    end

    def received_count
      received_requests.size
    end

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

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
proto_pharm-0.7.0	lib/proto_pharm/request_stub.rb