Exemplo n.º 1
0
 /** @see LoopIterationListener#iterationStart(LoopIterationEvent) */
 @Override
 public void iterationStart(LoopIterationEvent event) {
   // Cannot use getThreadContext() as not cloned per thread
   JMeterVariables variables = JMeterContextService.getContext().getVariables();
   long start = getStart();
   long end = getEnd();
   long increment = getIncrement();
   if (!isPerUser()) {
     synchronized (this) {
       if (globalCounter == Long.MIN_VALUE || globalCounter > end) {
         globalCounter = start;
       }
       variables.put(getVarName(), formatNumber(globalCounter));
       globalCounter += increment;
     }
   } else {
     long current = perTheadNumber.get().longValue();
     if (isResetOnThreadGroupIteration()) {
       int iteration = variables.getIteration();
       Long lastIterationNumber = perTheadLastIterationNumber.get();
       if (iteration != lastIterationNumber.longValue()) {
         // reset
         current = getStart();
       }
       perTheadLastIterationNumber.set(Long.valueOf(iteration));
     }
     variables.put(getVarName(), formatNumber(current));
     current += increment;
     if (current > end) {
       current = start;
     }
     perTheadNumber.set(Long.valueOf(current));
   }
 }
Exemplo n.º 2
0
 /**
  * Creates the variables:<br>
  * basename_gn, where n=0...# of groups<br>
  * basename_g = number of groups (apart from g0)
  */
 private void saveGroups(JMeterVariables vars, String basename, MatchResult match) {
   StringBuilder buf = new StringBuilder();
   buf.append(basename);
   buf.append("_g"); // $NON-NLS-1$
   int pfxlen = buf.length();
   String prevString = vars.get(buf.toString());
   int previous = 0;
   if (prevString != null) {
     try {
       previous = Integer.parseInt(prevString);
     } catch (NumberFormatException e) {
       log.warn("Could not parse " + prevString + " " + e);
     }
   }
   // Note: match.groups() includes group 0
   final int groups = match.groups();
   for (int x = 0; x < groups; x++) {
     buf.append(x);
     vars.put(buf.toString(), match.group(x));
     buf.setLength(pfxlen);
   }
   vars.put(buf.toString(), Integer.toString(groups - 1));
   for (int i = groups; i <= previous; i++) {
     buf.append(i);
     vars.remove(buf.toString()); // remove the remaining _gn vars
     buf.setLength(pfxlen);
   }
 }
Exemplo n.º 3
0
  // @Override
  public SampleResult runTest(JavaSamplerContext arg0) {
    JMeterVariables vars = JMeterContextService.getContext().getVariables();
    vars.put("demo", "demoVariableContent");

    setupValues(arg0);
    SampleResult sampleResult = new SampleResult();
    sampleResult.sampleStart();

    System.out.println("/level/du/parents: " + level + ", " + du + ", " + parents);
    // JUnitCore junit = new JUnitCore();
    // Result result = junit.run(AuthenticationTest.class);

    //		if (result.getFailureCount() > 0) {
    //			sampleResult.setSuccessful(false);
    //			sampleResult.setResponseCode("301");
    //			sampleResult.setResponseMessage("FailureCount: "
    //					+ result.getFailureCount());
    //		} else {
    sampleResult.setSuccessful(true);
    sampleResult.setResponseCodeOK();
    sampleResult.setResponseMessageOK();
    //		}
    sampleResult.sampleEnd();
    return sampleResult;
  }
Exemplo n.º 4
0
 private void saveGroups(MatchResult result) {
   if (result != null) {
     JMeterVariables vars = getVariables();
     for (int x = 0; x < result.groups(); x++) {
       vars.put(name + "_g" + x, result.group(x));
     }
   }
 }
Exemplo n.º 5
0
  /** {@inheritDoc} */
  @Override
  public synchronized String execute(SampleResult previousResult, Sampler currentSampler)
      throws InvalidVariableException {

    if (bshInterpreter == null) // did we find BeanShell?
    {
      throw new InvalidVariableException("BeanShell not found");
    }

    JMeterContext jmctx = JMeterContextService.getContext();
    JMeterVariables vars = jmctx.getVariables();

    String script = ((CompoundVariable) values[0]).execute();
    String varName = ""; // $NON-NLS-1$
    if (values.length > 1) {
      varName = ((CompoundVariable) values[1]).execute().trim();
    }

    String resultStr = ""; // $NON-NLS-1$

    log.debug("Script=" + script);

    try {

      // Pass in some variables
      if (currentSampler != null) {
        bshInterpreter.set("Sampler", currentSampler); // $NON-NLS-1$
      }

      if (previousResult != null) {
        bshInterpreter.set("SampleResult", previousResult); // $NON-NLS-1$
      }

      // Allow access to context and variables directly
      bshInterpreter.set("ctx", jmctx); // $NON-NLS-1$
      bshInterpreter.set("vars", vars); // $NON-NLS-1$
      bshInterpreter.set("props", JMeterUtils.getJMeterProperties()); // $NON-NLS-1$
      bshInterpreter.set("threadName", Thread.currentThread().getName()); // $NON-NLS-1$

      // Execute the script
      Object bshOut = bshInterpreter.eval(script);
      if (bshOut != null) {
        resultStr = bshOut.toString();
      }
      if (vars != null && varName.length() > 0) { // vars will be null on TestPlan
        vars.put(varName, resultStr);
      }
    } catch (Exception ex) // Mainly for bsh.EvalError
    {
      log.warn("Error running BSH script", ex);
    }

    log.debug("Output=" + resultStr);
    return resultStr;
  }
