You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@inlong.apache.org by do...@apache.org on 2022/04/16 14:31:26 UTC

[incubator-inlong] branch master updated: [INLONG-3765][Audit] Fix audit-proxy TubeSinkTest ut (#3766)

This is an automated email from the ASF dual-hosted git repository.

dockerzhang pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-inlong.git


The following commit(s) were added to refs/heads/master by this push:
     new 91cf3e1c8 [INLONG-3765][Audit] Fix audit-proxy TubeSinkTest ut (#3766)
91cf3e1c8 is described below

commit 91cf3e1c8bcb45eb70a44037998c8d5f89999b0c
Author: xueyingzhang <86...@users.noreply.github.com>
AuthorDate: Sat Apr 16 22:31:21 2022 +0800

    [INLONG-3765][Audit] Fix audit-proxy TubeSinkTest ut (#3766)
---
 .../test/java/org/apache/inlong/audit/sink/TubeSinkTest.java | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/inlong-audit/audit-proxy/src/test/java/org/apache/inlong/audit/sink/TubeSinkTest.java b/inlong-audit/audit-proxy/src/test/java/org/apache/inlong/audit/sink/TubeSinkTest.java
index 69f601b30..7b6f68008 100644
--- a/inlong-audit/audit-proxy/src/test/java/org/apache/inlong/audit/sink/TubeSinkTest.java
+++ b/inlong-audit/audit-proxy/src/test/java/org/apache/inlong/audit/sink/TubeSinkTest.java
@@ -31,21 +31,21 @@ import org.apache.flume.lifecycle.LifecycleController;
 import org.apache.flume.lifecycle.LifecycleState;
 import org.junit.Before;
 import org.junit.Test;
-import org.slf4j.Logger;
-import org.slf4j.LoggerFactory;
+import org.powermock.api.mockito.PowerMockito;
 import org.testng.Assert;
 
 public class TubeSinkTest {
 
-    private static final Logger LOG = LoggerFactory.getLogger(TubeSink.class);
-
     private TubeSink tubeSink;
     private Channel channel;
     private Context context;
 
     @Before
-    public void setUp() {
-        tubeSink = new TubeSink();
+    public void setUp() throws Exception {
+        tubeSink = PowerMockito.mock(TubeSink.class);
+        PowerMockito.doNothing().when(tubeSink, "start");
+        PowerMockito.when(tubeSink.process()).thenReturn(Sink.Status.READY);
+        PowerMockito.when(tubeSink.getLifecycleState()).thenReturn(LifecycleState.ERROR);
         channel = new MemoryChannel();
         context = new Context();