com.thoughtworks.gauge.processor.MethodExecutionMessageProcessor Maven / Gradle / Ivy
/*----------------------------------------------------------------
* Copyright (c) ThoughtWorks, Inc.
* Licensed under the Apache License, Version 2.0
* See LICENSE.txt in the project root for license information.
*----------------------------------------------------------------*/
package com.thoughtworks.gauge.processor;
import com.thoughtworks.gauge.ClassInstanceManager;
import com.thoughtworks.gauge.ExecutionContext;
import com.thoughtworks.gauge.execution.HooksExecutor;
import com.thoughtworks.gauge.execution.MethodExecutor;
import com.thoughtworks.gauge.hook.Hook;
import gauge.messages.Messages;
import gauge.messages.Spec;
import java.lang.reflect.Method;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
public abstract class MethodExecutionMessageProcessor {
private ClassInstanceManager instanceManager;
public MethodExecutionMessageProcessor(ClassInstanceManager instanceManager) {
this.instanceManager = instanceManager;
}
public Messages.Message execute(Method method, Messages.Message message, Object... args) {
HashSet methods = new HashSet();
methods.add(method);
return execute(methods, message, args);
}
protected ClassInstanceManager getInstanceManager() {
return instanceManager;
}
public Messages.Message execute(Set methods, Messages.Message message, Object... args) {
MethodExecutor methodExecutor = new MethodExecutor(instanceManager);
long totalExecutionTime = 0;
for (Method method : methods) {
Spec.ProtoExecutionResult result = methodExecutor.execute(method, args);
totalExecutionTime += result.getExecutionTime();
if (result.getFailed()) {
return createMessageWithExecutionStatusResponse(message, result);
}
}
Spec.ProtoExecutionResult passingExecution = Spec.ProtoExecutionResult.newBuilder().setFailed(false).setExecutionTime(totalExecutionTime).build();
return createMessageWithExecutionStatusResponse(message, passingExecution);
}
public Messages.Message executeHooks(List hooks, Messages.Message message, ExecutionContext executionInfo) {
Spec.ProtoExecutionResult executionStatusResponse = new HooksExecutor(hooks, executionInfo, instanceManager).execute();
return createMessageWithExecutionStatusResponse(message, executionStatusResponse);
}
public Messages.Message createMessageWithExecutionStatusResponse(Messages.Message receivedMessage, Spec.ProtoExecutionResult result) {
return Messages.Message.newBuilder()
.setMessageId(receivedMessage.getMessageId())
.setMessageType(Messages.Message.MessageType.ExecutionStatusResponse)
.setExecutionStatusResponse(Messages.ExecutionStatusResponse.newBuilder().setExecutionResult(result).build())
.build();
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy