pinpoint-apm#745 filter wizard bug fix. incorrect agent mapping

- backport pinpoint-apm#732 - add testcase
emeroad · Jul 20, 2015 · d81077b · d81077b
1 parent 08095d9
commit d81077b
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/web/src/test/java/com/navercorp/pinpoint/web/applicationmap/ServerInstanceListTest.java b/web/src/test/java/com/navercorp/pinpoint/web/applicationmap/ServerInstanceListTest.java
@@ -17,7 +17,7 @@
 package com.navercorp.pinpoint.web.applicationmap;
 
 import com.navercorp.pinpoint.common.bo.AgentInfoBo;
-import com.navercorp.pinpoint.common.trace.ServiceType;
+import com.navercorp.pinpoint.common.ServiceType;
 import org.junit.Assert;
 import org.junit.Test;
 
@@ -54,14 +54,12 @@ public void testGetAgentIdList() throws Exception {
 
     public static AgentInfoBo createAgentInfo(String agentId, String hostName) {
         AgentInfoBo.Builder agentInfoBuilder = new AgentInfoBo.Builder();
-        agentInfoBuilder.setAgentId(agentId);
+        agentInfoBuilder.agentId(agentId);
 
         ServiceType serviceType = ServiceType.TEST_STAND_ALONE;
-        agentInfoBuilder.setServiceTypeCode(serviceType.getCode());
-        // TODO FIX api
-        agentInfoBuilder.setServiceType(serviceType);
+        agentInfoBuilder.serviceType(serviceType);
 
-        agentInfoBuilder.setHostName(hostName);
+        agentInfoBuilder.hostName(hostName);
 
         return agentInfoBuilder.build();
     }