Log a warning when MarksDB password isn't configured

-------------
Created by MOE: https://github.com/google/moe
MOE_MIGRATED_REVID=231287550
This commit is contained in:
mcilwain 2019-01-28 14:17:10 -08:00 committed by Michael Muller
parent 44e822ed86
commit 88aa13d1a4
3 changed files with 17 additions and 8 deletions

View file

@ -18,6 +18,7 @@ import static com.google.common.base.Verify.verifyNotNull;
import static google.registry.util.UrlFetchUtils.setAuthorizationHeader;
import com.google.appengine.api.urlfetch.HTTPRequest;
import com.google.common.flogger.FluentLogger;
import google.registry.keyring.api.KeyModule.Key;
import google.registry.model.registry.Registry;
import java.util.Optional;
@ -26,8 +27,14 @@ import javax.inject.Inject;
/** Helper class for setting the authorization header on a MarksDB LORDN request. */
final class LordnRequestInitializer {
@Inject @Key("marksdbLordnPassword") Optional<String> marksdbLordnPassword;
@Inject LordnRequestInitializer() {}
private static final FluentLogger logger = FluentLogger.forEnclosingClass();
private final Optional<String> marksdbLordnPassword;
@Inject
LordnRequestInitializer(@Key("marksdbLordnPassword") Optional<String> marksdbLordnPassword) {
this.marksdbLordnPassword = marksdbLordnPassword;
}
/** Initializes a URL fetch request for talking to the MarksDB server. */
void initialize(HTTPRequest request, String tld) {
@ -41,6 +48,8 @@ final class LordnRequestInitializer {
"lordnUsername is not set for %s.", Registry.get(tld).getTld());
return Optional.of(String.format("%s:%s", lordnUsername, marksdbLordnPassword.get()));
} else {
logger.atWarning().log(
"Cannot set credentials for LORDN request because password isn't configured.");
return Optional.empty();
}
}

View file

@ -104,7 +104,8 @@ public class NordnUploadActionTest {
@Captor private ArgumentCaptor<HTTPRequest> httpRequestCaptor;
private final FakeClock clock = new FakeClock(DateTime.parse("2010-05-01T10:11:12Z"));
private final LordnRequestInitializer lordnRequestInitializer = new LordnRequestInitializer();
private final LordnRequestInitializer lordnRequestInitializer =
new LordnRequestInitializer(Optional.of("attack"));
private final NordnUploadAction action = new NordnUploadAction();
@Before
@ -117,7 +118,6 @@ public class NordnUploadActionTest {
persistResource(loadRegistrar("TheRegistrar").asBuilder().setIanaIdentifier(99999L).build());
createTld("tld");
persistResource(Registry.get("tld").asBuilder().setLordnUsername("lolcat").build());
lordnRequestInitializer.marksdbLordnPassword = Optional.of("attack");
action.clock = clock;
action.fetchService = fetchService;
action.lordnRequestInitializer = lordnRequestInitializer;
@ -210,7 +210,7 @@ public class NordnUploadActionTest {
@Test
public void testRun_noPassword_doesntSendAuthorizationHeader() throws Exception {
lordnRequestInitializer.marksdbLordnPassword = Optional.empty();
action.lordnRequestInitializer = new LordnRequestInitializer(Optional.empty());
persistClaimsModeDomain();
action.run();
assertThat(getHeaderFirst(getCapturedHttpRequest(), AUTHORIZATION)).isEmpty();

View file

@ -87,7 +87,8 @@ public class NordnVerifyActionTest {
@Captor private ArgumentCaptor<HTTPRequest> httpRequestCaptor;
private final FakeResponse response = new FakeResponse();
private final LordnRequestInitializer lordnRequestInitializer = new LordnRequestInitializer();
private final LordnRequestInitializer lordnRequestInitializer =
new LordnRequestInitializer(Optional.of("attack"));
private final NordnVerifyAction action = new NordnVerifyAction();
@Before
@ -97,7 +98,6 @@ public class NordnVerifyActionTest {
when(fetchService.fetch(any(HTTPRequest.class))).thenReturn(httpResponse);
createTld("gtld");
persistResource(Registry.get("gtld").asBuilder().setLordnUsername("lolcat").build());
lordnRequestInitializer.marksdbLordnPassword = Optional.of("attack");
action.tld = "gtld";
action.fetchService = fetchService;
action.lordnRequestInitializer = lordnRequestInitializer;
@ -125,7 +125,7 @@ public class NordnVerifyActionTest {
@Test
public void testSuccess_noLordnPassword_doesntSetAuthorizationHeader() throws Exception {
lordnRequestInitializer.marksdbLordnPassword = Optional.empty();
action.lordnRequestInitializer = new LordnRequestInitializer(Optional.empty());
action.run();
assertThat(getHeaderFirst(getCapturedHttpRequest(), AUTHORIZATION)).isEmpty();
}