FINERACT-1678-Config-Bulk-Event-Loan-COB

fineract-provider/src/test/java/org/apache/fineract/cob/COBBusinessStepServiceStepDefinitions.java

@@ -35,6 +35,7 @@ import org.apache.fineract.cob.domain.BatchBusinessStep;
 import org.apache.fineract.cob.domain.BatchBusinessStepRepository;
 import org.apache.fineract.cob.exceptions.BusinessStepException;
 import org.apache.fineract.cob.loan.LoanCOBBusinessStep;
+import org.apache.fineract.infrastructure.configuration.domain.ConfigurationDomainService;
 import org.apache.fineract.infrastructure.core.domain.AbstractAuditableCustom;
 import org.apache.fineract.infrastructure.core.domain.ActionContext;
 import org.apache.fineract.infrastructure.core.service.ThreadLocalContextUtil;
@@ -51,8 +52,9 @@ public class COBBusinessStepServiceStepDefinitions implements En {
     private ListableBeanFactory beanFactory = mock(ListableBeanFactory.class);
     private BatchBusinessStepRepository batchBusinessStepRepository = mock(BatchBusinessStepRepository.class);
     private BusinessEventNotifierService businessEventNotifierService = mock(BusinessEventNotifierService.class);
+    private ConfigurationDomainService configurationDomainService = mock(ConfigurationDomainService.class);
     private final COBBusinessStepService businessStepService = new COBBusinessStepServiceImpl(batchBusinessStepRepository,
-            applicationContext, beanFactory, businessEventNotifierService);
+            applicationContext, beanFactory, businessEventNotifierService, configurationDomainService);
     private COBBusinessStep cobBusinessStep = mock(COBBusinessStep.class);
     private COBBusinessStep notRegistereCobBusinessStep = mock(COBBusinessStep.class);
     private TreeMap<Long, String> executionMap;
@@ -88,6 +90,7 @@ public class COBBusinessStepServiceStepDefinitions implements En {
             lenient().when(this.applicationContext.getBean("test")).thenReturn(cobBusinessStep);
             lenient().when(this.applicationContext.getBean("notExist")).thenThrow(BeanCreationException.class);
             lenient().when(this.cobBusinessStep.execute(this.item)).thenReturn(outputItem);
+            lenient().when(this.configurationDomainService.isCOBBulkEventEnabled()).thenReturn(true);
 
             ThreadLocalContextUtil.setActionContext(ActionContext.DEFAULT);
         });

fineract-provider/src/test/java/org/apache/fineract/cob/service/COBBulkEventConfigurationTest.java

