클라이언트 유량제어 eapim-online 이동

src/test/java/com/eactive/eai/custom/inbound/processor/ClientDualInflowControlManagerMetricsTest.java

@@ -0,0 +1,196 @@
+package com.eactive.eai.custom.inbound.processor;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertNotNull;
+import static org.junit.jupiter.api.Assertions.assertNull;
+import static org.junit.jupiter.api.Assertions.assertTrue;
+
+import java.time.Duration;
+import java.util.Map;
+import java.util.concurrent.ConcurrentHashMap;
+
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import org.springframework.test.util.ReflectionTestUtils;
+
+import com.eactive.eai.common.inflow.InflowTargetVO;
+import com.eactive.eai.common.inflow.dual.DualCustomBucket;
+import com.eactive.eai.common.inflow.dual.DualInflowControlManager;
+import com.eactive.eai.custom.inflow.ClientDualInflowControlManager;
+
+import io.github.bucket4j.Bandwidth;
+import io.github.bucket4j.Bucket4j;
+import io.github.bucket4j.local.LocalBucket;
+
+/**
+ * ClientDualInflowControlManager 클라이언트 메트릭 단위 테스트.
+ */
+class ClientDualInflowControlManagerMetricsTest {
+
+    private ClientDualInflowControlManager manager;
+
+    @BeforeEach
+    void setUp() {
+        manager = new ClientDualInflowControlManager();
+    }
+
+    // -------------------------------------------------------------------------
+    // Helpers
+    // -------------------------------------------------------------------------
+
+    private InflowTargetVO makeTargetVO(String name) {
+        InflowTargetVO vo = new InflowTargetVO();
+        vo.setName(name);
+        vo.setThresholdPerSecond(100);
+        vo.setThreshold(10000);
+        vo.setThresholdTimeUnit("DAY");
+        vo.setActivate(true);
+        return vo;
+    }
+
+    private LocalBucket stableBucket(long capacity) {
+        return Bucket4j.builder()
+                .addLimit(Bandwidth.simple(capacity, Duration.ofHours(1)))
+                .build();
+    }
+
+    private LocalBucket exhaustedBucket(long capacity) {
+        LocalBucket b = stableBucket(capacity);
+        b.tryConsume(capacity);
+        return b;
+    }
+
+    private DualCustomBucket passDualBucket(String name) {
+        return new DualCustomBucket(stableBucket(100), null, makeTargetVO(name));
+    }
+
+    private DualCustomBucket blockedPerSecondDualBucket(String name) {
+        return new DualCustomBucket(exhaustedBucket(1), null, makeTargetVO(name));
+    }
+
+    private DualCustomBucket blockedThresholdDualBucket(String name) {
+        return new DualCustomBucket(stableBucket(100), exhaustedBucket(1), makeTargetVO(name));
+    }
+
+    private void injectClientMap(Map<String, DualCustomBucket> map) {
+        ReflectionTestUtils.setField(manager, "clientBucketMap", map);
+    }
+
+    // =========================================================================
+    // isClientPassDetail() — 카운터 증가
+    // =========================================================================
+
+    @Test
+    void isClientPassDetail_bucketNotRegistered_noMetricsCreated() {
+        manager.isClientPassDetail("UNKNOWN");
+
+        assertNull(manager.getClientMetrics("UNKNOWN"));
+    }
+
+    @Test
+    void isClientPassDetail_pass_incrementsAllowed() {
+        Map<String, DualCustomBucket> map = new ConcurrentHashMap<>();
+        map.put("C1", passDualBucket("C1"));
+        injectClientMap(map);
+
+        manager.isClientPassDetail("C1");
+
+        assertEquals(1, manager.getClientMetrics("C1").allowed.get());
+    }
+
+    @Test
+    void isClientPassDetail_blockedThreshold_incrementsRejectedThreshold() {
+        Map<String, DualCustomBucket> map = new ConcurrentHashMap<>();
+        map.put("C1", blockedThresholdDualBucket("C1"));
+        injectClientMap(map);
+
+        manager.isClientPassDetail("C1");
+
+        assertEquals(1, manager.getClientMetrics("C1").rejectedThreshold.get());
+    }
+
+    @Test
+    void isClientPassDetail_blockedPerSecond_incrementsRejectedPerSecond() {
+        Map<String, DualCustomBucket> map = new ConcurrentHashMap<>();
+        map.put("C1", blockedPerSecondDualBucket("C1"));
+        injectClientMap(map);
+
+        manager.isClientPassDetail("C1");
+
+        assertEquals(1, manager.getClientMetrics("C1").rejectedPerSecond.get());
+    }
+
+    // =========================================================================
+    // getAllClientMetrics — 조회 및 불변성
+    // =========================================================================
+
+    @Test
+    void getAllClientMetrics_empty_returnsEmptyMap() {
+        assertTrue(manager.getAllClientMetrics().isEmpty());
+    }
+
+    @Test
+    void getAllClientMetrics_afterCalls_containsEntries() {
+        Map<String, DualCustomBucket> map = new ConcurrentHashMap<>();
+        map.put("C1", passDualBucket("C1"));
+        map.put("C2", passDualBucket("C2"));
+        injectClientMap(map);
+
+        manager.isClientPassDetail("C1");
+        manager.isClientPassDetail("C2");
+
+        Map<String, DualInflowControlManager.TargetMetrics> all = manager.getAllClientMetrics();
+        assertEquals(2, all.size());
+        assertTrue(all.containsKey("C1"));
+        assertTrue(all.containsKey("C2"));
+    }
+
+    // =========================================================================
+    // resetClientMetrics / resetAllClientMetrics
+    // =========================================================================
+
+    @Test
+    void resetClientMetrics_resetsTargetClient() {
+        Map<String, DualCustomBucket> map = new ConcurrentHashMap<>();
+        map.put("C1", passDualBucket("C1"));
+        injectClientMap(map);
+
+        manager.isClientPassDetail("C1");
+        manager.resetClientMetrics("C1");
+
+        assertEquals(0, manager.getClientMetrics("C1").allowed.get());
+    }
+
+    @Test
+    void resetAllClientMetrics_resetsAll() {
+        Map<String, DualCustomBucket> map = new ConcurrentHashMap<>();
+        map.put("C1", passDualBucket("C1"));
+        map.put("C2", passDualBucket("C2"));
+        injectClientMap(map);
+
+        manager.isClientPassDetail("C1");
+        manager.isClientPassDetail("C2");
+        manager.resetAllClientMetrics();
+
+        assertEquals(0, manager.getClientMetrics("C1").allowed.get());
+        assertEquals(0, manager.getClientMetrics("C2").allowed.get());
+    }
+
+    // =========================================================================
+    // removeClient — 메트릭 자동 삭제
+    // =========================================================================
+
+    @Test
+    void removeClient_cleansUpMetrics() {
+        Map<String, DualCustomBucket> map = new ConcurrentHashMap<>();
+        map.put("C1", passDualBucket("C1"));
+        injectClientMap(map);
+
+        manager.isClientPassDetail("C1");
+        assertNotNull(manager.getClientMetrics("C1"));
+
+        manager.removeClient("C1");
+
+        assertNull(manager.getClientMetrics("C1"));
+    }
+}

