diff --git a/apm-collector/apm-collector-agent/agent-grpc/agent-grpc-provider/src/test/java/org/apache/skywalking/apm/collector/agent/grpc/provider/handler/mock/RegisterMock.java b/apm-collector/apm-collector-agent/agent-grpc/agent-grpc-provider/src/test/java/org/apache/skywalking/apm/collector/agent/grpc/provider/handler/mock/RegisterMock.java
index 6bf5d1690..c27301eac 100644
--- a/apm-collector/apm-collector-agent/agent-grpc/agent-grpc-provider/src/test/java/org/apache/skywalking/apm/collector/agent/grpc/provider/handler/mock/RegisterMock.java
+++ b/apm-collector/apm-collector-agent/agent-grpc/agent-grpc-provider/src/test/java/org/apache/skywalking/apm/collector/agent/grpc/provider/handler/mock/RegisterMock.java
@@ -22,6 +22,7 @@ import io.grpc.ManagedChannel;
 import java.util.UUID;
 import org.apache.skywalking.apm.network.proto.Application;
 import org.apache.skywalking.apm.network.proto.ApplicationInstance;
+import org.apache.skywalking.apm.network.proto.ApplicationInstanceMapping;
 import org.apache.skywalking.apm.network.proto.ApplicationMapping;
 import org.apache.skywalking.apm.network.proto.ApplicationRegisterServiceGrpc;
 import org.apache.skywalking.apm.network.proto.InstanceDiscoveryServiceGrpc;
@@ -29,18 +30,23 @@ import org.apache.skywalking.apm.network.proto.OSInfo;
 import org.apache.skywalking.apm.network.proto.ServiceNameCollection;
 import org.apache.skywalking.apm.network.proto.ServiceNameDiscoveryServiceGrpc;
 import org.apache.skywalking.apm.network.proto.ServiceNameElement;
