use of org.finos.legend.engine.plan.execution.stores.inMemory.plugin.InMemoryStoreExecutionState in project legend-engine by finos.
the class ServiceTestRunner method executeTestAsserts.
@Prometheus(name = "service test execute", doc = "Execution duration summary within service test execution")
private RichServiceTestResult executeTestAsserts(SingleExecutionPlan executionPlan, List<TestContainer> asserts, RichIterable<? extends String> sqlStatements, Scope scope) throws IOException {
long start = System.currentTimeMillis();
if (ExecutionNodeTDSResultHelper.isResultTDS(executionPlan.rootExecutionNode) || (executionPlan.rootExecutionNode.isResultPrimitiveType() && "String".equals(executionPlan.rootExecutionNode.getDataTypeResultType()))) {
// Java
String packageName = "org.finos.legend.tests.generated";
String className = "TestSuite";
String javaCode = ServiceTestGenerationHelper.generateJavaForAsserts(asserts, this.service, this.pureModel, packageName, className);
Class<?> assertsClass;
RichServiceTestResult testRun;
try {
assertsClass = compileJavaForAsserts(packageName, className, javaCode);
} catch (JavaCompileException e) {
MetricsHandler.incrementErrorCount("test_execute", 0);
throw new RuntimeException("Error compiling test asserts for " + this.service.getPath(), e);
}
scope.span().log("Java asserts generated and compiled");
TestExecutionScope execScope = null;
try {
// Setup test database if needed
if (sqlStatements != null) {
execScope = TestExecutionScope.setupTestServer(sqlStatements, scope);
}
// Run tests
Map<String, TestResult> results = Maps.mutable.empty();
Map<String, Exception> assertExceptions = Maps.mutable.empty();
for (Pair<TestContainer, Integer> tc : LazyIterate.zipWithIndex(asserts)) {
// Build Param Map
Map<String, Result> parameters = Maps.mutable.empty();
if (this.service.execution instanceof PureExecution) {
parameters = ListIterate.zip(((PureExecution) this.service.execution).func.parameters, tc.getOne().parametersValues).toMap(p -> p.getOne().name, p -> // Condition evoked in case of studio-flow
p.getTwo() instanceof Collection ? new ConstantResult(ListIterate.collect(((Collection) p.getTwo()).values, v -> v.accept(new ValueSpecificationToResultVisitor()).getValue())) : // Condition evoked in case of pureIDE-flow
p.getTwo() instanceof PureList ? new ConstantResult(ListIterate.collect(((PureList) p.getTwo()).values, v -> v.accept(new ValueSpecificationToResultVisitor()).getValue())) : p.getTwo().accept(new ValueSpecificationToResultVisitor()));
}
// Execute Plan
ExecutionState testExecutionState = new ExecutionState(parameters, Lists.mutable.withAll(executionPlan.templateFunctions), Lists.mutable.with(new RelationalStoreExecutionState(new RelationalStoreState(execScope == null ? -1 : execScope.getPort())), new InMemoryStoreExecutionState(new InMemoryStoreState()), new ServiceStoreExecutionState(new ServiceStoreState())));
Result result = this.executor.execute(executionPlan, testExecutionState, null, null);
org.finos.legend.pure.m3.coreinstance.meta.pure.mapping.Result<Object> pureResult = result.accept(new ResultToPureResultVisitor());
// Execute Assert
String testName = ServiceTestGenerationHelper.getAssertMethodName(tc.getTwo());
scope.span().setTag(testName, resultToString(pureResult, this.pureModel.getExecutionSupport()));
TestResult testResult;
try {
Boolean assertResult = (Boolean) assertsClass.getMethod(testName, org.finos.legend.pure.m3.coreinstance.meta.pure.mapping.Result.class, ExecutionSupport.class).invoke(null, pureResult, pureModel.getExecutionSupport());
testResult = assertResult ? TestResult.SUCCESS : TestResult.FAILURE;
scope.span().setTag(testName + "_assert", assertResult);
} catch (Exception e) {
StringWriter out = new StringWriter();
PrintWriter writer = new PrintWriter(out);
e.printStackTrace(writer);
e.printStackTrace();
testResult = TestResult.ERROR;
assertExceptions.put(testName, e);
scope.span().setTag(testName + "_assert", out.toString());
}
results.put(testName, testResult);
}
testRun = new RichServiceTestResult(service.getPath(), results, assertExceptions, null, executionPlan, javaCode);
scope.span().log("Finished running tests " + results);
MetricsHandler.observeServerOperation("test_execute", metricsContext, start, System.currentTimeMillis());
} catch (Exception e) {
LOGGER.error("Error running tests", e);
MetricsHandler.incrementErrorCount("test_execute", 0);
throw (e instanceof RuntimeException) ? (RuntimeException) e : new RuntimeException(e);
} finally {
if (execScope != null) {
execScope.close();
}
MetricsHandler.observe("service test execute", start, System.currentTimeMillis());
}
return testRun;
} else {
return new RichServiceTestResult(this.service.getPath(), Collections.emptyMap(), Collections.emptyMap(), null, executionPlan, "");
}
}
Aggregations