mirror of
https://github.com/google/nomulus.git
synced 2025-05-02 21:17:50 +02:00
aae1e42da8
Previously we had a few customized Gradle build task to manage the Docker container for provisioning browser and ChromeDriverService used by WebDriver tests. This CL changed to use a java library from testcontainers.org to achieve the same purpose. The main benefit of it is that we can expect to run the WebDriver tests from IDE going forward. Also, this CL refactored the structure of WebDriver related classes to have JUnit rule to manage the lifecycle of WebDriver instance, this is also compatible with the API from testcontainers library. ------------- Created by MOE: https://github.com/google/moe MOE_MIGRATED_REVID=241539861
70 lines
2.4 KiB
Java
70 lines
2.4 KiB
Java
// Copyright 2017 The Nomulus Authors. All Rights Reserved.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package google.registry.util;
|
|
|
|
import static com.google.common.base.Throwables.throwIfUnchecked;
|
|
import static java.util.concurrent.Executors.newCachedThreadPool;
|
|
|
|
import com.google.common.util.concurrent.SimpleTimeLimiter;
|
|
import java.io.IOException;
|
|
import java.net.HttpURLConnection;
|
|
import java.net.URL;
|
|
import java.util.concurrent.TimeUnit;
|
|
|
|
/** An utility to probe a given url until it becomes available. */
|
|
public final class UrlChecker {
|
|
|
|
private static final int READ_TIMEOUT_MS = 1000;
|
|
private static final int CONNECT_TIMEOUT_MS = 500;
|
|
|
|
/** Probes {@code url} until it becomes available. */
|
|
public static void waitUntilAvailable(final URL url, int timeoutMs) {
|
|
try {
|
|
Void unusedReturnValue = SimpleTimeLimiter.create(newCachedThreadPool())
|
|
.callWithTimeout(
|
|
() -> {
|
|
int exponentialBackoffMs = 1;
|
|
while (true) {
|
|
if (isAvailable(url)) {
|
|
return null;
|
|
}
|
|
Thread.sleep(exponentialBackoffMs *= 2);
|
|
}
|
|
},
|
|
timeoutMs,
|
|
TimeUnit.MILLISECONDS);
|
|
} catch (Exception e) {
|
|
throwIfUnchecked(e);
|
|
throw new RuntimeException(e);
|
|
}
|
|
}
|
|
|
|
/** Returns {@code true} if page is available and returns {@code 200 OK}. */
|
|
static boolean isAvailable(URL url) throws IOException {
|
|
HttpURLConnection connection = (HttpURLConnection) url.openConnection();
|
|
connection.setConnectTimeout(CONNECT_TIMEOUT_MS);
|
|
connection.setReadTimeout(READ_TIMEOUT_MS);
|
|
try {
|
|
connection.connect();
|
|
return connection.getResponseCode() == HttpURLConnection.HTTP_OK;
|
|
} catch (IOException e) {
|
|
return false;
|
|
} finally {
|
|
connection.disconnect();
|
|
}
|
|
}
|
|
|
|
private UrlChecker() {}
|
|
}
|