Exemplo n.º 6
0
  /** {@inheritDoc} */
  @Override
  public synchronized String execute(SampleResult previousResult, Sampler currentSampler)
      throws InvalidVariableException {
    JMeterVariables vars = getVariables();
    String res = ((CompoundVariable) values[0]).execute().toUpperCase();

    if (vars != null && values.length > 1) {
      String varName = ((CompoundVariable) values[1]).execute().trim();
      vars.put(varName, res);
    }

    return res;
  }
Exemplo n.º 7
0
 private String generateResult(MatchResult match) {
   saveGroups(match);
   StringBuffer result = new StringBuffer();
   for (int a = 0; a < template.length; a++) {
     if (template[a] instanceof String) {
       result.append(template[a]);
     } else {
       result.append(match.group(((Integer) template[a]).intValue()));
     }
   }
   JMeterVariables vars = getVariables();
   vars.put(name, result.toString());
   return result.toString();
 }
  /** {@inheritDoc} */
  @Override
  public synchronized String execute(SampleResult previousResult, Sampler currentSampler)
      throws InvalidVariableException {

    String fileName = ((CompoundVariable) values[0]).execute();

    String encoding = null; // means platform default
    if (values.length >= ENCODING) {
      encoding = ((CompoundVariable) values[ENCODING - 1]).execute().trim();
      if (encoding.length() <= 0) { // empty encoding, return to platorm default
        encoding = null;
      }
    }

    String myName = ""; // $NON-NLS-1$
    if (values.length >= PARAM_NAME) {
      myName = ((CompoundVariable) values[PARAM_NAME - 1]).execute().trim();
    }

    String myValue = ERR_IND;

    try {
      myValue = FileUtils.readFileToString(new File(fileName), encoding);
    } catch (IOException e) {
      log.warn("Could not read file: " + fileName + " " + e.getMessage(), e);
      throw new JMeterStopThreadException("End of sequence", e);
    }

    if (myName.length() > 0) {
      JMeterVariables vars = getVariables();
      if (vars != null) { // Can be null if called from Config item testEnded() method
        vars.put(myName, myValue);
      }
    }

    if (log.isDebugEnabled()) {
      String tn = Thread.currentThread().getName();
      log.debug(
          tn
              + " name:" //$NON-NLS-1$
              + myName
              + " value:"
              + myValue); //$NON-NLS-1$
    }

    return myValue;
  }
  @Override
  public synchronized void process() {
    if (file == null) {
      log.info("Creating file object: " + getFileName());
      try {
        file = new FileInputStream(getFileName()).getChannel();
      } catch (FileNotFoundException ex) {
        log.error(getFileName(), ex);
        return;
      }
    }
    String rawData;

    try {
      rawData = readNextChunk(getNextChunkSize());
    } catch (EndOfFileException ex) {
      if (getRewindOnEOF()) {
        if (log.isDebugEnabled()) {
          log.debug("Rewind file");
        }
        try {
          file.position(0);
        } catch (IOException ex1) {
          log.error("Cannot rewind", ex1);
        }
        process();
        return;
      } else {
        log.info("End of file reached: " + getFileName());
        if (JMeterContextService.getContext().getThread() != null) {
          JMeterContextService.getContext().getThread().stop();
        }
        throw new RuntimeEOFException("End of file reached", ex);
      }
    } catch (IOException ex) {
      log.error("Error reading next chunk", ex);
      throw new RuntimeException("Error reading next chunk", ex);
    }

    final JMeterVariables vars = JMeterContextService.getContext().getVariables();
    if (vars != null) {
      vars.put(getVarName(), rawData);
    }
  }
  /** {@inheritDoc} */
  @Override
  public synchronized String execute(SampleResult previousResult, Sampler currentSampler)
      throws InvalidVariableException {

    /*
     * boolean fullHostName = false; if (((CompoundFunction) values[0])
     * .execute() .toLowerCase() .equals("true")) { fullHostName = true; }
     */

    String machineName = JMeterUtils.getLocalHostName();

    if (values.length >= 1) { // we have a variable name
      JMeterVariables vars = getVariables();
      if (vars != null) { // May be null if function is used on TestPlan
        String varName = ((CompoundVariable) values[0]).execute().trim();
        if (varName.length() > 0) {
          vars.put(varName, machineName);
        }
      }
    }
    return machineName;
  }
  @Test
  public void testProcess_from_var() {
    System.out.println("process fromvar");
    JMeterContext context = JMeterContextService.getContext();
    JMeterVariables vars = context.getVariables();

    SampleResult res = new SampleResult();
    res.setResponseData("".getBytes());
    context.setPreviousResult(res);

    vars.put("SVAR", json);

    JSONPathExtractor instance = new JSONPathExtractor();
    instance.setDefaultValue("DEFAULT");
    instance.setVar("test");
    instance.setJsonPath("$.store.book[*].author");
    instance.setSubject(JSONPathExtractor.SUBJECT_VARIABLE);
    instance.setSrcVariableName("SVAR");
    instance.process();
    assertEquals(
        "[\"Nigel Rees\",\"Evelyn Waugh\",\"Herman Melville\",\"J. R. R. Tolkien\"]",
        vars.get("test"));
  }
 private void setValues() {
   synchronized (lock) {
     if (log.isDebugEnabled()) {
       log.debug(
           Thread.currentThread().getName() + " Running up named: " + getName()); // $NON-NLS-1$
     }
     PropertyIterator namesIter = getNames().iterator();
     PropertyIterator valueIter = getValues().iterator();
     JMeterVariables jmvars = getThreadContext().getVariables();
     while (namesIter.hasNext() && valueIter.hasNext()) {
       String name = namesIter.next().getStringValue();
       String value = valueIter.next().getStringValue();
       if (log.isDebugEnabled()) {
         log.debug(
             Thread.currentThread().getName()
                 + " saving variable: "
                 + name
                 + "="
                 + value); //$NON-NLS-1$
       }
       jmvars.put(name, value);
     }
   }
 }
