FINERACT-2081: Support read-only transactions via Batch API

2024-12-17 14:49:18 +01:00
parent 8716751c88
commit d03973165d
5 changed files with 125 additions and 2 deletions
@@ -0,0 +1,118 @@
+/**
+ * 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.batch.service;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertTrue;
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.Mockito.times;
+import static org.mockito.Mockito.when;
+
+import jakarta.persistence.EntityManager;
+import jakarta.ws.rs.core.UriInfo;
+import java.util.List;
+import org.apache.fineract.batch.command.CommandStrategy;
+import org.apache.fineract.batch.command.CommandStrategyProvider;
+import org.apache.fineract.batch.domain.BatchRequest;
+import org.apache.fineract.batch.domain.BatchResponse;
+import org.apache.fineract.infrastructure.core.filters.BatchRequestPreprocessor;
+import org.apache.fineract.infrastructure.core.persistence.ExtendedJpaTransactionManager;
+import org.apache.fineract.infrastructure.core.serialization.FromJsonHelper;
+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.Mockito;
+import org.mockito.junit.jupiter.MockitoExtension;
+import org.springframework.transaction.support.DefaultTransactionStatus;
+
+@ExtendWith(MockitoExtension.class)
+class BatchApiServiceImplTest {
+
+    @Mock
+    private CommandStrategyProvider strategyProvider;
+    @Mock
+    private ExtendedJpaTransactionManager transactionManager;
+    @Mock
+    private EntityManager entityManager;
+    @Mock
+    private CommandStrategy commandStrategy;
+    @Mock
+    private UriInfo uriInfo;
+    private final ResolutionHelper resolutionHelper = Mockito.spy(new ResolutionHelper(new FromJsonHelper()));
+    private final List<BatchRequestPreprocessor> batchPreprocessors = Mockito.spy(List.of());
+    @InjectMocks
+    private BatchApiServiceImpl batchApiService;
+    private BatchRequest request;
+    private BatchResponse response;
+
+    @BeforeEach
+    void setUp() {
+        request = new BatchRequest();
+        request.setRequestId(1L);
+        request.setMethod("POST");
+        request.setRelativeUrl("/random_api");
+        response = new BatchResponse();
+        response.setRequestId(1L);
+        response.setStatusCode(200);
+        response.setBody("Success");
+    }
+
+    @AfterEach
+    void tearDown() {
+        Mockito.reset(resolutionHelper);
+        Mockito.reset(batchPreprocessors);
+        Mockito.reset(entityManager);
+        Mockito.reset(commandStrategy);
+        Mockito.reset(strategyProvider);
+        Mockito.reset(transactionManager);
+    }
+
+    @Test
+    void testHandleBatchRequestsWithEnclosingTransaction() {
+        List<BatchRequest> requestList = List.of(request);
+        when(strategyProvider.getCommandStrategy(any())).thenReturn(commandStrategy);
+        when(commandStrategy.execute(any(), any())).thenReturn(response);
+        // Regular transaction
+        when(transactionManager.getTransaction(any()))
+                .thenReturn(new DefaultTransactionStatus("txn_name", null, true, true, false, false, false, null));
+        List<BatchResponse> result = batchApiService.handleBatchRequestsWithEnclosingTransaction(requestList, uriInfo);
+        assertEquals(1, result.size());
+        assertEquals(200, result.get(0).getStatusCode());
+        assertTrue(result.get(0).getBody().contains("Success"));
+        Mockito.verify(entityManager, times(1)).flush();
+    }
+
+    @Test
+    void testHandleBatchRequestsWithEnclosingTransactionReadOnly() {
+        List<BatchRequest> requestList = List.of(request);
+        when(strategyProvider.getCommandStrategy(any())).thenReturn(commandStrategy);
+        when(commandStrategy.execute(any(), any())).thenReturn(response);
+        // Read-only transaction
+        when(transactionManager.getTransaction(any()))
+                .thenReturn(new DefaultTransactionStatus("txn_name", null, true, true, false, true, false, null));
+        List<BatchResponse> result = batchApiService.handleBatchRequestsWithEnclosingTransaction(requestList, uriInfo);
+        assertEquals(1, result.size());
+        assertEquals(200, result.get(0).getStatusCode());
+        assertTrue(result.get(0).getBody().contains("Success"));
+        Mockito.verifyNoInteractions(entityManager);
+    }
+}