+import org.apache.skywalking.apm.network.proto.ServiceNameMappingCollection;
 import org.joda.time.DateTime;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
 
 /**
  * @author peng-yongsheng
  */
 class RegisterMock {
 
+    private static final Logger logger = LoggerFactory.getLogger(RegisterMock.class);
+
     private ApplicationRegisterServiceGrpc.ApplicationRegisterServiceBlockingStub applicationRegisterServiceBlockingStub;
     private InstanceDiscoveryServiceGrpc.InstanceDiscoveryServiceBlockingStub instanceDiscoveryServiceBlockingStub;
     private ServiceNameDiscoveryServiceGrpc.ServiceNameDiscoveryServiceBlockingStub serviceNameDiscoveryServiceBlockingStub;
 
-    void mock(ManagedChannel channel) {
+    void mock(ManagedChannel channel) throws InterruptedException {
         applicationRegisterServiceBlockingStub = ApplicationRegisterServiceGrpc.newBlockingStub(channel);
         instanceDiscoveryServiceBlockingStub = InstanceDiscoveryServiceGrpc.newBlockingStub(channel);
         serviceNameDiscoveryServiceBlockingStub = ServiceNameDiscoveryServiceGrpc.newBlockingStub(channel);
@@ -48,10 +54,17 @@ class RegisterMock {
         registerProvider();
     }
 
-    private void registerConsumer() {
+    private void registerConsumer() throws InterruptedException {
         Application.Builder application = Application.newBuilder();
         application.setApplicationCode("dubbox-consumer");
-        ApplicationMapping applicationMapping = applicationRegisterServiceBlockingStub.applicationCodeRegister(application.build());
+
+        ApplicationMapping applicationMapping;
+        do {
+            applicationMapping = applicationRegisterServiceBlockingStub.applicationCodeRegister(application.build());
+            logger.debug("application id: {}", applicationMapping.getApplication().getValue());
+            Thread.sleep(20);
+        }
+        while (applicationMapping.getApplication().getValue() == 0);
 
         ApplicationInstance.Builder instance = ApplicationInstance.newBuilder();
         instance.setApplicationId(applicationMapping.getApplication().getValue());
@@ -65,20 +78,35 @@ class RegisterMock {
         osInfo.addIpv4S("10.0.0.3");
         osInfo.addIpv4S("10.0.0.4");
         instance.setOsinfo(osInfo);
-        instanceDiscoveryServiceBlockingStub.registerInstance(instance.build());
+
+        ApplicationInstanceMapping instanceMapping;
+        do {
+            instanceMapping = instanceDiscoveryServiceBlockingStub.registerInstance(instance.build());
+            logger.debug("instance id: {}", instanceMapping.getApplicationInstanceId());
+            Thread.sleep(20);
+        }
+        while (instanceMapping.getApplicationInstanceId() == 0);
 
         ServiceNameCollection.Builder serviceNameCollection = ServiceNameCollection.newBuilder();
         ServiceNameElement.Builder serviceNameElement = ServiceNameElement.newBuilder();
         serviceNameElement.setApplicationId(applicationMapping.getApplication().getValue());
         serviceNameElement.setServiceName("org.skywaking.apm.testcase.dubbo.services.GreetService.doBusiness()");
         serviceNameCollection.addElements(serviceNameElement);
-        serviceNameDiscoveryServiceBlockingStub.discovery(serviceNameCollection.build());
+
+        registerServiceName(serviceNameCollection);
     }
 
-    private void registerProvider() {
+    private void registerProvider() throws InterruptedException {
         Application.Builder application = Application.newBuilder();
         application.setApplicationCode("dubbox-provider");
-        ApplicationMapping applicationMapping = applicationRegisterServiceBlockingStub.applicationCodeRegister(application.build());
+
+        ApplicationMapping applicationMapping;
+        do {
+            applicationMapping = applicationRegisterServiceBlockingStub.applicationCodeRegister(application.build());
+            logger.debug("application id: {}", applicationMapping.getApplication().getValue());
+            Thread.sleep(20);
+        }
+        while (applicationMapping.getApplication().getValue() == 0);
 
         ApplicationInstance.Builder instance = ApplicationInstance.newBuilder();
         instance.setApplicationId(applicationMapping.getApplication().getValue());
@@ -92,13 +120,34 @@ class RegisterMock {
         osInfo.addIpv4S("10.0.0.1");
         osInfo.addIpv4S("10.0.0.2");
         instance.setOsinfo(osInfo);
-        instanceDiscoveryServiceBlockingStub.registerInstance(instance.build());
+
+        ApplicationInstanceMapping instanceMapping;
+        do {
+            instanceMapping = instanceDiscoveryServiceBlockingStub.registerInstance(instance.build());
+            logger.debug("instance id: {}", instanceMapping.getApplicationInstanceId());
+            Thread.sleep(20);
+        }
+        while (instanceMapping.getApplicationInstanceId() == 0);
 
         ServiceNameCollection.Builder serviceNameCollection = ServiceNameCollection.newBuilder();
         ServiceNameElement.Builder serviceNameElement = ServiceNameElement.newBuilder();
         serviceNameElement.setApplicationId(applicationMapping.getApplication().getValue());
         serviceNameElement.setServiceName("org.skywaking.apm.testcase.dubbo.services.GreetService.doBusiness()");
         serviceNameCollection.addElements(serviceNameElement);
-        serviceNameDiscoveryServiceBlockingStub.discovery(serviceNameCollection.build());
+
+        registerServiceName(serviceNameCollection);
+    }
+
+    private void registerServiceName(ServiceNameCollection.Builder serviceNameCollection) throws InterruptedException {
+        ServiceNameMappingCollection serviceNameMappingCollection;
+        do {
+            serviceNameMappingCollection = serviceNameDiscoveryServiceBlockingStub.discovery(serviceNameCollection.build());
+            logger.debug("service name mapping collection size: {}", serviceNameMappingCollection.getElementsCount());
+            if (serviceNameMappingCollection.getElementsCount() > 0) {
+                logger.debug("service id: {}", serviceNameMappingCollection.getElements(0).getServiceId());
+            }
+            Thread.sleep(20);
+        }
+        while (serviceNameMappingCollection.getElementsCount() == 0 || serviceNameMappingCollection.getElements(0).getServiceId() == 0);
     }
 }