Add answer in annotation

Signed-off-by: Jorge Bescos Gascon <[email protected]>

Author: jbescos

microprofile/testing/mock-beans/src/main/java/io/helidon/microprofile/testing/mockbeans/MockBean.java

@@ -21,6 +21,8 @@
 import java.lang.annotation.RetentionPolicy;
 import java.lang.annotation.Target;
 
+import org.mockito.Answers;
+
 /**
  * A field annotated with @MockBean will be mocked by Mockito
  * and injected in every place it is referenced.
@@ -29,4 +31,10 @@
 @Target({ElementType.FIELD, ElementType.PARAMETER})
 public @interface MockBean {
 
+    /**
+     * The {@link Answers} type to use on the mock.
+     * Defaults to {@link Answers#RETURNS_DEFAULTS}
+     * @return the answer type
+     */
+    Answers answer() default Answers.RETURNS_DEFAULTS;
 }

microprofile/testing/mock-beans/src/main/java/io/helidon/microprofile/testing/mockbeans/MockBeansCdiExtension.java

@@ -16,8 +16,9 @@
 package io.helidon.microprofile.testing.mockbeans;
 
 import java.lang.reflect.Field;
-import java.util.HashSet;
+import java.util.HashMap;
 import java.util.List;
+import java.util.Map;
 import java.util.Set;
 
 import jakarta.enterprise.context.ApplicationScoped;
@@ -38,7 +39,7 @@
  */
 public class MockBeansCdiExtension implements Extension {
 
-    private final Set<Class<?>> mocks = new HashSet<>();
+    private final Map<Class<?>, MockBean> mocks = new HashMap<>();
 
     void processMockBean(@Observes @WithAnnotations(MockBean.class) ProcessAnnotatedType<?> obj) throws Exception {
         var configurator = obj.configureAnnotatedType();
@@ -49,7 +50,7 @@ void processMockBean(@Observes @WithAnnotations(MockBean.class) ProcessAnnotated
                 // Adds @Inject to be more user friendly
                 field.add(InjectLiteral.INSTANCE);
                 Class<?> fieldType = f.getType();
-                mocks.add(fieldType);
+                mocks.put(fieldType, mockBean);
             }
         });
         configurator.constructors().forEach(constructor -> {
@@ -62,16 +63,16 @@ private void processMockBeanParameters(List<? extends AnnotatedParameter<?>> par
             MockBean mockBean = parameter.getAnnotation(MockBean.class);
             if (mockBean != null) {
                 Class<?> parameterType = parameter.getJavaParameter().getType();
-                mocks.add(parameterType);
+                mocks.put(parameterType, mockBean);
             }
         });
     }
 
     void registerOtherBeans(@Observes AfterBeanDiscovery event, BeanManager beanManager) {
         // Register all mocks
-        mocks.forEach(type -> {
+        mocks.entrySet().forEach(entry -> {
             event.addBean()
-                .addType(type)
+                .addType(entry.getKey())
                 .scope(ApplicationScoped.class)
                 .alternative(true)
                 .createWith(inst -> {
@@ -80,9 +81,9 @@ void registerOtherBeans(@Observes AfterBeanDiscovery event, BeanManager beanMana
                         Bean<?> bean = beans.iterator().next();
                         MockSettings mockSettings = (MockSettings) beanManager.getReference(bean, MockSettings.class,
                                 beanManager.createCreationalContext(null));
-                        return Mockito.mock(type, mockSettings);
+                        return Mockito.mock(entry.getKey(), mockSettings);
                     } else {
-                        return Mockito.mock(type);
+                        return Mockito.mock(entry.getKey(), Mockito.withSettings().defaultAnswer(entry.getValue().answer()));
                     }
                 })
                 .priority(0);

microprofile/tests/testing/junit5/src/test/java/io/helidon/microprofile/tests/testing/junit5/TestMockBeanAnswer.java

@@ -0,0 +1,75 @@
+/*
+ * Copyright (c) 2024 Oracle and/or its affiliates.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package io.helidon.microprofile.tests.testing.junit5;
+
+import static org.hamcrest.CoreMatchers.is;
+import static org.hamcrest.MatcherAssert.assertThat;
+
+import jakarta.enterprise.inject.Produces;
+import jakarta.inject.Inject;
+import jakarta.ws.rs.GET;
+import jakarta.ws.rs.Path;
+import jakarta.ws.rs.client.WebTarget;
+
+import io.helidon.microprofile.testing.junit5.AddBean;
+import io.helidon.microprofile.testing.junit5.HelidonTest;
+import io.helidon.microprofile.testing.mockbeans.MockBean;
+
+import org.junit.jupiter.api.Test;
+import org.mockito.Answers;
+import org.mockito.MockSettings;
+import org.mockito.Mockito;
+
+@HelidonTest
+@AddBean(TestMockBeanAnswer.Resource.class)
+@AddBean(TestMockBeanAnswer.Service.class)
+class TestMockBeanAnswer {
+
+    @MockBean(answer = Answers.CALLS_REAL_METHODS)
+    private Service service;
+    @Inject
+    private WebTarget target;
+
+    @Test
+    void injectionTest() {
+        String response = target.path("/test").request().get(String.class);
+        assertThat(response, is("Not Mocked"));
+        Mockito.when(service.test()).thenReturn("Mocked");
+        response = target.path("/test").request().get(String.class);
+        assertThat(response, is("Mocked"));
+    }
+
+    @Path("/test")
+    public static class Resource {
+
+        @Inject
+        private Service service;
+
+        @GET
+        public String test() {
+            return service.test();
+        }
+    }
+
+    static class Service {
+
+        String test() {
+            return "Not Mocked";
+        }
+
+    }
+}