spring-ydb-retry/src/test/java/tech/ydb/retry/YdbRetryPolicyConfigTest.java - main - githubmirror/ydb-java-dialects

spring-ydb-retry/src/test/java/tech/ydb/retry/YdbRetryPolicyConfigTest.java
270 строк11 KB
Kirill Kurdyukov
Add spring-ydb-retry module (v0.10.0) (#226)
15 июн 2026, 14:09
15 июн 2026, 14:093e8a6c8
package tech.ydb.retry;

import java.lang.reflect.Method;
import org.junit.jupiter.api.Test;
import org.springframework.core.annotation.AnnotatedElementUtils;

import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertFalse;
import static org.junit.jupiter.api.Assertions.assertSame;
import static org.junit.jupiter.api.Assertions.assertThrows;
import static org.junit.jupiter.api.Assertions.assertTrue;
import static tech.ydb.retry.YdbRetryPolicyConfig.DEFAULT_FAST_BACKOFF_BASE_MS;
import static tech.ydb.retry.YdbRetryPolicyConfig.DEFAULT_FAST_CAP_BACKOFF_MS;
import static tech.ydb.retry.YdbRetryPolicyConfig.DEFAULT_MAX_ATTEMPTS;
import static tech.ydb.retry.YdbRetryPolicyConfig.DEFAULT_SLOW_BACKOFF_BASE_MS;
import static tech.ydb.retry.YdbRetryPolicyConfig.DEFAULT_SLOW_CAP_BACKOFF_MS;

class YdbRetryPolicyConfigTest extends InterceptorTestSupport {

    @Test
    void defaultConstructorShouldSetDefaultValues() {
        YdbRetryPolicyConfig config = new YdbRetryPolicyConfig();

        assertEquals(DEFAULT_MAX_ATTEMPTS, config.getMaxAttempts());
        assertEquals(DEFAULT_SLOW_BACKOFF_BASE_MS, config.getSlowBackoffBaseMs());
        assertEquals(DEFAULT_FAST_BACKOFF_BASE_MS, config.getFastBackoffBaseMs());
        assertEquals(DEFAULT_SLOW_CAP_BACKOFF_MS, config.getSlowCapBackoffMs());
        assertEquals(DEFAULT_FAST_CAP_BACKOFF_MS, config.getFastCapBackoffMs());
    }

    @Test
    void customConstructorShouldSetValues() {
        YdbRetryPolicyConfig config = new YdbRetryPolicyConfig(true, 5, 100, 20, 2000, 300);

        assertEquals(5, config.getMaxAttempts());
        assertEquals(100, config.getSlowBackoffBaseMs());
        assertEquals(20, config.getFastBackoffBaseMs());
        assertEquals(2000, config.getSlowCapBackoffMs());
        assertEquals(300, config.getFastCapBackoffMs());
    }

    @Test
    void shouldAcceptZeroMaxAttempts() {
        YdbRetryPolicyConfig config = new YdbRetryPolicyConfig(true, 0, 0, 0, 0, 0);
        assertEquals(0, config.getMaxAttempts());
    }

    @Test
    void shouldThrowWhenMaxAttemptsIsNegative() {
        assertThrows(
                IllegalArgumentException.class, () -> new YdbRetryPolicyConfig(true, -1, 0, 0, 0, 0));
    }

    @Test
    void shouldThrowWhenSlowBackoffBaseIsNegative() {
        assertThrows(
                IllegalArgumentException.class, () -> new YdbRetryPolicyConfig(true, 1, -1, 0, 0, 0));
    }

    @Test
    void shouldThrowWhenFastBackoffBaseIsNegative() {
        assertThrows(
                IllegalArgumentException.class, () -> new YdbRetryPolicyConfig(true, 1, 0, -1, 0, 0));
    }

    @Test
    void shouldThrowWhenSlowCapIsNegative() {
        assertThrows(
                IllegalArgumentException.class, () -> new YdbRetryPolicyConfig(true, 1, 0, 0, -1, 0));
    }

    @Test
    void shouldThrowWhenFastCapIsNegative() {
        assertThrows(
                IllegalArgumentException.class, () -> new YdbRetryPolicyConfig(true, 1, 0, 0, 0, -1));
    }

    @Test
    void mergeWithNullShouldReturnSameInstance() {
        YdbRetryPolicyConfig config = new YdbRetryPolicyConfig();
        assertSame(config, config.merge(null));
    }

    @Test
    void mergeWithDefaultAnnotationShouldKeepConfigValues() throws NoSuchMethodException {
        YdbRetryPolicyConfig original = new YdbRetryPolicyConfig(true, 5, 100, 20, 2000, 300);

        Method method = YdbTransactionalTestService.class.getMethod("defaultRetry");
        YdbTransactional annotation =
                AnnotatedElementUtils.findMergedAnnotation(method, YdbTransactional.class);

        YdbRetryPolicyConfig merged = original.merge(annotation);

        assertEquals(5, merged.getMaxAttempts());
        assertEquals(100, merged.getSlowBackoffBaseMs());
        assertEquals(20, merged.getFastBackoffBaseMs());
        assertEquals(2000, merged.getSlowCapBackoffMs());
        assertEquals(300, merged.getFastCapBackoffMs());
    }

    @Test
    void mergeWithCustomAnnotationShouldOverride() throws NoSuchMethodException {
        YdbRetryPolicyConfig original = new YdbRetryPolicyConfig(true, 5, 100, 20, 2000, 300);

        Method method = YdbTransactionalTestService.class.getMethod("ydbNewTransactionSettings");
        YdbTransactional annotation =
                AnnotatedElementUtils.findMergedAnnotation(method, YdbTransactional.class);

        YdbRetryPolicyConfig merged = original.merge(annotation);

        assertEquals(100, merged.getMaxAttempts());
        assertEquals(200, merged.getSlowBackoffBaseMs());
        assertEquals(10, merged.getFastBackoffBaseMs());
        assertEquals(10000, merged.getSlowCapBackoffMs());
        assertEquals(12, merged.getFastCapBackoffMs());
    }

    @Test
    void mergeWithPartialOverrideShouldOnlyChangeSpecifiedValues() throws NoSuchMethodException {
        YdbRetryPolicyConfig original = new YdbRetryPolicyConfig(true, 5, 100, 20, 2000, 300);

        Method method = YdbTransactionalTestService.class.getMethod("ydbCustomRetry");
        YdbTransactional annotation =
                AnnotatedElementUtils.findMergedAnnotation(method, YdbTransactional.class);

        YdbRetryPolicyConfig merged = original.merge(annotation);

        // only maxAttempts should change
        assertEquals(3, merged.getMaxAttempts());
        assertEquals(100, merged.getSlowBackoffBaseMs());
        assertEquals(20, merged.getFastBackoffBaseMs());
        assertEquals(2000, merged.getSlowCapBackoffMs());
        assertEquals(300, merged.getFastCapBackoffMs());
    }

    @Test
    void shouldThrowWhenYdbTransactionalMaxAttemptsIsNegative() throws NoSuchMethodException {
        YdbRetryPolicyConfig original = new YdbRetryPolicyConfig(true, 5, 100, 20, 2000, 300);

        Method method = YdbTransactionalTestService.class.getMethod("ydbNegativeMaxAttempts");
        YdbTransactional annotation =
                AnnotatedElementUtils.findMergedAnnotation(method, YdbTransactional.class);

        assertThrows(IllegalArgumentException.class, () -> original.merge(annotation));
    }

    @Test
    void shouldInheritGlobalMaxAttemptsWhenAnnotationIsZero() throws NoSuchMethodException {
        YdbRetryPolicyConfig original = new YdbRetryPolicyConfig(true, 5, 100, 20, 2000, 300);

        Method method = YdbTransactionalTestService.class.getMethod("ydbZeroMaxAttempts");
        YdbTransactional annotation =
                AnnotatedElementUtils.findMergedAnnotation(method, YdbTransactional.class);

        YdbRetryPolicyConfig merged = original.merge(annotation);

        assertEquals(5, merged.getMaxAttempts());
    }

    @Test
    void ceilingShouldBeComputedFromCapValues() {
        YdbRetryPolicyConfig config = new YdbRetryPolicyConfig(true, 5, 100, 20, 2000, 300);

        int expectedSlowCeiling = (int) Math.ceil(Math.log(2001) / Math.log(2));
        int expectedFastCeiling = (int) Math.ceil(Math.log(301) / Math.log(2));

        assertEquals(expectedSlowCeiling, config.getSlowCeiling());
        assertEquals(expectedFastCeiling, config.getFastCeiling());
    }

    @Test
    void ceilingForSmallCapShouldBeOne() {
        YdbRetryPolicyConfig config = new YdbRetryPolicyConfig(true, 1, 0, 0, 1, 1);
        assertEquals(1, config.getSlowCeiling());
        assertEquals(1, config.getFastCeiling());
    }

    @Test
    void ceilingForCapTwoShouldBeTwo() {
        YdbRetryPolicyConfig config = new YdbRetryPolicyConfig(true, 1, 0, 0, 2, 2);
        assertEquals(2, config.getSlowCeiling());
        assertEquals(2, config.getFastCeiling());
    }

    @Test
    void ceilingForZeroCapShouldBeZero() {
        YdbRetryPolicyConfig config = new YdbRetryPolicyConfig(true, 1, 0, 0, 0, 0);
        assertEquals(0, config.getSlowCeiling());
        assertEquals(0, config.getFastCeiling());
    }

    @Test
    void defaultConstructorShouldSetEnabledTrue() {
        YdbRetryPolicyConfig config = new YdbRetryPolicyConfig();
        assertTrue(config.isEnabled());
    }

    @Test
    void constructorShouldSetEnabledFalse() {
        YdbRetryPolicyConfig config = new YdbRetryPolicyConfig(false, 5, 100, 20, 2000, 300);
        assertFalse(config.isEnabled());
    }

    @Test
    void mergeShouldPreserveEnabledFromBaseConfig() throws NoSuchMethodException {
        YdbRetryPolicyConfig original = new YdbRetryPolicyConfig(false, 5, 100, 20, 2000, 300);

        Method method = YdbTransactionalTestService.class.getMethod("ydbCustomRetry");
        YdbTransactional annotation =
                AnnotatedElementUtils.findMergedAnnotation(method, YdbTransactional.class);

        YdbRetryPolicyConfig merged = original.merge(annotation);

        assertFalse(merged.isEnabled());
    }

    @Test
    void mergeShouldKeepEnabledTrueWhenConfigEnabled() throws NoSuchMethodException {
        YdbRetryPolicyConfig original = new YdbRetryPolicyConfig(true, 5, 100, 20, 2000, 300);

        Method method = YdbTransactionalTestService.class.getMethod("defaultRetry");
        YdbTransactional annotation =
                AnnotatedElementUtils.findMergedAnnotation(method, YdbTransactional.class);

        YdbRetryPolicyConfig merged = original.merge(annotation);

        assertTrue(merged.isEnabled());
    }

    @Test
    void mergeWithDefaultAnnotationShouldKeepEnabledFalseWhenConfigDisabled()
            throws NoSuchMethodException {
        YdbRetryPolicyConfig original = new YdbRetryPolicyConfig(false, 5, 100, 20, 2000, 300);

        Method method = YdbTransactionalTestService.class.getMethod("defaultRetry");
        YdbTransactional annotation =
                AnnotatedElementUtils.findMergedAnnotation(method, YdbTransactional.class);

        YdbRetryPolicyConfig merged = original.merge(annotation);

        assertFalse(merged.isEnabled());
    }

    @Test
    void mergeWithDisabledAnnotationShouldSetEnabledFalse() throws NoSuchMethodException {
        YdbRetryPolicyConfig original = new YdbRetryPolicyConfig(true, 5, 100, 20, 2000, 300);

        Method method = YdbTransactionalTestService.class.getMethod("ydbRetryDisabled");
        YdbTransactional annotation =
                AnnotatedElementUtils.findMergedAnnotation(method, YdbTransactional.class);

        YdbRetryPolicyConfig merged = original.merge(annotation);

        assertFalse(merged.isEnabled());
    }

    @Test
    void mergeWithEnabledAnnotationShouldNotOverrideDisabledGlobalConfig()
            throws NoSuchMethodException {
        YdbRetryPolicyConfig original = new YdbRetryPolicyConfig(false, 5, 100, 20, 2000, 300);

        Method method = YdbTransactionalTestService.class.getMethod("ydbRetryEnabled");
        YdbTransactional annotation =
                AnnotatedElementUtils.findMergedAnnotation(method, YdbTransactional.class);

        YdbRetryPolicyConfig merged = original.merge(annotation);

        assertFalse(merged.isEnabled());
    }
}