From c5eb330c1cbacde50c6a51eb2e068e162fd65446 Mon Sep 17 00:00:00 2001
From: pengys5 <8082209@qq.com>
Date: Wed, 26 Apr 2017 11:40:26 +0800
Subject: [PATCH] testcase

---
 .../collector/actor/AbstractWorker.java       | 11 +++++
 .../worker/RecordAnalysisMember.java          |  9 ++--
 .../worker/RecordPersistenceMember.java       |  8 +---
 .../RecordPersistenceMemberTestCase.java      | 41 +++++++++++++++++++
 .../globaltrace/GlobalTraceIndexTestCase.java |  7 ++++
 .../worker/node/NodeCompIndexTestCase.java    |  7 ++++
 .../worker/node/NodeMappingIndexTestCase.java |  7 ++++
 .../worker/noderef/NodeRefIndexTestCase.java  | 11 +++--
 .../noderef/NodeRefResSumIndexTestCase.java   |  7 ++++
 .../segment/SegmentCostIndexTestCase.java     | 11 +++--
 .../SegmentExceptionIndexTestCase.java        | 11 +++--
 .../worker/segment/SegmentIndexTestCase.java  |  7 ++++
 .../analysis/SegmentCostAnalysisTestCase.java | 35 ++++++++++++++++
 13 files changed, 154 insertions(+), 18 deletions(-)
 create mode 100644 skywalking-collector/skywalking-collector-worker/src/test/java/com/a/eye/skywalking/collector/worker/RecordPersistenceMemberTestCase.java
 create mode 100644 skywalking-collector/skywalking-collector-worker/src/test/java/com/a/eye/skywalking/collector/worker/segment/analysis/SegmentCostAnalysisTestCase.java

diff --git a/skywalking-collector/skywalking-collector-cluster/src/main/java/com/a/eye/skywalking/collector/actor/AbstractWorker.java b/skywalking-collector/skywalking-collector-cluster/src/main/java/com/a/eye/skywalking/collector/actor/AbstractWorker.java
index e5a5f8a03..db842e428 100644
--- a/skywalking-collector/skywalking-collector-cluster/src/main/java/com/a/eye/skywalking/collector/actor/AbstractWorker.java
+++ b/skywalking-collector/skywalking-collector-cluster/src/main/java/com/a/eye/skywalking/collector/actor/AbstractWorker.java
@@ -1,9 +1,15 @@
 package com.a.eye.skywalking.collector.actor;
 
