fix ut

apache · Feb 19, 2025 · d08aa79 · d08aa79
1 parent 9a2edef
commit d08aa79
Showing 1 changed file with 2 additions and 7 deletions.
diff --git a/...t/src/test/java/org/apache/bigtop/manager/agent/grpc/interceptor/TaskInterceptorTest.java b/...t/src/test/java/org/apache/bigtop/manager/agent/grpc/interceptor/TaskInterceptorTest.java
@@ -20,6 +20,7 @@
 
 import org.apache.bigtop.manager.common.utils.ProjectPathUtils;
 
+import org.apache.bigtop.manager.grpc.generated.TaskLogRequest;
 import org.junit.jupiter.api.AfterEach;
 import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.api.Test;
@@ -92,7 +93,7 @@ public void testTruncateLogFileFileDoesNotExist() throws IOException {
 
     @Test
     public void testIsTaskLogRequestTaskRequest() {
-        TaskLogRequest taskLogRequest = new TaskLogRequest();
+        TaskLogRequest taskLogRequest = TaskLogRequest.newBuilder().setTaskId(1L).build();
 
         Boolean result = taskInterceptor.isTaskRequest(taskLogRequest);
 
@@ -129,10 +130,4 @@ public Long getTaskId() {
             return 1L;
         }
     }
-
-    private static class TaskLogRequest {
-        public Long getTaskId() {
-            return 1L;
-        }
-    }
 }