ruby-jmeter 2.0

data/lib/ruby-jmeter/dsl/transaction_controller.rb

@@ -0,0 +1,26 @@
+module RubyJmeter
+  class DSL
+    def transaction_controller(params={}, &block)
+      node = RubyJmeter::TransactionController.new(params)
+      attach_node(node, &block)
+    end
+  end
+
+  class TransactionController
+    attr_accessor :doc
+    include Helper
+
+    def initialize(params={})
+      params[:name] ||= 'TransactionController'
+      @doc = Nokogiri::XML(<<-EOS.strip_heredoc)
+<TransactionController guiclass="TransactionControllerGui" testclass="TransactionController" testname="#{params[:name]}" enabled="true">
+  <boolProp name="TransactionController.parent">true</boolProp>
+  <boolProp name="TransactionController.includeTimers">false</boolProp>
+</TransactionController>)
+      EOS
+      update params
+      update_at_xpath params if params[:update_at_xpath]
+    end
+  end
+
+end

data/lib/ruby-jmeter/dsl/uniform_random_timer.rb

@@ -0,0 +1,26 @@
+module RubyJmeter
+  class DSL
+    def uniform_random_timer(params={}, &block)
+      node = RubyJmeter::UniformRandomTimer.new(params)
+      attach_node(node, &block)
+    end
+  end
+
+  class UniformRandomTimer
+    attr_accessor :doc
+    include Helper
+
+    def initialize(params={})
+      params[:name] ||= 'UniformRandomTimer'
+      @doc = Nokogiri::XML(<<-EOS.strip_heredoc)
+<UniformRandomTimer guiclass="UniformRandomTimerGui" testclass="UniformRandomTimer" testname="#{params[:name]}" enabled="true">
+  <stringProp name="ConstantTimer.delay">0</stringProp>
+  <stringProp name="RandomTimer.range">100.0</stringProp>
+</UniformRandomTimer>)
+      EOS
+      update params
+      update_at_xpath params if params[:update_at_xpath]
+    end
+  end
+
+end

data/lib/ruby-jmeter/dsl/user_defined_variables.rb

@@ -0,0 +1,32 @@
+module RubyJmeter
+  class DSL
+    def user_defined_variables(params={}, &block)
+      node = RubyJmeter::UserDefinedVariables.new(params)
+      attach_node(node, &block)
+    end
+  end
+
+  class UserDefinedVariables
+    attr_accessor :doc
+    include Helper
+
+    def initialize(params={})
+      params[:name] ||= 'UserDefinedVariables'
+      @doc = Nokogiri::XML(<<-EOS.strip_heredoc)
+<Arguments guiclass="ArgumentsPanel" testclass="Arguments" testname="#{params[:name]}" enabled="true">
+  <collectionProp name="Arguments.arguments">
+    <elementProp name="testguid" elementType="Argument">
+      <stringProp name="Argument.name"/>
+      <stringProp name="Argument.value"/>
+      <stringProp name="Argument.metadata">=</stringProp>
+    </elementProp>
+  </collectionProp>
+  <stringProp name="TestPlan.comments"/>
+</Arguments>)
+      EOS
+      update params
+      update_at_xpath params if params[:update_at_xpath]
+    end
+  end
+
+end

data/lib/ruby-jmeter/dsl/user_parameters.rb

@@ -0,0 +1,29 @@
+module RubyJmeter
+  class DSL
+    def user_parameters(params={}, &block)
+      node = RubyJmeter::UserParameters.new(params)
+      attach_node(node, &block)
+    end
+  end
+
+  class UserParameters
+    attr_accessor :doc
+    include Helper
+
+    def initialize(params={})
+      params[:name] ||= 'UserParameters'
+      @doc = Nokogiri::XML(<<-EOS.strip_heredoc)
+<UserParameters guiclass="UserParametersGui" testclass="UserParameters" testname="#{params[:name]}" enabled="true">
+  <collectionProp name="UserParameters.names"/>
+  <collectionProp name="UserParameters.thread_values">
+    <collectionProp name="1"/>
+  </collectionProp>
+  <boolProp name="UserParameters.per_iteration">false</boolProp>
+</UserParameters>)
+      EOS
+      update params
+      update_at_xpath params if params[:update_at_xpath]
+    end
+  end
+
+end

