mirror of
https://github.com/google/nomulus.git
synced 2025-05-07 23:38:21 +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
84 lines
2.9 KiB
Java
84 lines
2.9 KiB
Java
// Copyright 2019 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.webdriver;
|
|
|
|
import static com.google.common.base.Preconditions.checkNotNull;
|
|
|
|
import google.registry.util.UrlChecker;
|
|
import java.net.MalformedURLException;
|
|
import java.net.URL;
|
|
import java.time.Duration;
|
|
import java.time.temporal.ChronoUnit;
|
|
import org.junit.rules.ExternalResource;
|
|
import org.openqa.selenium.WebDriver;
|
|
import org.openqa.selenium.chrome.ChromeOptions;
|
|
import org.openqa.selenium.remote.RemoteWebDriver;
|
|
import org.testcontainers.containers.BindMode;
|
|
import org.testcontainers.containers.GenericContainer;
|
|
import org.testcontainers.containers.wait.strategy.Wait;
|
|
|
|
/** JUnit rule for managing Docker container used by WebDriver tests. */
|
|
class DockerWebDriverRule extends ExternalResource {
|
|
|
|
// This port number is defined in this Dockerfile:
|
|
// https://github.com/SeleniumHQ/docker-selenium/blob/master/StandaloneChrome/Dockerfile#L21
|
|
private static final int CHROME_DRIVER_SERVICE_PORT = 4444;
|
|
private static final URL WEB_DRIVER_URL = getWebDriverUrl();
|
|
private WebDriver webDriver;
|
|
|
|
private static URL getWebDriverUrl() {
|
|
GenericContainer container =
|
|
new GenericContainer("selenium/standalone-chrome")
|
|
.withFileSystemBind("/dev/shm", "/dev/shm", BindMode.READ_WRITE)
|
|
.withExposedPorts(CHROME_DRIVER_SERVICE_PORT)
|
|
.waitingFor(Wait.forHttp("/").withStartupTimeout(Duration.of(20, ChronoUnit.SECONDS)));
|
|
container.start();
|
|
URL url;
|
|
try {
|
|
url =
|
|
new URL(
|
|
String.format(
|
|
"http://localhost:%d/wd/hub",
|
|
container.getMappedPort(CHROME_DRIVER_SERVICE_PORT)));
|
|
} catch (MalformedURLException e) {
|
|
throw new IllegalArgumentException(e);
|
|
}
|
|
UrlChecker.waitUntilAvailable(url, 20000);
|
|
return url;
|
|
}
|
|
|
|
@Override
|
|
protected void before() {
|
|
ChromeOptions chromeOptions = new ChromeOptions().setHeadless(true);
|
|
webDriver = new RemoteWebDriver(WEB_DRIVER_URL, chromeOptions);
|
|
}
|
|
|
|
@Override
|
|
protected void after() {
|
|
try {
|
|
webDriver.quit();
|
|
} finally {
|
|
webDriver = null;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Returns {@link WebDriver} instance connected to the {@link
|
|
* org.openqa.selenium.chrome.ChromeDriverService} running in the container.
|
|
*/
|
|
public WebDriver getWebDriver() {
|
|
return checkNotNull(webDriver);
|
|
}
|
|
}
|