com.neotys.neoload.model.readers.loadrunner.selectionstatement.ConditionContextVisitor Maven / Gradle / Ivy

Go to download

package com.neotys.neoload.model.readers.loadrunner.selectionstatement;

import java.util.List;

import com.neotys.neoload.model.core.Element;
import com.neotys.neoload.model.parsers.CPP14BaseVisitor;
import com.neotys.neoload.model.parsers.CPP14Parser.ConditionContext;
import com.neotys.neoload.model.parsers.CPP14Parser.MethodcallContext;
import com.neotys.neoload.model.readers.loadrunner.LoadRunnerVUVisitor;

public class ConditionContextVisitor extends CPP14BaseVisitor {
	
	private final LoadRunnerVUVisitor visitor;
	
	public ConditionContextVisitor(final LoadRunnerVUVisitor visitor) {
		this.visitor = visitor;
	}

	@Override
	public Element visitCondition(final ConditionContext ctx) {
		return ctx.expression().accept(this);
	}
	
	@Override
	public Element visitMethodcall(MethodcallContext ctx) {
		final List elements = visitor.visitMethodcall(ctx);
		if(elements.size() == 1){
			return elements.get(0);			
		}
		return null;
	}	
}