data/lib/ruby-jmeter/dsl/view_results_in_table.rb

@@ -0,0 +1,55 @@
+module RubyJmeter
+  class DSL
+    def view_results_in_table(params={}, &block)
+      node = RubyJmeter::ViewResultsInTable.new(params)
+      attach_node(node, &block)
+    end
+  end
+
+  class ViewResultsInTable
+    attr_accessor :doc
+    include Helper
+
+    def initialize(params={})
+      params[:name] ||= 'ViewResultsInTable'
+      @doc = Nokogiri::XML(<<-EOS.strip_heredoc)
+<ResultCollector guiclass="TableVisualizer" testclass="ResultCollector" testname="#{params[:name]}" enabled="true">
+  <boolProp name="ResultCollector.error_logging">false</boolProp>
+  <objProp>
+    <name>saveConfig</name>
+    <value class="SampleSaveConfiguration">
+      <time>true</time>
+      <latency>true</latency>
+      <timestamp>true</timestamp>
+      <success>true</success>
+      <label>true</label>
+      <code>true</code>
+      <message>false</message>
+      <threadName>true</threadName>
+      <dataType>false</dataType>
+      <encoding>false</encoding>
+      <assertions>false</assertions>
+      <subresults>false</subresults>
+      <responseData>false</responseData>
+      <samplerData>false</samplerData>
+      <xml>false</xml>
+      <fieldNames>false</fieldNames>
+      <responseHeaders>false</responseHeaders>
+      <requestHeaders>false</requestHeaders>
+      <responseDataOnError>false</responseDataOnError>
+      <saveAssertionResultsFailureMessage>false</saveAssertionResultsFailureMessage>
+      <assertionsResultsToSave>0</assertionsResultsToSave>
+      <bytes>true</bytes>
+      <threadCounts>true</threadCounts>
+      <sampleCount>true</sampleCount>
+    </value>
+  </objProp>
+  <stringProp name="filename"/>
+</ResultCollector>)
+      EOS
+      update params
+      update_at_xpath params if params[:update_at_xpath]
+    end
+  end
+
+end

data/lib/ruby-jmeter/dsl/view_results_tree.rb

@@ -0,0 +1,55 @@
+module RubyJmeter
+  class DSL
+    def view_results_tree(params={}, &block)
+      node = RubyJmeter::ViewResultsTree.new(params)
+      attach_node(node, &block)
+    end
+  end
+
+  class ViewResultsTree
+    attr_accessor :doc
+    include Helper
+
+    def initialize(params={})
+      params[:name] ||= 'ViewResultsTree'
+      @doc = Nokogiri::XML(<<-EOS.strip_heredoc)
+<ResultCollector guiclass="ViewResultsFullVisualizer" testclass="ResultCollector" testname="#{params[:name]}" enabled="true">
+  <boolProp name="ResultCollector.error_logging">false</boolProp>
+  <objProp>
+    <name>saveConfig</name>
+    <value class="SampleSaveConfiguration">
+      <time>true</time>
+      <latency>true</latency>
+      <timestamp>true</timestamp>
+      <success>true</success>
+      <label>true</label>
+      <code>true</code>
+      <message>false</message>
+      <threadName>true</threadName>
+      <dataType>false</dataType>
+      <encoding>false</encoding>
+      <assertions>false</assertions>
+      <subresults>false</subresults>
+      <responseData>false</responseData>
+      <samplerData>false</samplerData>
+      <xml>false</xml>
+      <fieldNames>false</fieldNames>
+      <responseHeaders>false</responseHeaders>
+      <requestHeaders>false</requestHeaders>
+      <responseDataOnError>false</responseDataOnError>
+      <saveAssertionResultsFailureMessage>false</saveAssertionResultsFailureMessage>
+      <assertionsResultsToSave>0</assertionsResultsToSave>
+      <bytes>true</bytes>
+      <threadCounts>true</threadCounts>
+      <sampleCount>true</sampleCount>
+    </value>
+  </objProp>
+  <stringProp name="filename"/>
+</ResultCollector>)
+      EOS
+      update params
+      update_at_xpath params if params[:update_at_xpath]
+    end
+  end
+
+end

