lib/ruby-jmeter/dsl/counter.rb in ruby-jmeter-2.0.3 vs lib/ruby-jmeter/dsl/counter.rb in ruby-jmeter-2.0.4

@@ -9,23 +9,23 @@
   class Counter
     attr_accessor :doc
     include Helper
 
     def initialize(params={})
-      params[:name] ||= 'Counter'
+      testname = params.kind_of?(Array) ? 'Counter' : (params[:name] || 'Counter')
       @doc = Nokogiri::XML(<<-EOS.strip_heredoc)
-<CounterConfig guiclass="CounterConfigGui" testclass="CounterConfig" testname="#{params[:name]}" enabled="true">
+<CounterConfig guiclass="CounterConfigGui" testclass="CounterConfig" testname="#{testname}" enabled="true">
   <stringProp name="CounterConfig.start"/>
   <stringProp name="CounterConfig.end"/>
   <stringProp name="CounterConfig.incr"/>
   <stringProp name="CounterConfig.name"/>
   <stringProp name="CounterConfig.format"/>
   <boolProp name="CounterConfig.per_user">true</boolProp>
   <boolProp name="CounterConfig.reset_on_tg_iteration">true</boolProp>
 </CounterConfig>)
       EOS
       update params
-      update_at_xpath params if params[:update_at_xpath]
+      update_at_xpath params if params.is_a?(Hash) && params[:update_at_xpath]
     end
   end
 
 end