org.testng.internal.ScriptSelectorFactory Maven / Gradle / Ivy

package org.testng.internal;

import java.util.Map;
import java.util.ServiceLoader;
import javax.script.ScriptEngineFactory;
import org.testng.TestNGException;
import org.testng.collections.Maps;
import org.testng.xml.XmlScript;

public final class ScriptSelectorFactory {

  private static final Map ENGINE_FACTORIES = Maps.newHashMap();

  private ScriptSelectorFactory() {}

  public static ScriptMethodSelector getScriptSelector(XmlScript script) {

    if (script == null || script.getLanguage() == null) {
      throw new IllegalArgumentException("Language name must not be null");
    }

    String languageName = script.getLanguage().toLowerCase();
    ScriptEngineFactory engineFactory = ENGINE_FACTORIES.get(languageName);
    if (engineFactory == null) {
      ServiceLoader loader = ServiceLoader.load(ScriptEngineFactory.class);
      for (ScriptEngineFactory factory : loader) {
        ENGINE_FACTORIES.put(factory.getLanguageName().toLowerCase(), factory);
      }

      engineFactory = ENGINE_FACTORIES.get(languageName);
    }

    if (engineFactory == null) {
      throw new TestNGException(
          "No engine found for language: "
              + script.getLanguage()
              + ". "
              + "Please check your dependencies and have a look at "
              + "https://github.com/cbeust/testng/wiki/Supported-script-engines");
    }

    return new ScriptMethodSelector(engineFactory.getScriptEngine(), script.getExpression());
  }
}