+import org.apache.logging.log4j.LogManager;
+import org.apache.logging.log4j.Logger;
+
 /**
  * @author pengys5
  */
 public abstract class AbstractWorker {
+
+    private final Logger logger;
+
     private final LocalWorkerContext selfContext;
 
     private final Role role;
@@ -14,6 +20,11 @@ public abstract class AbstractWorker {
         this.role = role;
         this.clusterContext = clusterContext;
         this.selfContext = selfContext;
+        this.logger = LogManager.getFormatterLogger(role.roleName());
+    }
+
+    final public Logger logger() {
+        return logger;
     }
 
     public abstract void preStart() throws ProviderNotFoundException;
diff --git a/skywalking-collector/skywalking-collector-worker/src/main/java/com/a/eye/skywalking/collector/worker/RecordAnalysisMember.java b/skywalking-collector/skywalking-collector-worker/src/main/java/com/a/eye/skywalking/collector/worker/RecordAnalysisMember.java
index c4c912c8e..0f2edfcab 100644
--- a/skywalking-collector/skywalking-collector-worker/src/main/java/com/a/eye/skywalking/collector/worker/RecordAnalysisMember.java
+++ b/skywalking-collector/skywalking-collector-worker/src/main/java/com/a/eye/skywalking/collector/worker/RecordAnalysisMember.java
@@ -6,12 +6,16 @@ import com.a.eye.skywalking.collector.actor.Role;
 import com.a.eye.skywalking.collector.actor.WorkerRefs;
 import com.a.eye.skywalking.collector.worker.storage.RecordAnalysisData;
 import com.google.gson.JsonObject;
+import org.apache.logging.log4j.LogManager;
+import org.apache.logging.log4j.Logger;
 
 /**
  * @author pengys5
  */
 public abstract class RecordAnalysisMember extends AnalysisMember {
 
+    private Logger logger = LogManager.getFormatterLogger(RecordAnalysisMember.class);
+
     private RecordAnalysisData recordAnalysisData = new RecordAnalysisData();
 
     public RecordAnalysisMember(Role role, ClusterWorkerContext clusterContext, LocalWorkerContext selfContext) {
@@ -26,13 +30,12 @@ public abstract class RecordAnalysisMember extends AnalysisMember {
         return recordAnalysisData;
     }
 
-    @Override
-    final protected void aggregation() throws Exception {
+    @Override final protected void aggregation() throws Exception {
         getRecordAnalysisData().asMap().forEach((key, value) -> {
             try {
                 aggWorkRefs().tell(value);
             } catch (Exception e) {
-                e.printStackTrace();
+                logger.error(e);
             }
         });
         getRecordAnalysisData().asMap().clear();
diff --git a/skywalking-collector/skywalking-collector-worker/src/main/java/com/a/eye/skywalking/collector/worker/RecordPersistenceMember.java b/skywalking-collector/skywalking-collector-worker/src/main/java/com/a/eye/skywalking/collector/worker/RecordPersistenceMember.java
index 84cf9ade2..c0dc184c8 100644
--- a/skywalking-collector/skywalking-collector-worker/src/main/java/com/a/eye/skywalking/collector/worker/RecordPersistenceMember.java
+++ b/skywalking-collector/skywalking-collector-worker/src/main/java/com/a/eye/skywalking/collector/worker/RecordPersistenceMember.java
@@ -8,8 +8,6 @@ import com.a.eye.skywalking.collector.worker.storage.RecordData;
 import com.a.eye.skywalking.collector.worker.storage.RecordPersistenceData;
 import java.util.List;
 import java.util.Map;
-import org.apache.logging.log4j.LogManager;
-import org.apache.logging.log4j.Logger;
 import org.elasticsearch.action.index.IndexRequestBuilder;
 import org.elasticsearch.client.Client;
 
@@ -18,8 +16,6 @@ import org.elasticsearch.client.Client;
  */
 public abstract class RecordPersistenceMember extends PersistenceMember<RecordPersistenceData, RecordData> {
 
-    private Logger logger = LogManager.getFormatterLogger(RecordPersistenceMember.class);
-
     public RecordPersistenceMember(Role role, ClusterWorkerContext clusterContext, LocalWorkerContext selfContext) {
         super(role, clusterContext, selfContext);
     }
@@ -33,13 +29,13 @@ public abstract class RecordPersistenceMember extends PersistenceMember<RecordPe
     public void analyse(Object message) throws Exception {
         if (message instanceof RecordData) {
             RecordData recordData = (RecordData) message;
-            logger.debug("setRecord: id: %s, data: %s", recordData.getId(), recordData.getRecord());
+            logger().debug("setRecord: id: %s, data: %s", recordData.getId(), recordData.getRecord());
             RecordPersistenceData data = getPersistenceData();
             data.hold();
             data.getOrCreate(recordData.getId()).setRecord(recordData.getRecord());
             data.release();
         } else {
-            logger.error("message unhandled");
+            logger().error("message unhandled");
         }
     }
 
diff --git a/skywalking-collector/skywalking-collector-worker/src/test/java/com/a/eye/skywalking/collector/worker/RecordPersistenceMemberTestCase.java b/skywalking-collector/skywalking-collector-worker/src/test/java/com/a/eye/skywalking/collector/worker/RecordPersistenceMemberTestCase.java
new file mode 100644
index 000000000..c1d27e9c1
--- /dev/null
+++ b/skywalking-collector/skywalking-collector-worker/src/test/java/com/a/eye/skywalking/collector/worker/RecordPersistenceMemberTestCase.java
@@ -0,0 +1,41 @@
+package com.a.eye.skywalking.collector.worker;
+
+import com.a.eye.skywalking.collector.worker.storage.RecordData;
+import com.a.eye.skywalking.collector.worker.storage.RecordPersistenceData;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.powermock.api.mockito.PowerMockito;
+import org.powermock.core.classloader.annotations.PrepareForTest;
+import org.powermock.modules.junit4.PowerMockRunner;
+import org.apache.logging.log4j.Logger;
+
+import static org.mockito.Mockito.*;
+
+/**
+ * @author pengys5
+ */
+@RunWith(PowerMockRunner.class)
+@PrepareForTest(RecordPersistenceMember.class)
+public class RecordPersistenceMemberTestCase {
+
+    @Test
+    public void analyse() throws Exception {
+        RecordPersistenceMember member = PowerMockito.mock(RecordPersistenceMember.class, CALLS_REAL_METHODS);
+        Logger logger = mock(Logger.class);
+        when(member.logger()).thenReturn(logger);
+
+        RecordPersistenceData data = mock(RecordPersistenceData.class);
+        RecordData recordData = new RecordData("test1");
+        when(data.getOrCreate(anyString())).thenReturn(recordData);
+
+        when(member.getPersistenceData()).thenReturn(data);
+
+        RecordData recordData_1 = new RecordData("test2");
+        member.analyse(recordData_1);
+
+        verify(data).hold();
+        verify(data).release();
+
+        member.analyse(new Object());
+    }
+}
diff --git a/skywalking-collector/skywalking-collector-worker/src/test/java/com/a/eye/skywalking/collector/worker/globaltrace/GlobalTraceIndexTestCase.java b/skywalking-collector/skywalking-collector-worker/src/test/java/com/a/eye/skywalking/collector/worker/globaltrace/GlobalTraceIndexTestCase.java
index cc4e9dbf7..871495e4e 100644
--- a/skywalking-collector/skywalking-collector-worker/src/test/java/com/a/eye/skywalking/collector/worker/globaltrace/GlobalTraceIndexTestCase.java
+++ b/skywalking-collector/skywalking-collector-worker/src/test/java/com/a/eye/skywalking/collector/worker/globaltrace/GlobalTraceIndexTestCase.java
@@ -1,5 +1,6 @@
 package com.a.eye.skywalking.collector.worker.globaltrace;
 
+import com.a.eye.skywalking.collector.worker.config.EsConfig;
 import org.junit.Assert;
 import org.junit.Test;
 
@@ -22,4 +23,10 @@ public class GlobalTraceIndexTestCase {
         GlobalTraceIndex index = new GlobalTraceIndex();
         Assert.assertEquals("{\"properties\":{\"subSegIds\":{\"type\":\"keyword\"}}}", index.createMappingBuilder().string());
     }
+
+    @Test
+    public void refreshInterval() {
+        GlobalTraceIndex index = new GlobalTraceIndex();
+        Assert.assertEquals(EsConfig.Es.Index.RefreshInterval.GlobalTraceIndex.VALUE.intValue(), index.refreshInterval());
+    }
 }
diff --git a/skywalking-collector/skywalking-collector-worker/src/test/java/com/a/eye/skywalking/collector/worker/node/NodeCompIndexTestCase.java b/skywalking-collector/skywalking-collector-worker/src/test/java/com/a/eye/skywalking/collector/worker/node/NodeCompIndexTestCase.java
index 7a8782b0e..4c4628ba4 100644
--- a/skywalking-collector/skywalking-collector-worker/src/test/java/com/a/eye/skywalking/collector/worker/node/NodeCompIndexTestCase.java
+++ b/skywalking-collector/skywalking-collector-worker/src/test/java/com/a/eye/skywalking/collector/worker/node/NodeCompIndexTestCase.java
@@ -1,5 +1,6 @@
 package com.a.eye.skywalking.collector.worker.node;
 
+import com.a.eye.skywalking.collector.worker.config.EsConfig;
 import java.io.IOException;
 import org.junit.Assert;
 import org.junit.Test;
@@ -21,4 +22,10 @@ public class NodeCompIndexTestCase {
         NodeCompIndex index = new NodeCompIndex();
         Assert.assertEquals("{\"properties\":{\"name\":{\"type\":\"keyword\"},\"peers\":{\"type\":\"keyword\"},\"aggId\":{\"type\":\"keyword\"}}}", index.createMappingBuilder().string());
     }
+
+    @Test
+    public void refreshInterval() {
+        NodeCompIndex index = new NodeCompIndex();
+        Assert.assertEquals(EsConfig.Es.Index.RefreshInterval.NodeCompIndex.VALUE.intValue(), index.refreshInterval());
+    }
 }
diff --git a/skywalking-collector/skywalking-collector-worker/src/test/java/com/a/eye/skywalking/collector/worker/node/NodeMappingIndexTestCase.java b/skywalking-collector/skywalking-collector-worker/src/test/java/com/a/eye/skywalking/collector/worker/node/NodeMappingIndexTestCase.java
index b69d379a1..e7e015cf4 100644
--- a/skywalking-collector/skywalking-collector-worker/src/test/java/com/a/eye/skywalking/collector/worker/node/NodeMappingIndexTestCase.java
+++ b/skywalking-collector/skywalking-collector-worker/src/test/java/com/a/eye/skywalking/collector/worker/node/NodeMappingIndexTestCase.java
@@ -1,5 +1,6 @@
 package com.a.eye.skywalking.collector.worker.node;
 
+import com.a.eye.skywalking.collector.worker.config.EsConfig;
 import org.junit.Assert;
 import org.junit.Test;
 
@@ -22,4 +23,10 @@ public class NodeMappingIndexTestCase {
         NodeMappingIndex index = new NodeMappingIndex();
         Assert.assertEquals("{\"properties\":{\"code\":{\"type\":\"string\",\"index\":\"not_analyzed\"},\"peers\":{\"type\":\"string\",\"index\":\"not_analyzed\"},\"aggId\":{\"type\":\"string\",\"index\":\"not_analyzed\"},\"timeSlice\":{\"type\":\"long\",\"index\":\"not_analyzed\"}}}", index.createMappingBuilder().string());
     }
+
+    @Test
+    public void refreshInterval() {
+        NodeMappingIndex index = new NodeMappingIndex();
+        Assert.assertEquals(EsConfig.Es.Index.RefreshInterval.NodeMappingIndex.VALUE.intValue(), index.refreshInterval());
+    }
 }
diff --git a/skywalking-collector/skywalking-collector-worker/src/test/java/com/a/eye/skywalking/collector/worker/noderef/NodeRefIndexTestCase.java b/skywalking-collector/skywalking-collector-worker/src/test/java/com/a/eye/skywalking/collector/worker/noderef/NodeRefIndexTestCase.java
index 2420f64d2..edc3080d6 100644
--- a/skywalking-collector/skywalking-collector-worker/src/test/java/com/a/eye/skywalking/collector/worker/noderef/NodeRefIndexTestCase.java
+++ b/skywalking-collector/skywalking-collector-worker/src/test/java/com/a/eye/skywalking/collector/worker/noderef/NodeRefIndexTestCase.java
@@ -1,11 +1,10 @@
 package com.a.eye.skywalking.collector.worker.noderef;
 
-import com.a.eye.skywalking.collector.worker.globaltrace.GlobalTraceIndex;
+import com.a.eye.skywalking.collector.worker.config.EsConfig;
+import java.io.IOException;
 import org.junit.Assert;
 import org.junit.Test;
 
-import java.io.IOException;
-
 /**
  * @author pengys5
  */
@@ -23,4 +22,10 @@ public class NodeRefIndexTestCase {
         NodeRefIndex index = new NodeRefIndex();
         Assert.assertEquals("{\"properties\":{\"front\":{\"type\":\"keyword\"},\"frontIsRealCode\":{\"type\":\"boolean\",\"index\":\"not_analyzed\"},\"behind\":{\"type\":\"keyword\"},\"behindIsRealCode\":{\"type\":\"boolean\",\"index\":\"not_analyzed\"},\"aggId\":{\"type\":\"keyword\"},\"timeSlice\":{\"type\":\"long\",\"index\":\"not_analyzed\"}}}", index.createMappingBuilder().string());
     }
+
+    @Test
+    public void refreshInterval() {
+        NodeRefIndex index = new NodeRefIndex();
+        Assert.assertEquals(EsConfig.Es.Index.RefreshInterval.NodeRefIndex.VALUE.intValue(), index.refreshInterval());
+    }
 }
diff --git a/skywalking-collector/skywalking-collector-worker/src/test/java/com/a/eye/skywalking/collector/worker/noderef/NodeRefResSumIndexTestCase.java b/skywalking-collector/skywalking-collector-worker/src/test/java/com/a/eye/skywalking/collector/worker/noderef/NodeRefResSumIndexTestCase.java
index 2d0c6c4c7..d32c66556 100644
--- a/skywalking-collector/skywalking-collector-worker/src/test/java/com/a/eye/skywalking/collector/worker/noderef/NodeRefResSumIndexTestCase.java
+++ b/skywalking-collector/skywalking-collector-worker/src/test/java/com/a/eye/skywalking/collector/worker/noderef/NodeRefResSumIndexTestCase.java
@@ -1,5 +1,6 @@
 package com.a.eye.skywalking.collector.worker.noderef;
 
+import com.a.eye.skywalking.collector.worker.config.EsConfig;
 import org.junit.Assert;
 import org.junit.Test;
 
@@ -22,4 +23,10 @@ public class NodeRefResSumIndexTestCase {
         NodeRefResSumIndex index = new NodeRefResSumIndex();
         Assert.assertEquals("{\"properties\":{\"oneSecondLess\":{\"type\":\"long\",\"index\":\"not_analyzed\"},\"threeSecondLess\":{\"type\":\"long\",\"index\":\"not_analyzed\"},\"fiveSecondLess\":{\"type\":\"long\",\"index\":\"not_analyzed\"},\"fiveSecondGreater\":{\"type\":\"long\",\"index\":\"not_analyzed\"},\"error\":{\"type\":\"long\",\"index\":\"not_analyzed\"},\"summary\":{\"type\":\"long\",\"index\":\"not_analyzed\"},\"aggId\":{\"type\":\"keyword\"},\"timeSlice\":{\"type\":\"long\",\"index\":\"not_analyzed\"}}}", index.createMappingBuilder().string());
     }
+
+    @Test
+    public void refreshInterval() {
+        NodeRefResSumIndex index = new NodeRefResSumIndex();
+        Assert.assertEquals(EsConfig.Es.Index.RefreshInterval.NodeRefResSumIndex.VALUE.intValue(), index.refreshInterval());
+    }
 }
diff --git a/skywalking-collector/skywalking-collector-worker/src/test/java/com/a/eye/skywalking/collector/worker/segment/SegmentCostIndexTestCase.java b/skywalking-collector/skywalking-collector-worker/src/test/java/com/a/eye/skywalking/collector/worker/segment/SegmentCostIndexTestCase.java
index 917cd881f..50f02f040 100644
--- a/skywalking-collector/skywalking-collector-worker/src/test/java/com/a/eye/skywalking/collector/worker/segment/SegmentCostIndexTestCase.java
+++ b/skywalking-collector/skywalking-collector-worker/src/test/java/com/a/eye/skywalking/collector/worker/segment/SegmentCostIndexTestCase.java
@@ -1,11 +1,10 @@
 package com.a.eye.skywalking.collector.worker.segment;
 
-import com.a.eye.skywalking.collector.worker.globaltrace.GlobalTraceIndex;
+import com.a.eye.skywalking.collector.worker.config.EsConfig;
+import java.io.IOException;
 import org.junit.Assert;
 import org.junit.Test;
 
-import java.io.IOException;
-
 /**
  * @author pengys5
  */
@@ -23,4 +22,10 @@ public class SegmentCostIndexTestCase {
         SegmentCostIndex index = new SegmentCostIndex();
         Assert.assertEquals("{\"properties\":{\"segId\":{\"type\":\"keyword\"},\"startTime\":{\"type\":\"long\",\"index\":\"not_analyzed\"},\"END_TIME\":{\"type\":\"long\",\"index\":\"not_analyzed\"},\"operationName\":{\"type\":\"keyword\"},\"cost\":{\"type\":\"long\",\"index\":\"not_analyzed\"}}}", index.createMappingBuilder().string());
     }
+
+    @Test
+    public void refreshInterval() {
+        SegmentCostIndex index = new SegmentCostIndex();
+        Assert.assertEquals(EsConfig.Es.Index.RefreshInterval.SegmentCostIndex.VALUE.intValue(), index.refreshInterval());
+    }
 }