@@ -0,0 +1,159 @@
+/**
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements. See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership. The ASF licenses this file
+ * to you 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 org.apache.fineract.cob.service;
+
+import static org.junit.jupiter.api.Assertions.assertThrows;
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.ArgumentMatchers.anyString;
+import static org.mockito.Mockito.doThrow;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.times;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
+
+import java.time.LocalDate;
+import java.time.ZoneId;
+import java.util.HashMap;
+import java.util.Map;
+import java.util.TreeMap;
+import org.apache.fineract.cob.COBBusinessStep;
+import org.apache.fineract.cob.COBBusinessStepServiceImpl;
+import org.apache.fineract.cob.domain.BatchBusinessStepRepository;
+import org.apache.fineract.cob.exceptions.BusinessStepException;
+import org.apache.fineract.infrastructure.businessdate.domain.BusinessDateType;
+import org.apache.fineract.infrastructure.configuration.domain.ConfigurationDomainService;
+import org.apache.fineract.infrastructure.core.domain.ActionContext;
+import org.apache.fineract.infrastructure.core.domain.FineractPlatformTenant;
+import org.apache.fineract.infrastructure.core.service.ThreadLocalContextUtil;
+import org.apache.fineract.infrastructure.event.business.service.BusinessEventNotifierService;
+import org.apache.fineract.portfolio.loanaccount.domain.Loan;
+import org.junit.jupiter.api.AfterEach;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.extension.ExtendWith;
+import org.mockito.InjectMocks;
+import org.mockito.Mock;
+import org.mockito.junit.jupiter.MockitoExtension;
+import org.springframework.beans.factory.ListableBeanFactory;
+import org.springframework.context.ApplicationContext;
+
+@ExtendWith(MockitoExtension.class)
+public class COBBulkEventConfigurationTest {
+
+    @Mock
+    private BatchBusinessStepRepository batchBusinessStepRepository;
+    @Mock
+    private ApplicationContext applicationContext;
+    @Mock
+    private ListableBeanFactory beanFactory;
+    @Mock
+    private BusinessEventNotifierService businessEventNotifierService;
+    @Mock
+    private ConfigurationDomainService configurationDomainService;
+    @InjectMocks
+    private COBBusinessStepServiceImpl underTest;
+
+    @BeforeEach
+    public void setUp() {
+        ThreadLocalContextUtil.setTenant(new FineractPlatformTenant(1L, "default", "Default", "Asia/Kolkata", null));
+        ThreadLocalContextUtil
+                .setBusinessDates(new HashMap<>(Map.of(BusinessDateType.BUSINESS_DATE, LocalDate.now(ZoneId.systemDefault()))));
+    }
+
+    @AfterEach
+    public void tearDown() {
+        ThreadLocalContextUtil.setActionContext(ActionContext.DEFAULT);
+    }
+
+    @Test
+    public void testGivenBulkEventEnabledWhenCOBRunThenEventsAreRecorded() {
+        // given
+        Loan loan = mock(Loan.class);
+        TreeMap<Long, String> dummyExecutionMap = new TreeMap<>();
+        dummyExecutionMap.put(1L, "dummyBusinessStep");
+        COBBusinessStep businessStep = mock(COBBusinessStep.class);
+        when(configurationDomainService.isCOBBulkEventEnabled()).thenReturn(true);
+        when(applicationContext.getBean(anyString())).thenReturn(businessStep);
+        when(businessStep.execute(any())).thenReturn(loan);
+
+        // when
+        underTest.run(dummyExecutionMap, loan);
+
+        // then
+        verify(businessEventNotifierService, times(1)).startExternalEventRecording();
+        verify(businessEventNotifierService, times(1)).stopExternalEventRecording();
+    }
+
+    @Test
+    public void testGivenBulkEventDisabledWhenCOBRunThenEventsAreNotRecorded() {
+        // given
+        Loan loan = mock(Loan.class);
+        TreeMap<Long, String> dummyExecutionMap = new TreeMap<>();
+        dummyExecutionMap.put(1L, "dummyBusinessStep");
+        COBBusinessStep businessStep = mock(COBBusinessStep.class);
+        when(configurationDomainService.isCOBBulkEventEnabled()).thenReturn(false);
+        when(applicationContext.getBean(anyString())).thenReturn(businessStep);
+        when(businessStep.execute(any())).thenReturn(loan);
+
+        // when
+        underTest.run(dummyExecutionMap, loan);
+
+        // then
+        verify(businessEventNotifierService, times(0)).startExternalEventRecording();
+        verify(businessEventNotifierService, times(0)).stopExternalEventRecording();
+    }
+
+    @Test
+    public void testGivenBulkEventEnabledWhenCOBRunExceptionThenEventRecordingReset() {
+        // given
+        Loan loan = mock(Loan.class);
+        TreeMap<Long, String> dummyExecutionMap = new TreeMap<>();
+        dummyExecutionMap.put(1L, "dummyBusinessStep");
+        COBBusinessStep businessStep = mock(COBBusinessStep.class);
+        when(configurationDomainService.isCOBBulkEventEnabled()).thenReturn(true);
+        when(applicationContext.getBean(anyString())).thenReturn(businessStep);
+
+        doThrow(new BusinessStepException("Test exception")).when(businessStep).execute(any());
+
+        // when
+        assertThrows(BusinessStepException.class, () -> underTest.run(dummyExecutionMap, loan));
+
+        // then
+        verify(businessEventNotifierService, times(1)).resetEventRecording();
+    }
+
+    @Test
+    public void testGivenBulkEventDisabledWhenCOBRunExceptionThenEventRecordingResetNotCalled() {
+        // given
+        Loan loan = mock(Loan.class);
+        TreeMap<Long, String> dummyExecutionMap = new TreeMap<>();
+        dummyExecutionMap.put(1L, "dummyBusinessStep");
+        COBBusinessStep businessStep = mock(COBBusinessStep.class);
+        when(configurationDomainService.isCOBBulkEventEnabled()).thenReturn(false);
+        when(applicationContext.getBean(anyString())).thenReturn(businessStep);
+        doThrow(new BusinessStepException("Test exception")).when(businessStep).execute(any());
+
+        // when
+        assertThrows(BusinessStepException.class, () -> underTest.run(dummyExecutionMap, loan));
+
+        // then
+        verify(businessEventNotifierService, times(0)).resetEventRecording();
+    }
+
+}