module JmeterPerf
  module Plugins
    class RedisDataSet
      attr_accessor :doc
      include JmeterPerf::Helpers::XmlDocumentUpdater
      def initialize(params = {})
        testname = params.is_a?(Array) ? "Redis Data Set Config" : (params[:name] || "Redis Data Set Config")
        params[:getMode] ||= "1" unless params[:remove] == true
        @doc = Nokogiri::XML(
          JmeterPerf::Helpers::String.strip_heredoc(
            <<-XML
              <kg.apc.jmeter.config.redis.RedisDataSet guiclass="TestBeanGUI" testclass="kg.apc.jmeter.config.redis.RedisDataSet" testname="#{testname}" enabled="true">
                <stringProp name="database">0</stringProp>
                <stringProp name="delimiter">,</stringProp>
                <intProp name="getMode">0</intProp>
                <stringProp name="host"></stringProp
                <intProp name="maxActive">20</intProp>
                <intProp name="maxIdle">10</intProp>
                <longProp name="maxWait">30000</longProp>
                <longProp name="minEvictableIdleTimeMillis">60000</longProp>
                <intProp name="minIdle">0</intProp>
                <intProp name="numTestsPerEvictionRun">0</intProp>
                <stringProp name="password"></stringProp>
                <stringProp name="port">6379</stringProp>
                <stringProp name="redisKey"></stringProp>
                <longProp name="softMinEvictableIdleTimeMillis">60000</longProp>
                <boolProp name="testOnBorrow">false</boolProp>
                <boolProp name="testOnReturn">false</boolProp>
                <boolProp name="testWhileIdle">false</boolProp>
                <longProp name="timeBetweenEvictionRunsMillis">30000</longProp>
                <stringProp name="timeout">2000</stringProp>
                <stringProp name="variableNames"></stringProp>
                <intProp name="whenExhaustedAction">2</intProp>
              </kg.apc.jmeter.config.redis.RedisDataSet>
            XML
          )
        )
        update params
        update_at_xpath params if params.is_a?(Hash) && params[:update_at_xpath]
      end
    end
  end
end