Class IntegrationTest

s2a/src/test/java/io/grpc/s2a/IntegrationTest.java:62–256

import org.junit.runners.JUnit4;

@RunWith(JUnit4.class)
public final class IntegrationTest {
  private static final Logger logger = Logger.getLogger(FakeS2AServer.class.getName());
  private String s2aAddress;
  private Server s2aServer;
  private String s2aDelayAddress;
  private Server s2aDelayServer;
  private String mtlsS2AAddress;
  private Server mtlsS2AServer;
  private String serverAddress;
  private Server server;

  @Before
  public void setUp() throws Exception {
    s2aServer = ServerBuilder.forPort(0).addService(new FakeS2AServer()).build().start();
    int s2aPort = s2aServer.getPort();
    s2aAddress = "localhost:" + s2aPort;
    logger.info("S2A service listening on localhost:" + s2aPort);
    ClassLoader classLoader = IntegrationTest.class.getClassLoader();
    InputStream s2aCert = classLoader.getResourceAsStream("server_cert.pem");
    InputStream s2aKey = classLoader.getResourceAsStream("server_key.pem");
    InputStream rootCert = classLoader.getResourceAsStream("root_cert.pem");
    ServerCredentials s2aCreds =
        TlsServerCredentials.newBuilder()
            .keyManager(s2aCert, s2aKey)
            .trustManager(rootCert)
            .clientAuth(TlsServerCredentials.ClientAuth.REQUIRE)
            .build();
    mtlsS2AServer = NettyServerBuilder.forPort(0, s2aCreds).addService(new FakeS2AServer()).build();
    mtlsS2AServer.start();
    int mtlsS2APort = mtlsS2AServer.getPort();
    mtlsS2AAddress = "localhost:" + mtlsS2APort;
    logger.info("mTLS S2A service listening on localhost:" + mtlsS2APort);

    int s2aDelayPort = Utils.pickUnusedPort();
    s2aDelayAddress = "localhost:" + s2aDelayPort;
    s2aDelayServer = ServerBuilder.forPort(s2aDelayPort).addService(new FakeS2AServer()).build();

    server =
        NettyServerBuilder.forPort(0)
            .addService(new SimpleServiceImpl())
            .sslContext(buildSslContext())
            .build()
            .start();
    int serverPort = server.getPort();
    serverAddress = "localhost:" + serverPort;
    logger.info("Simple Service listening on localhost:" + serverPort);
  }

  @After
  public void tearDown() throws Exception {
    server.shutdown();
    s2aServer.shutdown();
    s2aDelayServer.shutdown();
    mtlsS2AServer.shutdown();

    server.awaitTermination(10, SECONDS);
    s2aServer.awaitTermination(10, SECONDS);