Add mock-module test cases.

skywalking-sniffer/skywalking-sniffer-mock/src/main/java/com/a/eye/skywalking/sniffer/mock/context/MockTracerContextListener.java

@@ -30,6 +30,15 @@ public class MockTracerContextListener implements TracerContextListener {
         Assert.assertEquals(size, finishedTraceSegments.size());
     }
 
+    /**
+     * Assert the given index is a valid index of {@link #finishedTraceSegments}
+     *
+     * @param index the given index.
+     */
+    public void assertValidIndex(int index){
+        Assert.assertTrue(index < finishedTraceSegments.size());
+    }
+
     /**
      * Assert the {@link TraceSegment} at the given index of {@link #finishedTraceSegments},
      * and run the given {@link SegmentAssert#call(TraceSegment)} to assert.
@@ -38,7 +47,7 @@ public class MockTracerContextListener implements TracerContextListener {
      * @param segmentAssert the given assert.
      */
     public void assertTraceSegment(int index, SegmentAssert segmentAssert){
-        assertSize(index + 1);
+        assertValidIndex(index);
         segmentAssert.call(finishedTraceSegments.get(index));
     }
 

skywalking-sniffer/skywalking-sniffer-mock/src/test/java/com/a/eye/skywalking/sniffer/mock/MockTracerContextListenerTestCase.java

 package com.a.eye.skywalking.sniffer.mock;
 
 import com.a.eye.skywalking.sniffer.mock.context.MockTracerContextListener;
+import com.a.eye.skywalking.sniffer.mock.context.SegmentAssert;
 import com.a.eye.skywalking.sniffer.mock.trace.TraceSegmentBuilderFactory;
+import com.a.eye.skywalking.trace.TraceSegment;
 import org.junit.Assert;
 import org.junit.Test;
 
@@ -16,4 +18,35 @@ public class MockTracerContextListenerTestCase {
 
         Assert.assertNotNull(listener.getFinished(0));
     }
+
+    @Test(expected = AssertionError.class)
+    public void testAssertSize(){
+        MockTracerContextListener listener = new MockTracerContextListener();
+        listener.afterFinished(TraceSegmentBuilderFactory.INSTANCE.singleTomcat404Trace());
+
+        listener.assertSize(0);
+    }
+
+    @Test
+    public void testAssertTraceSegment(){
+        MockTracerContextListener listener = new MockTracerContextListener();
+        listener.afterFinished(TraceSegmentBuilderFactory.INSTANCE.singleTomcat404Trace());
+        listener.afterFinished(TraceSegmentBuilderFactory.INSTANCE.singleTomcat500Trace());
+
+        listener.assertTraceSegment(0, new SegmentAssert() {
+            @Override public void call(TraceSegment finishedSegment) {
+                Assert.assertNotNull(finishedSegment);
+            }
+        });
+    }
+
+    @Test(expected = AssertionError.class)
+    public void testClear(){
+        MockTracerContextListener listener = new MockTracerContextListener();
+        listener.afterFinished(TraceSegmentBuilderFactory.INSTANCE.singleTomcat404Trace());
+        listener.afterFinished(TraceSegmentBuilderFactory.INSTANCE.singleTomcat500Trace());
+
+        listener.clear();
+        listener.assertValidIndex(0);
+    }
 }