fix PMMLRestResourceGeneratorTest

tiagodolphine · Nov 16, 2020 · 5e4aa33 · 5e4aa33
1 parent 93bfcb2
commit 5e4aa33
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/...odegen/src/test/java/org/kie/kogito/codegen/prediction/PMMLRestResourceGeneratorTest.java b/...odegen/src/test/java/org/kie/kogito/codegen/prediction/PMMLRestResourceGeneratorTest.java
@@ -33,6 +33,7 @@
 import org.junit.jupiter.api.BeforeAll;
 import org.junit.jupiter.api.Test;
 import org.kie.dmn.feel.codegen.feel11.CodegenStringUtil;
+import org.kie.kogito.codegen.di.CDIDependencyInjectionAnnotator;
 import org.kie.kogito.codegen.di.DependencyInjectionAnnotator;
 import org.kie.pmml.commons.model.KiePMMLModel;
 
@@ -88,7 +89,7 @@ void constructor() {
 
     @Test
     void generateWithDependencyInjection() {
-        String retrieved = pmmlRestResourceGenerator.withDependencyInjection(mock(DependencyInjectionAnnotator.class)).generate();
+        String retrieved = pmmlRestResourceGenerator.withDependencyInjection(new CDIDependencyInjectionAnnotator()).generate();
         commonEvaluateGenerate(retrieved);
         String expected = "Application application;";
         assertTrue(retrieved.contains(expected));
@@ -105,7 +106,7 @@ void generateWithoutDependencyInjection() {
     @Test
     void getNameURL() {
         String classPrefix = getSanitizedClassName(KIE_PMML_MODEL.getName());
-        String expected = URLEncoder.encode(classPrefix).replaceAll("\\+", "%20");
+        String expected = URLEncoder.encode(classPrefix).replaceAll("\\+", " ");
         assertEquals(expected, pmmlRestResourceGenerator.getNameURL());
     }
 
@@ -157,7 +158,7 @@ void setPathValue() {
         pmmlRestResourceGenerator.setPathValue(template);
         try {
             String classPrefix = getSanitizedClassName(KIE_PMML_MODEL.getName());
-            String expected = URLEncoder.encode(classPrefix).replaceAll("\\+", "%20");
+            String expected = URLEncoder.encode(classPrefix).replaceAll("\\+", " ");
             assertEquals(expected, retrieved.getMemberValue().asStringLiteralExpr().asString());
         } catch (Exception e) {
             fail(e);