Merge pull request #1127 from william-tran/schemeless-uri

spring-cloud-netflix-core/src/main/java/org/springframework/cloud/netflix/ribbon/RibbonLoadBalancerClient.java

@@ -114,7 +114,10 @@ public class RibbonLoadBalancerClient implements LoadBalancerClient {
 	private boolean isSecure(Server server, String serviceId) {
 		IClientConfig config = this.clientFactory.getClientConfig(serviceId);
 		if (config != null) {
-			return config.get(CommonClientConfigKey.IsSecure, false);
+			Boolean isSecure = config.get(CommonClientConfigKey.IsSecure);
+			if (isSecure != null) {
+				return isSecure;
+			}
 		}
 
 		return serverIntrospector(serviceId).isSecure(server);

spring-cloud-netflix-core/src/test/java/org/springframework/cloud/netflix/ribbon/RibbonLoadBalancerClientTests.java

@@ -112,7 +112,7 @@ public class RibbonLoadBalancerClientTests {
 	public void testReconstructUriWithSecureClientConfig() {
 		RibbonServer server = getRibbonServer();
 		IClientConfig config = mock(IClientConfig.class);
-		when(config.get(CommonClientConfigKey.IsSecure, false)).thenReturn(true);
+		when(config.get(CommonClientConfigKey.IsSecure)).thenReturn(true);
 		when(clientFactory.getClientConfig(server.getServiceId())).thenReturn(config);
 
 		RibbonLoadBalancerClient client = getRibbonLoadBalancerClient(server);
@@ -125,6 +125,33 @@ public class RibbonLoadBalancerClientTests {
 	}
 
 	@Test
+	@SneakyThrows
+	public void testReconstructSecureUriWithoutScheme() {
+		testReconstructSchemelessUriWithoutClientConfig(getSecureRibbonServer(), "https");
+	}
+
+	@Test
+	@SneakyThrows
+	public void testReconstructUnsecureSchemelessUri() {
+		testReconstructSchemelessUriWithoutClientConfig(getRibbonServer(), "http");
+	}
+
+	@SneakyThrows
+	public void testReconstructSchemelessUriWithoutClientConfig(RibbonServer server, String expectedScheme) {
+		IClientConfig config = mock(IClientConfig.class);
+		when(config.get(CommonClientConfigKey.IsSecure)).thenReturn(null);
+		when(clientFactory.getClientConfig(server.getServiceId())).thenReturn(config);
+
+		RibbonLoadBalancerClient client = getRibbonLoadBalancerClient(server);
+		ServiceInstance serviceInstance = client.choose(server.getServiceId());
+		URI uri = client.reconstructURI(serviceInstance,
+				new URI("//" + server.getServiceId()));
+		assertEquals(server.getHost(), uri.getHost());
+		assertEquals(server.getPort(), uri.getPort());
+		assertEquals(expectedScheme, uri.getScheme());
+	}
+
+	@Test
 	public void testChoose() {
 		RibbonServer server = getRibbonServer();
 		RibbonLoadBalancerClient client = getRibbonLoadBalancerClient(server);
@@ -206,6 +233,11 @@ public class RibbonLoadBalancerClientTests {
 				Collections.singletonMap("mykey", "myvalue"));
 	}
 
+	protected RibbonServer getSecureRibbonServer() {
+		return new RibbonServer("testService", new Server("myhost", 8443), false,
+				Collections.singletonMap("mykey", "myvalue"));
+	}
+
 	protected void verifyServerStats() {
 		verify(this.serverStats).incrementActiveRequestsCount();
 		verify(this.serverStats).decrementActiveRequestsCount();