data/lib/ruby-jmeter/dsl/while_controller.rb

@@ -0,0 +1,25 @@
+module RubyJmeter
+  class DSL
+    def while_controller(params={}, &block)
+      node = RubyJmeter::WhileController.new(params)
+      attach_node(node, &block)
+    end
+  end
+
+  class WhileController
+    attr_accessor :doc
+    include Helper
+
+    def initialize(params={})
+      params[:name] ||= 'WhileController'
+      @doc = Nokogiri::XML(<<-EOS.strip_heredoc)
+<WhileController guiclass="WhileControllerGui" testclass="WhileController" testname="#{params[:name]}" enabled="true">
+  <stringProp name="WhileController.condition"/>
+</WhileController>)
+      EOS
+      update params
+      update_at_xpath params if params[:update_at_xpath]
+    end
+  end
+
+end

data/lib/ruby-jmeter/dsl/xml_assertion.rb

@@ -0,0 +1,23 @@
+module RubyJmeter
+  class DSL
+    def xml_assertion(params={}, &block)
+      node = RubyJmeter::XmlAssertion.new(params)
+      attach_node(node, &block)
+    end
+  end
+
+  class XmlAssertion
+    attr_accessor :doc
+    include Helper
+
+    def initialize(params={})
+      params[:name] ||= 'XmlAssertion'
+      @doc = Nokogiri::XML(<<-EOS.strip_heredoc)
+<XMLAssertion guiclass="XMLAssertionGui" testclass="XMLAssertion" testname="#{params[:name]}" enabled="true"/>)
+      EOS
+      update params
+      update_at_xpath params if params[:update_at_xpath]
+    end
+  end
+
+end

data/lib/ruby-jmeter/dsl/xml_schema_assertion.rb

@@ -0,0 +1,25 @@
+module RubyJmeter
+  class DSL
+    def xml_schema_assertion(params={}, &block)
+      node = RubyJmeter::XmlSchemaAssertion.new(params)
+      attach_node(node, &block)
+    end
+  end
+
+  class XmlSchemaAssertion
+    attr_accessor :doc
+    include Helper
+
+    def initialize(params={})
+      params[:name] ||= 'XmlSchemaAssertion'
+      @doc = Nokogiri::XML(<<-EOS.strip_heredoc)
+<XMLSchemaAssertion guiclass="XMLSchemaAssertionGUI" testclass="XMLSchemaAssertion" testname="#{params[:name]}" enabled="true">
+  <stringProp name="xmlschema_assertion_filename"/>
+</XMLSchemaAssertion>)
+      EOS
+      update params
+      update_at_xpath params if params[:update_at_xpath]
+    end
+  end
+
+end

data/lib/ruby-jmeter/dsl/xpath_assertion.rb

@@ -0,0 +1,30 @@
+module RubyJmeter
+  class DSL
+    def xpath_assertion(params={}, &block)
+      node = RubyJmeter::XpathAssertion.new(params)
+      attach_node(node, &block)
+    end
+  end
+
+  class XpathAssertion
+    attr_accessor :doc
+    include Helper
+
+    def initialize(params={})
+      params[:name] ||= 'XpathAssertion'
+      @doc = Nokogiri::XML(<<-EOS.strip_heredoc)
+<XPathAssertion guiclass="XPathAssertionGui" testclass="XPathAssertion" testname="#{params[:name]}" enabled="true">
+  <boolProp name="XPath.negate">false</boolProp>
+  <stringProp name="XPath.xpath">/</stringProp>
+  <boolProp name="XPath.validate">false</boolProp>
+  <boolProp name="XPath.whitespace">false</boolProp>
+  <boolProp name="XPath.tolerant">false</boolProp>
+  <boolProp name="XPath.namespace">false</boolProp>
+</XPathAssertion>)
+      EOS
+      update params
+      update_at_xpath params if params[:update_at_xpath]
+    end
+  end
+
+end

data/lib/ruby-jmeter/dsl/xpath_extractor.rb

