From 24010d134b9d01fffc9d596ea213c1b79cbc8619 Mon Sep 17 00:00:00 2001
From: ascrutae <nzt48.xin@gmail.com>
Date: Thu, 27 Apr 2017 16:28:45 +0800
Subject: [PATCH] fix issue that test compile failed

---
 .../assist/NoConcurrencyAccessObjectTest.java | 46 ++++++++++++-------
 1 file changed, 29 insertions(+), 17 deletions(-)

diff --git a/skywalking-sniffer/skywalking-api/src/test/java/com/a/eye/skywalking/api/plugin/assist/NoConcurrencyAccessObjectTest.java b/skywalking-sniffer/skywalking-api/src/test/java/com/a/eye/skywalking/api/plugin/assist/NoConcurrencyAccessObjectTest.java
index fcd83db592..91f2bc3efe 100644
--- a/skywalking-sniffer/skywalking-api/src/test/java/com/a/eye/skywalking/api/plugin/assist/NoConcurrencyAccessObjectTest.java
+++ b/skywalking-sniffer/skywalking-api/src/test/java/com/a/eye/skywalking/api/plugin/assist/NoConcurrencyAccessObjectTest.java
@@ -2,41 +2,53 @@ package com.a.eye.skywalking.api.plugin.assist;
 
 import com.a.eye.skywalking.api.plugin.interceptor.EnhancedClassInstanceContext;
 import com.a.eye.skywalking.api.plugin.interceptor.assist.NoConcurrencyAccessObject;
+import com.a.eye.skywalking.api.plugin.interceptor.enhance.InstanceMethodInvokeContext;
 import org.junit.Assert;
 import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.mockito.Mock;
+import org.mockito.runners.MockitoJUnitRunner;
 
 /**
  * @author wusheng
  */
+@RunWith(MockitoJUnitRunner.class)
 public class NoConcurrencyAccessObjectTest {
+
+    @Mock
+    private InstanceMethodInvokeContext invokeContext;
+
     @Test
     public void testEntraExitCounter() {
-        NoConcurrencyAccessObject object = new NoConcurrencyAccessObject();
         final EnhancedClassInstanceContext context = new EnhancedClassInstanceContext();
-        object.whenEnter(context, new Runnable() {
+        NoConcurrencyAccessObject first = new NoConcurrencyAccessObject(){
+
             @Override
-            public void run() {
+            protected void enter(EnhancedClassInstanceContext context, InstanceMethodInvokeContext interceptorContext) {
                 context.set("firstEntrance", true);
             }
-        });
-        object.whenEnter(context, new Runnable() {
-            @Override
-            public void run() {
-                context.set("secondEntrance", true);
-            }
-        });
-        object.whenExist(context, new Runnable() {
-            @Override
-            public void run() {
+
+            @Override protected void exit() {
                 context.set("firstExit", true);
             }
-        });
-        object.whenExist(context, new Runnable() {
+        };
+
+        NoConcurrencyAccessObject second = new NoConcurrencyAccessObject(){
+
             @Override
-            public void run() {
+            protected void enter(EnhancedClassInstanceContext context, InstanceMethodInvokeContext interceptorContext) {
+                context.set("secondEntrance", true);
+            }
+
+            @Override protected void exit() {
                 context.set("lastEntrance", true);
             }
-        });
+        };
+
+        first.whenEnter(context, invokeContext);
+        second.whenEnter(context, invokeContext);
+        first.whenExist(context);
+        second.whenExist(context);
 
         Assert.assertTrue(!context.isContain("secondEntrance"));
         Assert.assertTrue(!context.isContain("firstExit"));
-- 
GitLab