diff --git a/skywalking-collector/skywalking-collector-worker/src/test/java/com/a/eye/skywalking/collector/worker/segment/SegmentExceptionIndexTestCase.java b/skywalking-collector/skywalking-collector-worker/src/test/java/com/a/eye/skywalking/collector/worker/segment/SegmentExceptionIndexTestCase.java
index 5a1f57ef3..bec336196 100644
--- a/skywalking-collector/skywalking-collector-worker/src/test/java/com/a/eye/skywalking/collector/worker/segment/SegmentExceptionIndexTestCase.java
+++ b/skywalking-collector/skywalking-collector-worker/src/test/java/com/a/eye/skywalking/collector/worker/segment/SegmentExceptionIndexTestCase.java
@@ -1,11 +1,10 @@
 package com.a.eye.skywalking.collector.worker.segment;
 
-import com.a.eye.skywalking.collector.worker.globaltrace.GlobalTraceIndex;
+import com.a.eye.skywalking.collector.worker.config.EsConfig;
+import java.io.IOException;
 import org.junit.Assert;
 import org.junit.Test;
 
-import java.io.IOException;
-
 /**
  * @author pengys5
  */
@@ -23,4 +22,10 @@ public class SegmentExceptionIndexTestCase {
         SegmentExceptionIndex index = new SegmentExceptionIndex();
         Assert.assertEquals("{\"properties\":{\"segId\":{\"type\":\"keyword\"},\"isError\":{\"type\":\"boolean\",\"index\":\"not_analyzed\"},\"errorKind\":{\"type\":\"keyword\"}}}", index.createMappingBuilder().string());
     }