@@ -0,0 +1,30 @@
+module RubyJmeter
+  class DSL
+    def xpath_extractor(params={}, &block)
+      node = RubyJmeter::XpathExtractor.new(params)
+      attach_node(node, &block)
+    end
+  end
+
+  class XpathExtractor
+    attr_accessor :doc
+    include Helper
+
+    def initialize(params={})
+      params[:name] ||= 'XpathExtractor'
+      @doc = Nokogiri::XML(<<-EOS.strip_heredoc)
+<XPathExtractor guiclass="XPathExtractorGui" testclass="XPathExtractor" testname="#{params[:name]}" enabled="true">
+  <stringProp name="XPathExtractor.default"/>
+  <stringProp name="XPathExtractor.refname"/>
+  <stringProp name="XPathExtractor.xpathQuery"/>
+  <boolProp name="XPathExtractor.validate">false</boolProp>
+  <boolProp name="XPathExtractor.tolerant">false</boolProp>
+  <boolProp name="XPathExtractor.namespace">false</boolProp>
+</XPathExtractor>)
+      EOS
+      update params
+      update_at_xpath params if params[:update_at_xpath]
+    end
+  end
+
+end

data/lib/ruby-jmeter/helpers/fallback_content_proxy.rb

@@ -0,0 +1,54 @@
+require 'set'
+
+module RubyJmeter
+
+  class FallbackContextProxy
+    NON_PROXIED_METHODS = Set[:object_id, :__send__, :__id__, :==, :equal?, :"!", :"!=", :instance_eval,
+                              :instance_variables, :instance_variable_get, :instance_variable_set,
+                              :remove_instance_variable]
+
+    NON_PROXIED_INSTANCE_VARIABLES = Set[:@__receiver__, :@__fallback__]
+
+    instance_methods.each do |method|
+      unless NON_PROXIED_METHODS.include?(method.to_sym)
+        undef_method(method)
+      end
+    end
+
+    def initialize(receiver, fallback)
+      @__receiver__ = receiver
+      @__fallback__ = fallback
+    end
+
+    def id
+      @__receiver__.__send__(:id)
+    end
+
+    # Special case due to `Kernel#sub`'s existence
+    def sub(*args, &block)
+      __proxy_method__(:sub, *args, &block)
+    end
+
+    # Special case to allow proxy instance variables
+    def instance_variables
+      # Ruby 1.8.x returns string names, convert to symbols
+      super.map(&:to_sym) - NON_PROXIED_INSTANCE_VARIABLES.to_a
+    end
+
+    def method_missing(method, *args, &block)
+      __proxy_method__(method, *args, &block)
+    end
+
+    def __proxy_method__(method, *args, &block)
+      begin
+        @__receiver__.__send__(method.to_sym, *args, &block)
+      rescue ::NoMethodError => e
+        begin
+          @__fallback__.__send__(method.to_sym, *args, &block)
+        rescue ::NoMethodError
+          raise(e)
+        end
+      end
+    end
+  end
+end

data/lib/ruby-jmeter/helpers/helper.rb

@@ -0,0 +1,41 @@
+module RubyJmeter
+
+  def dsl_eval(dsl, &block)
+    block_context = eval("self", block.binding)
+    proxy_context = RubyJmeter::FallbackContextProxy.new(dsl, block_context)
+    begin
+      block_context.instance_variables.each { |ivar| proxy_context.instance_variable_set(ivar, block_context.instance_variable_get(ivar)) }
+      proxy_context.instance_eval(&block)
+    ensure
+      block_context.instance_variables.each { |ivar| block_context.instance_variable_set(ivar, proxy_context.instance_variable_get(ivar)) }
+    end
+    dsl
+  end
+
+  module_function :dsl_eval
+
+  module Helper
+    def update(params)
+      params.delete(:name)
+      params.each do |name, value|
+        node = @doc.children.xpath("//*[contains(@name,\"#{name.to_s}\")]")
+        node.first.content = value unless node.empty?
+      end
+    end
+
+    def update_at_xpath(params)
+      params[:update_at_xpath].each do |fragment|
+        @doc.at_xpath(fragment[:xpath]) << fragment[:value]
+      end
+    end
+
+    def enabled(params)
+      #default to true unless explicitly set to false
+      if params.has_key?(:enabled) && params[:enabled] == false
+        'false'
+      else
+        'true'
+      end
+    end
+  end
+end