Sha256: 49ea5b2ddf6a592551999182d10cb94b8265fbbd9985f9ea8614989e9c507cb2

Contents?: true

Size: 1.79 KB

Versions: 70

Compression:

Stored size: 1.79 KB

Contents

# frozen_string_literal: true

module Aws
  # @api private
  module Stubbing
    class StubData

      def initialize(operation)
        @rules = operation.output
        @pager = operation[:pager]
      end

      def stub(data = {})
        stub = EmptyStub.new(@rules).stub
        remove_paging_tokens(stub)
        remove_checksums(stub)
        apply_data(data, stub)
        stub
      end

      private

      def remove_checksums(stub)
        if @rules && @rules.shape.is_a?(Seahorse::Model::Shapes::StructureShape)
          @rules.shape.members.each do |key, member|
            if member.location == 'header' && member.location_name.start_with?('x-amz-checksum-')
              stub[key] = nil
            end
          end
        end
      end

      def remove_paging_tokens(stub)
        if @pager
          @pager.instance_variable_get("@tokens").keys.each do |path|
            if divide = (path[' || '] || path[' or '])
              path = path.split(divide)[0]
            end
            parts = path.split(/\b/)
            # if nested struct/expression, EmptyStub auto-pop "string"
            # currently not support remove "string" for nested/expression
            # as it requires reverse JMESPATH search
            stub[parts[0]] = nil if parts.size == 1
          end
          if more_results = @pager.instance_variable_get('@more_results')
            parts = more_results.split(/\b/)
            # if nested struct/expression, EmptyStub auto-pop false value
            # no further work needed
            stub[parts[0]] = false if parts.size == 1
          end
        end
      end

      def apply_data(data, stub)
        ParamValidator.new(@rules, validate_required: false, input: false).validate!(data)
        DataApplicator.new(@rules).apply_data(data, stub)
      end
    end
  end
end

Version data entries

70 entries across 70 versions & 1 rubygems

Version Path
aws-sdk-core-3.218.1 lib/aws-sdk-core/stubbing/stub_data.rb
aws-sdk-core-3.218.0 lib/aws-sdk-core/stubbing/stub_data.rb
aws-sdk-core-3.217.1 lib/aws-sdk-core/stubbing/stub_data.rb
aws-sdk-core-3.217.0 lib/aws-sdk-core/stubbing/stub_data.rb
aws-sdk-core-3.216.1 lib/aws-sdk-core/stubbing/stub_data.rb
aws-sdk-core-3.216.0 lib/aws-sdk-core/stubbing/stub_data.rb
aws-sdk-core-3.215.1 lib/aws-sdk-core/stubbing/stub_data.rb
aws-sdk-core-3.215.0 lib/aws-sdk-core/stubbing/stub_data.rb
aws-sdk-core-3.214.1 lib/aws-sdk-core/stubbing/stub_data.rb
aws-sdk-core-3.214.0 lib/aws-sdk-core/stubbing/stub_data.rb
aws-sdk-core-3.213.0 lib/aws-sdk-core/stubbing/stub_data.rb
aws-sdk-core-3.212.0 lib/aws-sdk-core/stubbing/stub_data.rb
aws-sdk-core-3.211.0 lib/aws-sdk-core/stubbing/stub_data.rb
aws-sdk-core-3.210.0 lib/aws-sdk-core/stubbing/stub_data.rb
aws-sdk-core-3.209.1 lib/aws-sdk-core/stubbing/stub_data.rb
aws-sdk-core-3.209.0 lib/aws-sdk-core/stubbing/stub_data.rb
aws-sdk-core-3.208.0 lib/aws-sdk-core/stubbing/stub_data.rb
aws-sdk-core-3.207.0 lib/aws-sdk-core/stubbing/stub_data.rb
aws-sdk-core-3.206.0 lib/aws-sdk-core/stubbing/stub_data.rb
aws-sdk-core-3.205.0 lib/aws-sdk-core/stubbing/stub_data.rb