+
+    @Test
+    public void refreshInterval() {
+        SegmentExceptionIndex index = new SegmentExceptionIndex();
+        Assert.assertEquals(EsConfig.Es.Index.RefreshInterval.SegmentExceptionIndex.VALUE.intValue(), index.refreshInterval());
+    }
 }
diff --git a/skywalking-collector/skywalking-collector-worker/src/test/java/com/a/eye/skywalking/collector/worker/segment/SegmentIndexTestCase.java b/skywalking-collector/skywalking-collector-worker/src/test/java/com/a/eye/skywalking/collector/worker/segment/SegmentIndexTestCase.java
index 89c8dc229..d16667c6c 100644
--- a/skywalking-collector/skywalking-collector-worker/src/test/java/com/a/eye/skywalking/collector/worker/segment/SegmentIndexTestCase.java
+++ b/skywalking-collector/skywalking-collector-worker/src/test/java/com/a/eye/skywalking/collector/worker/segment/SegmentIndexTestCase.java
@@ -1,5 +1,6 @@
 package com.a.eye.skywalking.collector.worker.segment;
 
+import com.a.eye.skywalking.collector.worker.config.EsConfig;
 import org.junit.Assert;
 import org.junit.Test;
 
@@ -22,4 +23,10 @@ public class SegmentIndexTestCase {
         SegmentIndex index = new SegmentIndex();
         Assert.assertEquals("{\"properties\":{\"traceSegmentId\":{\"type\":\"keyword\"},\"startTime\":{\"type\":\"date\",\"index\":\"not_analyzed\"},\"endTime\":{\"type\":\"date\",\"index\":\"not_analyzed\"},\"applicationCode\":{\"type\":\"keyword\"},\"minute\":{\"type\":\"long\",\"index\":\"not_analyzed\"},\"hour\":{\"type\":\"long\",\"index\":\"not_analyzed\"},\"day\":{\"type\":\"long\",\"index\":\"not_analyzed\"}}}", index.createMappingBuilder().string());
     }