Exemplo n.º 13
0
  /**
   * Parses the response data using regular expressions and saving the results into variables for
   * use later in the test.
   *
   * @see org.apache.jmeter.processor.PostProcessor#process()
   */
  @Override
  public void process() {
    initTemplate();
    JMeterContext context = getThreadContext();
    SampleResult previousResult = context.getPreviousResult();
    if (previousResult == null) {
      return;
    }
    log.debug("RegexExtractor processing result");

    // Fetch some variables
    JMeterVariables vars = context.getVariables();
    String refName = getRefName();
    int matchNumber = getMatchNumber();

    final String defaultValue = getDefaultValue();
    if (defaultValue.length() > 0) { // Only replace default if it is provided
      vars.put(refName, defaultValue);
    }
    Perl5Matcher matcher = JMeterUtils.getMatcher();
    String regex = getRegex();
    Pattern pattern = null;
    try {
      pattern = JMeterUtils.getPatternCache().getPattern(regex, Perl5Compiler.READ_ONLY_MASK);
      List<MatchResult> matches = processMatches(pattern, regex, previousResult, matchNumber, vars);
      int prevCount = 0;
      String prevString = vars.get(refName + REF_MATCH_NR);
      if (prevString != null) {
        vars.remove(refName + REF_MATCH_NR); // ensure old value is not left defined
        try {
          prevCount = Integer.parseInt(prevString);
        } catch (NumberFormatException e1) {
          log.warn("Could not parse " + prevString + " " + e1);
        }
      }
      int matchCount = 0; // Number of refName_n variable sets to keep
      try {
        MatchResult match;
        if (matchNumber >= 0) { // Original match behaviour
          match = getCorrectMatch(matches, matchNumber);
          if (match != null) {
            vars.put(refName, generateResult(match));
            saveGroups(vars, refName, match);
          } else {
            // refname has already been set to the default (if present)
            removeGroups(vars, refName);
          }
        } else // < 0 means we save all the matches
        {
          removeGroups(vars, refName); // remove any single matches
          matchCount = matches.size();
          vars.put(refName + REF_MATCH_NR, Integer.toString(matchCount)); // Save the count
          for (int i = 1; i <= matchCount; i++) {
            match = getCorrectMatch(matches, i);
            if (match != null) {
              final String refName_n =
                  new StringBuilder(refName).append(UNDERSCORE).append(i).toString();
              vars.put(refName_n, generateResult(match));
              saveGroups(vars, refName_n, match);
            }
          }
        }
        // Remove any left-over variables
        for (int i = matchCount + 1; i <= prevCount; i++) {
          final String refName_n =
              new StringBuilder(refName).append(UNDERSCORE).append(i).toString();
          vars.remove(refName_n);
          removeGroups(vars, refName_n);
        }
      } catch (RuntimeException e) {
        log.warn("Error while generating result");
      }
    } catch (MalformedCachePatternException e) {
      log.error("Error in pattern: " + regex);
    } finally {
      JMeterUtils.clearMatcherMemory(matcher, pattern);
    }
  }