fitnesse.testsystems.slim.SlimTestContextImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of fitnesse Show documentation
The fully integrated standalone wiki, and acceptance testing framework.
The newest version!
// Copyright (C) 2003-2009 by Object Mentor, Inc. All rights reserved.
// Released under the terms of the CPL Common Public License version 1.0.
package fitnesse.testsystems.slim;

import fitnesse.testsystems.ExecutionResult;
import fitnesse.testsystems.TestPage;
import fitnesse.testsystems.TestSummary;
import fitnesse.testsystems.slim.tables.ScenarioTable;
import fitnesse.testsystems.slim.tables.ScriptTable;
import fitnesse.util.TimeMeasurement;

import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.Comparator;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class SlimTestContextImpl implements SlimTestContext {
  private final Map symbols = new HashMap<>();
  private final Map scenarios = new HashMap<>(512);
  private final TestSummary testSummary = new TestSummary();
  private final TestPage pageToTest;
  private final TimeMeasurement timeMeasurement;
  private List scenariosWithInputs = null;
  private boolean isSorted = true;
  private String currentScriptActor;
  private Class currentScriptClass = ScriptTable.class;

  public SlimTestContextImpl(TestPage pageToTest) {
    this.pageToTest = pageToTest;
    this.timeMeasurement = new TimeMeasurement().start();
  }

  @Override
  public String getSymbol(String symbolName) {
    if (symbolName.startsWith("SECRET_")) {
      return "*****";
    }
    return symbols.get(symbolName);
  }

  @Override
  public Map getSymbols() {
    return symbols;
  }

  @Override
  public void setSymbol(String symbolName, String value) {
    symbols.put(symbolName, value);
  }

  @Override
  public void addScenario(String scenarioName, ScenarioTable scenarioTable) {
    ScenarioTable oldValue = scenarios.put(scenarioName, scenarioTable);
    if (scenariosWithInputs != null) {
      maintainScenariosWithInputs(oldValue, scenarioTable);
    }
  }

  @Override
  public ScenarioTable getScenario(String scenarioName) {
    return scenarios.get(scenarioName);
  }

  @Override
  public ScenarioTable getScenarioByPattern(String invokingString) {
    ScenarioTable result = null;
    for (ScenarioTable s : getScenariosWithMostArgumentsFirst()) {
      if (s.canMatchParameters(invokingString)) {
        result = s;
        break;
      }
    }
    return result;
  }

  private List getScenariosWithMostArgumentsFirst() {
    if (scenariosWithInputs == null) {
      initializeScenariosWithInputs();
    }
    if (!isSorted) {
      Collections.sort(scenariosWithInputs, new ScenarioTableLengthComparator());
      isSorted = true;
    }
    return scenariosWithInputs;
  }

  private static class ScenarioTableLengthComparator implements Comparator {
    @Override
    public int compare(ScenarioTable st1, ScenarioTable st2) {
      int size1 = st1.getInputs().size();
      int size2 = st2.getInputs().size();
      return size2 - size1;
    }
  }

  private void initializeScenariosWithInputs() {
    int initialCapacity = scenarios.size();
    scenariosWithInputs = new ArrayList<>(initialCapacity);
    isSorted = true;
    for (ScenarioTable table : scenarios.values()) {
      addToScenariosWithInputsIfNeeded(table);
    }
  }

  private void maintainScenariosWithInputs(ScenarioTable oldTable, ScenarioTable newTable) {
    if (oldTable != null && !oldTable.getInputs().isEmpty()) {
      scenariosWithInputs.remove(oldTable);
    }
    addToScenariosWithInputsIfNeeded(newTable);
  }

  private void addToScenariosWithInputsIfNeeded(ScenarioTable newTable) {
    if (!newTable.getInputs().isEmpty()) {
      scenariosWithInputs.add(newTable);
      isSorted = false;
    }
  }

  @Override
  public Collection getScenarios() {
    return scenarios.values();
  }

  @Override
  public void incrementPassedTestsCount() {
    increment(ExecutionResult.PASS);
  }

  @Override
  public void incrementFailedTestsCount() {
    increment(ExecutionResult.FAIL);
  }

  @Override
  public void incrementErroredTestsCount() {
    increment(ExecutionResult.ERROR);
  }

  @Override
  public void incrementIgnoredTestsCount() {
    increment(ExecutionResult.IGNORE);
  }

  @Override
  public void increment(ExecutionResult result) {
    this.testSummary.add(result);
  }

  @Override
  public void increment(TestSummary testSummary) {
    this.testSummary.add(testSummary);
  }

  public TestSummary getTestSummary() {
    testSummary.setRunTimeInMillis(timeMeasurement.elapsed());
    return testSummary;
  }

  @Override
  public TestPage getPageToTest() {
    return pageToTest;
  }

  @Override
  public void setCurrentScriptClass(Class currentScriptClass) {
    this.currentScriptClass = currentScriptClass;
  }

  @Override
  public Class getCurrentScriptClass() {
    return currentScriptClass;
  }

  @Override
  public void setCurrentScriptActor(String currentScriptActor) {
    this.currentScriptActor = currentScriptActor;
  }

  @Override
  public String getCurrentScriptActor() {
    return currentScriptActor;
  }
}