Contents

module Aws
  module Plugins

    # @seahorse.client.option [Boolean] :stub_responses (false)
    #   Causes the client to return stubbed responses. By default
    #   fake responses are generated and returned. You can specify
    #   the response data to return or errors to raise by calling
    #   {ClientStubs#stub_responses}. See {ClientStubs} for more information.
    #
    #   ** Please note ** When response stubbing is enabled, no HTTP
    #   requests are made, and retries are disabled.
    #
    class StubResponses < Seahorse::Client::Plugin

      option(:stub_responses, false)

      def add_handlers(handlers, config)
        handlers.add(Handler, step: :send) if config.stub_responses
      end

      def after_initialize(client)
        # disable retries when stubbing responses
        client.config.retry_limit = 0 if client.config.stub_responses
      end

      class Handler < Seahorse::Client::Handler

        def call(context)
          response = Seahorse::Client::Response.new(context: context)
          apply_stub(response, context.client.next_stub(context.operation_name))
          response
        end

        private

        def apply_stub(resp, stub)
          if Exception === stub
            resp.error = stub
          elsif Class === stub && stub.ancestors.include?(Errors::ServiceError)
            resp.error = stub.new(resp.context, 'stubbed error')
          elsif Class === stub && stub.ancestors.include?(Exception)
            resp.error = stub.new
          else
            resp.data = stub
          end
        end

      end
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
aws-sdk-core-2.0.3	lib/aws-sdk-core/plugins/stub_responses.rb
aws-sdk-core-2.0.2	lib/aws-sdk-core/plugins/stub_responses.rb