SpringCloud/spring-cloud-openfeign
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix broken test.

Browse Source
This commit is contained in:
Spencer Gibb
2016-11-07 11:24:52 -07:00
parent 93cd1e7a4a
commit aec8ec43ef
1 changed files with 35 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

69
spring-cloud-netflix-eureka-server/src/test/java/org/springframework/cloud/netflix/eureka/server/EurekaControllerReplicasTest.java
View File

@@ -9,7 +9,6 @@ import java.util.Collections;
import java.util.HashMap;
import java.util.Map;
import com.netflix.appinfo.ApplicationInfoManager;
import com.netflix.appinfo.DataCenterInfo;
import com.netflix.appinfo.InstanceInfo;
@@ -22,68 +21,70 @@ import org.junit.Test;
import com.netflix.eureka.util.StatusInfo;
public class EurekaControllerReplicasTest {
String noAuthList1 = "http://test1.com";
String noAuthList2 = noAuthList1+",http://test2.com";
String noAuthList2 = noAuthList1 + ",http://test2.com";
String authList1 = "http://user:pwd@test1.com";
String authList2 = authList1+",http://user2:pwd2@test2.com";
String authList2 = authList1 + ",http://user2:pwd2@test2.com";
String empty = new String();
private ApplicationInfoManager original;
private InstanceInfo instanceInfo;
@Before
public void setup() throws Exception {
this.original = ApplicationInfoManager.getInstance();
setInstance(mock(ApplicationInfoManager.class));
instanceInfo = mock(InstanceInfo.class);
}
@After
public void teardown() throws Exception {
setInstance(this.original);
instanceInfo = null;
}
@Test
public void testFilterReplicasNoAuth() throws Exception {
Map<String, Object> model=new HashMap<String, Object>();
StatusInfo statusInfo = StatusInfo.Builder.newBuilder().
add("registered-replicas", empty).
add("available-replicas",noAuthList1).
add("unavailable-replicas",noAuthList2).
build();
Map<String, Object> model = new HashMap<>();
StatusInfo statusInfo = StatusInfo.Builder.newBuilder()
.add("registered-replicas", empty)
.add("available-replicas", noAuthList1)
.add("unavailable-replicas", noAuthList2)
.withInstanceInfo(this.instanceInfo).build();
EurekaController controller = new EurekaController(null);
controller.filterReplicas(model,statusInfo);
controller.filterReplicas(model, statusInfo);
@SuppressWarnings("unchecked")
Map<String,String> results = (Map<String, String>) model.get("applicationStats");
assertEquals(empty,results.get("registered-replicas"));
assertEquals(noAuthList1,results.get("available-replicas"));
assertEquals(noAuthList2,results.get("unavailable-replicas"));
Map<String, String> results = (Map<String, String>) model.get("applicationStats");
assertEquals(empty, results.get("registered-replicas"));
assertEquals(noAuthList1, results.get("available-replicas"));
assertEquals(noAuthList2, results.get("unavailable-replicas"));
}
@Test
public void testFilterReplicasAuth() throws Exception {
Map<String, Object> model=new HashMap<String, Object>();
StatusInfo statusInfo = StatusInfo.Builder.newBuilder().
add("registered-replicas", authList2).
add("available-replicas",authList1).
add("unavailable-replicas",empty).
build();
Map<String, Object> model = new HashMap<>();
StatusInfo statusInfo = StatusInfo.Builder.newBuilder()
.add("registered-replicas", authList2)
.add("available-replicas", authList1)
.add("unavailable-replicas", empty)
.withInstanceInfo(instanceInfo).build();
EurekaController controller = new EurekaController(null);
controller.filterReplicas(model,statusInfo);
controller.filterReplicas(model, statusInfo);
@SuppressWarnings("unchecked")
Map<String,String> results = (Map<String, String>) model.get("applicationStats");
assertEquals(empty,results.get("unavailable-replicas"));
assertEquals(noAuthList1,results.get("available-replicas"));
assertEquals(noAuthList2,results.get("registered-replicas"));
Map<String, String> results = (Map<String, String>) model.get("applicationStats");
assertEquals(empty, results.get("unavailable-replicas"));
assertEquals(noAuthList1, results.get("available-replicas"));
assertEquals(noAuthList2, results.get("registered-replicas"));
}
}