# frozen_string_literal: true

module Benchmark
  module Gems
    class SNS < Benchmark::Gem

      def gem_name
        'aws-sdk-sns'
      end

      def client_module_name
        :SNS
      end

      def operation_benchmarks
        {
          publish_small: {
            setup: proc do |client|
              {topic_arn: 'topic', subject: "subject", message: 'message'}
            end,
            test: proc do |client, req|
              client.publish(req)
            end
          },
          publish_large: {
            setup: proc do |client|
              {topic_arn: 'topic', subject: "subject", message: 'message'*1000, message_attributes: (0..100).map { |i| ["key#{i}", {data_type: "String", string_value: "string data#{i}"}]}.to_h}
            end,
            test: proc do |client, req|
              client.publish(req)
            end
          },
        }
      end
    end
  end
end