Change method of accept method test.

apache · Oct 14, 2024 · e793f7e · e793f7e
1 parent de521ba
commit e793f7e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 243 deletions.
diff --git a/kie-dmn/kie-dmn-feel/src/test/java/org/kie/dmn/feel/lang/ast/RangeTypeNodeTest.java b/kie-dmn/kie-dmn-feel/src/test/java/org/kie/dmn/feel/lang/ast/RangeTypeNodeTest.java
@@ -23,8 +23,8 @@
 import org.junit.jupiter.api.Test;
 import org.kie.dmn.feel.lang.types.GenRangeType;
 import org.kie.dmn.feel.util.EvaluationContextTestUtil;
-import org.kie.dmn.feel.util.MockVisitor;
 import org.kie.dmn.feel.lang.types.BuiltInType;
+import org.mockito.Mockito;
 
 class RangeTypeNodeTest {
 
@@ -37,9 +37,10 @@ void evaluate() {
 
     @Test
     void accept() {
-        final MockVisitor visitor = new MockVisitor();
+        final Visitor visitor = Mockito.spy(Visitor.class);
         final RangeTypeNode rangeTypeNode = new RangeTypeNode(null, "sometext");
-        Assertions.assertThat(rangeTypeNode.accept(visitor)).isSameAs(rangeTypeNode);
+        rangeTypeNode.accept(visitor);
+        Mockito.verify(visitor).visit(rangeTypeNode);
     }
 
     @Test

diff --git a/kie-dmn/kie-dmn-feel/src/test/java/org/kie/dmn/feel/util/MockVisitor.java b/kie-dmn/kie-dmn-feel/src/test/java/org/kie/dmn/feel/util/MockVisitor.java