56 lines
2.6 KiB
Java
56 lines
2.6 KiB
Java
|
package com.common.service;
|
||
|
|
|
||
|
|
import org.junit.jupiter.api.Test;
|
||
|
|
import org.springframework.jdbc.core.JdbcTemplate;
|
||
|
|
import org.springframework.test.util.ReflectionTestUtils;
|
||
|
|
import org.springframework.transaction.annotation.Transactional;
|
||
|
|
|
||
|
|
import java.lang.reflect.Method;
|
||
|
|
import java.time.LocalDateTime;
|
||
|
|
|
||
|
|
import static org.junit.jupiter.api.Assertions.assertNotNull;
|
||
|
|
import static org.junit.jupiter.api.Assertions.assertTrue;
|
||
|
|
import static org.mockito.ArgumentMatchers.any;
|
||
|
|
import static org.mockito.ArgumentMatchers.anyString;
|
||
|
|
import static org.mockito.Mockito.mock;
|
||
|
|
import static org.mockito.Mockito.mockingDetails;
|
||
|
|
import static org.mockito.Mockito.when;
|
||
|
|
|
||
|
|
class AnalysisCheckpointServiceTest {
|
||
|
|
|
||
|
|
@Test
|
||
|
|
void ruleLeaseUsesFractionalDayInsteadOfIntegerDivision() {
|
||
|
|
JdbcTemplate jdbcTemplate = mock(JdbcTemplate.class);
|
||
|
|
when(jdbcTemplate.update(anyString(), any(Object[].class))).thenReturn(1);
|
||
|
|
AnalysisCheckpointService service = new AnalysisCheckpointService(jdbcTemplate);
|
||
|
|
ReflectionTestUtils.setField(service, "leaseSeconds", 600);
|
||
|
|
|
||
|
|
service.claimRuleWindow("rule-1", "1:window-v2",
|
||
|
|
LocalDateTime.of(2026, 8, 10, 17, 35),
|
||
|
|
LocalDateTime.of(2026, 8, 10, 17, 42));
|
||
|
|
|
||
|
|
assertTrue(mockingDetails(jdbcTemplate).getInvocations().stream()
|
||
|
|
.map(invocation -> String.valueOf(invocation.getRawArguments()[0]))
|
||
|
|
.anyMatch(sql -> sql.contains("lease_until=SYSDATE+(?/86400.0)")));
|
||
|
|
}
|
||
|
|
|
||
|
|
@Test
|
||
|
|
void terminalCheckpointUpdatesAreTransactional() throws Exception {
|
||
|
|
assertTransactional("completeRuleWindow", String.class, String.class,
|
||
|
|
LocalDateTime.class, LocalDateTime.class, long.class, long.class);
|
||
|
|
assertTransactional("failRuleWindow", String.class, String.class,
|
||
|
|
LocalDateTime.class, LocalDateTime.class, String.class, String.class);
|
||
|
|
assertTransactional("completeAlgorithmWindow", String.class, String.class,
|
||
|
|
LocalDateTime.class, LocalDateTime.class, long.class, long.class);
|
||
|
|
assertTransactional("failAlgorithmWindow", String.class, String.class,
|
||
|
|
LocalDateTime.class, LocalDateTime.class, String.class, String.class);
|
||
|
|
}
|
||
|
|
|
||
|
|
private void assertTransactional(String methodName, Class<?>... parameterTypes) throws Exception {
|
||
|
|
Method method = AnalysisCheckpointService.class.getMethod(methodName, parameterTypes);
|
||
|
|
Transactional annotation = method.getAnnotation(Transactional.class);
|
||
|
|
assertNotNull(annotation, methodName + " must commit when Hikari auto-commit is disabled");
|
||
|
|
assertTrue(annotation.rollbackFor().length == 0);
|
||
|
|
}
|
||
|
|
}
|