[RHDM-1237] Ensure all executable model classes are generated - Enabl…

…ed Camel test (kiegroup#2011)
gmunozfe · Feb 17, 2020 · 8255eaa · 8255eaa
1 parent 575e138
commit 8255eaa
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/kie-camel/src/test/java/org/kie/camel/embedded/camel/component/BatchTest.java b/kie-camel/src/test/java/org/kie/camel/embedded/camel/component/BatchTest.java
@@ -43,6 +43,7 @@
 import org.drools.core.common.InternalFactHandle;
 import org.drools.core.impl.StatefulKnowledgeSessionImpl;
 import org.drools.core.util.StringUtils;
+import org.drools.modelcompiler.ExecutableModelProject;
 import org.jbpm.process.core.context.variable.VariableScope;
 import org.jbpm.process.instance.context.variable.VariableScopeInstance;
 import org.junit.After;
@@ -248,7 +249,7 @@ private StatelessKieSession getStatelessKieSessionFromResource(Resource resource
 
         kfs.write(resource);
 
-        KieBuilder kieBuilder = ks.newKieBuilder(kfs).buildAll();
+        KieBuilder kieBuilder = ks.newKieBuilder(kfs).buildAll(ExecutableModelProject.class);
 
         List<Message> errors = kieBuilder.getResults().getMessages(Message.Level.ERROR);
         if (!errors.isEmpty()) {
@@ -268,7 +269,7 @@ private KieSession getKieSessionFromResource(Resource resource) throws Exception
 
         kfs.write(resource);
 
-        KieBuilder kieBuilder = ks.newKieBuilder(kfs).buildAll();
+        KieBuilder kieBuilder = ks.newKieBuilder(kfs).buildAll(ExecutableModelProject.class);
 
         List<Message> errors = kieBuilder.getResults().getMessages(Message.Level.ERROR);
         if (!errors.isEmpty()) {