Skip to content

Commit

Permalink
Merge branch '2.1.x' into 2.2.x
Browse files Browse the repository at this point in the history
Closes gh-19112
  • Loading branch information
snicoll committed Nov 25, 2019
2 parents 36f4d2f + d386ee0 commit a4ca1d0
Show file tree
Hide file tree
Showing 2 changed files with 35 additions and 8 deletions.
Expand Up @@ -189,7 +189,7 @@ public void setAddresses(String addresses) {
private List<Address> parseAddresses(String addresses) {
List<Address> parsedAddresses = new ArrayList<>();
for (String address : StringUtils.commaDelimitedListToStringArray(addresses)) {
parsedAddresses.add(new Address(address));
parsedAddresses.add(new Address(address, getSsl().isEnabled()));
}
return parsedAddresses;
}
Expand Down Expand Up @@ -390,7 +390,7 @@ public boolean determineEnabled() {
return isEnabled();
}
Address address = RabbitProperties.this.parsedAddresses.get(0);
return address.secureConnection;
return address.determineSslEnabled(isEnabled());
}

public void setEnabled(boolean enabled) {
Expand Down Expand Up @@ -989,14 +989,14 @@ private static final class Address {

private String virtualHost;

private boolean secureConnection;
private Boolean secureConnection;

private Address(String input) {
private Address(String input, boolean sslEnabled) {
input = input.trim();
input = trimPrefix(input);
input = parseUsernameAndPassword(input);
input = parseVirtualHost(input);
parseHostAndPort(input);
parseHostAndPort(input, sslEnabled);
}

private String trimPrefix(String input) {
Expand All @@ -1005,7 +1005,8 @@ private String trimPrefix(String input) {
return input.substring(PREFIX_AMQP_SECURE.length());
}
if (input.startsWith(PREFIX_AMQP)) {
input = input.substring(PREFIX_AMQP.length());
this.secureConnection = false;
return input.substring(PREFIX_AMQP.length());
}
return input;
}
Expand Down Expand Up @@ -1036,18 +1037,22 @@ private String parseVirtualHost(String input) {
return input;
}

private void parseHostAndPort(String input) {
private void parseHostAndPort(String input, boolean sslEnabled) {
int portIndex = input.indexOf(':');
if (portIndex == -1) {
this.host = input;
this.port = (this.secureConnection) ? DEFAULT_PORT_SECURE : DEFAULT_PORT;
this.port = (determineSslEnabled(sslEnabled)) ? DEFAULT_PORT_SECURE : DEFAULT_PORT;
}
else {
this.host = input.substring(0, portIndex);
this.port = Integer.valueOf(input.substring(portIndex + 1));
}
}

private boolean determineSslEnabled(boolean sslEnabled) {
return (this.secureConnection != null) ? this.secureConnection : sslEnabled;
}

}

}
Expand Up @@ -102,6 +102,14 @@ void determinePortUsingAmqpsReturnsPortOfFirstAddress() {
assertThat(this.properties.determinePort()).isEqualTo(5671);
}

@Test
void determinePortReturnsDefaultAmqpsPortWhenFirstAddressHasNoExplicitPortButSslEnabled() {
this.properties.getSsl().setEnabled(true);
this.properties.setPort(1234);
this.properties.setAddresses("rabbit1.example.com,rabbit2.example.com:2345");
assertThat(this.properties.determinePort()).isEqualTo(5671);
}

@Test
void virtualHostDefaultsToNull() {
assertThat(this.properties.getVirtualHost()).isNull();
Expand Down Expand Up @@ -241,6 +249,20 @@ void determineSslUsingAmqpsReturnsStateOfFirstAddress() {
assertThat(this.properties.getSsl().determineEnabled()).isTrue();
}

@Test
void sslDetermineEnabledIsTrueWhenAddressHasNoProtocolAndSslIsEnabled() {
this.properties.getSsl().setEnabled(true);
this.properties.setAddresses("root:password@otherhost");
assertThat(this.properties.getSsl().determineEnabled()).isTrue();
}

@Test
void sslDetermineEnabledIsFalseWhenAddressHasNoProtocolAndSslIsDisabled() {
this.properties.getSsl().setEnabled(false);
this.properties.setAddresses("root:password@otherhost");
assertThat(this.properties.getSsl().determineEnabled()).isFalse();
}

@Test
void determineSslUsingAmqpReturnsStateOfFirstAddress() {
this.properties.setAddresses("amqp://root:password@otherhost,amqps://root:password2@otherhost2");
Expand Down

0 comments on commit a4ca1d0

Please sign in to comment.