update test case

doleyzi · Jun 6, 2024 · 0d8d9d7 · 0d8d9d7
1 parent 8f33f9c
commit 0d8d9d7
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 15 deletions.
diff --git a/inlong-audit/audit-sdk/src/test/java/org/apache/inlong/audit/send/SenderManagerTest.java b/inlong-audit/audit-sdk/src/test/java/org/apache/inlong/audit/send/SenderManagerTest.java
@@ -18,7 +18,6 @@
 package org.apache.inlong.audit.send;
 
 import org.apache.inlong.audit.util.AuditConfig;
-import org.apache.inlong.audit.util.RequestIdUtils;
 
 import org.junit.Test;
 
@@ -28,19 +27,6 @@ public class SenderManagerTest {
 
     private AuditConfig testConfig = new AuditConfig();
 
-    @Test
-    public void nextRequestId() {
-        SenderManager testManager = new SenderManager(testConfig);
-        Long requestId = RequestIdUtils.nextRequestId();
-        assertTrue(requestId == 0);
-
-        requestId = RequestIdUtils.nextRequestId();
-        assertTrue(requestId == 1);
-
-        requestId = RequestIdUtils.nextRequestId();
-        assertTrue(requestId == 2);
-    }
-
     @Test
     public void clearBuffer() {
         SenderManager testManager = new SenderManager(testConfig);

diff --git a/inlong-audit/audit-sdk/src/test/java/org/apache/inlong/audit/util/RequestIdUtilsTest.java b/inlong-audit/audit-sdk/src/test/java/org/apache/inlong/audit/util/RequestIdUtilsTest.java
@@ -25,6 +25,14 @@ public class RequestIdUtilsTest {
 
     @Test
     public void testNextRequestId() {
-        assertTrue(RequestIdUtils.nextRequestId() < RequestIdUtils.nextRequestId());
+
+        Long requestId = RequestIdUtils.nextRequestId();
+        assertTrue(requestId == 0);
+
+        requestId = RequestIdUtils.nextRequestId();
+        assertTrue(requestId == 1);
+
+        requestId = RequestIdUtils.nextRequestId();
+        assertTrue(requestId == 2);
     }
 }