src/test/java/com/eactive/eai/custom/inbound/processor/DJBRequestProcessorTest.java

@@ -1,7 +1,12 @@
 package com.eactive.eai.custom.inbound.processor;
 
-import static org.junit.jupiter.api.Assertions.*;
-import static org.mockito.Mockito.*;
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertNull;
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.never;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
 
 import org.junit.jupiter.api.BeforeAll;
 import org.junit.jupiter.api.Test;
@@ -10,11 +15,11 @@ import org.springframework.test.util.ReflectionTestUtils;
 
 import com.eactive.eai.common.inflow.Bucket;
 import com.eactive.eai.common.inflow.InflowTargetVO;
-import com.eactive.eai.common.inflow.dual.ClientDualBucket;
 import com.eactive.eai.common.inflow.dual.DualBucket;
 import com.eactive.eai.common.inflow.dual.DualCustomBucket;
 import com.eactive.eai.common.server.EAIServerManager;
 import com.eactive.eai.common.util.ApplicationContextProvider;
+import com.eactive.eai.custom.inflow.ClientDualBucket;
 
 /**
  * DJBRequestProcessor 단위 테스트.

src/test/java/com/eactive/eai/custom/inflow/ClientInflowTargetBucketServiceTest.java

@@ -1,7 +1,11 @@
 package com.eactive.eai.custom.inflow;
 
-import static org.junit.jupiter.api.Assertions.*;
-import static org.mockito.Mockito.*;
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertNotNull;
+import static org.junit.jupiter.api.Assertions.assertNull;
+import static org.junit.jupiter.api.Assertions.assertTrue;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.when;
 
 import java.time.Duration;
 import java.util.List;
@@ -11,7 +15,6 @@ import java.util.concurrent.ConcurrentHashMap;
 import org.junit.jupiter.api.Test;
 
 import com.eactive.eai.common.inflow.InflowTargetVO;
-import com.eactive.eai.common.inflow.dual.ClientDualInflowControlManager;
 import com.eactive.eai.common.inflow.dual.DualCustomBucket;
 import com.eactive.eai.manage.inflow.GroupBucketStatusDTO;
 import com.eactive.eai.manage.inflow.TargetBucketStatusDTO;

src/test/java/com/eactive/eai/custom/inflow/ClientInflowTargetMetricServiceTest.java

@@ -1,14 +1,22 @@
 package com.eactive.eai.custom.inflow;
 
-import static org.junit.jupiter.api.Assertions.*;
-import static org.mockito.Mockito.*;
+import static org.junit.jupiter.api.Assertions.assertDoesNotThrow;
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertFalse;
+import static org.junit.jupiter.api.Assertions.assertNotNull;
+import static org.junit.jupiter.api.Assertions.assertNull;
+import static org.junit.jupiter.api.Assertions.assertTrue;
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.never;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
 
 import java.util.List;
 
 import org.junit.jupiter.api.Test;
 
 import com.eactive.eai.common.inflow.InflowTargetVO;
-import com.eactive.eai.common.inflow.dual.ClientDualInflowControlManager;
 import com.eactive.eai.common.inflow.dual.DualInflowControlManager;
 import com.eactive.eai.manage.inflow.TargetMetricDTO;