+
+    @Test
+    public void refreshInterval() {
+        SegmentIndex index = new SegmentIndex();
+        Assert.assertEquals(EsConfig.Es.Index.RefreshInterval.SegmentIndex.VALUE.intValue(), index.refreshInterval());
+    }
 }
diff --git a/skywalking-collector/skywalking-collector-worker/src/test/java/com/a/eye/skywalking/collector/worker/segment/analysis/SegmentCostAnalysisTestCase.java b/skywalking-collector/skywalking-collector-worker/src/test/java/com/a/eye/skywalking/collector/worker/segment/analysis/SegmentCostAnalysisTestCase.java
new file mode 100644
index 000000000..3ad7b0a7e
--- /dev/null
+++ b/skywalking-collector/skywalking-collector-worker/src/test/java/com/a/eye/skywalking/collector/worker/segment/analysis/SegmentCostAnalysisTestCase.java
@@ -0,0 +1,35 @@
+package com.a.eye.skywalking.collector.worker.segment.analysis;
+
+import com.a.eye.skywalking.collector.actor.LocalSyncWorkerRef;
+import com.a.eye.skywalking.collector.actor.LocalWorkerContext;
+import com.a.eye.skywalking.collector.actor.WorkerNotFoundException;
+import com.a.eye.skywalking.collector.actor.WorkerRef;
+import com.a.eye.skywalking.collector.actor.WorkerRefs;
+import com.a.eye.skywalking.collector.worker.segment.persistence.SegmentCostSave;
+import java.lang.reflect.Field;
+import java.util.List;
+import org.junit.Assert;
+import org.junit.Test;
+
+/**
+ * @author pengys5
+ */
+public class SegmentCostAnalysisTestCase {
+
+    @Test
+    public void aggWorkRefs() throws WorkerNotFoundException, NoSuchFieldException, IllegalAccessException {
+        LocalWorkerContext localWorkerContext = new LocalWorkerContext();
+
+        WorkerRef workerRef = new LocalSyncWorkerRef(SegmentCostSave.Role.INSTANCE, null);
+        localWorkerContext.put(workerRef);
+
+        SegmentCostAnalysis analysis = new SegmentCostAnalysis(SegmentCostAnalysis.Role.INSTANCE, null, localWorkerContext);
+        WorkerRefs workerRefs = analysis.aggWorkRefs();
+
+        Field testAField = workerRefs.getClass().getDeclaredField("workerRefs");
+        testAField.setAccessible(true);
+        List<LocalSyncWorkerRef> list = (List<LocalSyncWorkerRef>)testAField.get(workerRefs);
+
+        Assert.assertEquals(workerRef, list.get(0));
+    }
+}