使用嵌入式 tomcat 服务器进行 JUnit 测试,如何为 http 和 https 连接器指定自动端口?

2022-09-04 23:16:01

描述

我做了一个 JUnit 测试,重点是尝试测试对 SOAP Web 服务的调用。

我正在使用嵌入式tomcat服务器进行测试,以便使用模拟服务器运行我的测试。

我还同时使用http和https连接器。

我需要为这两个连接器使用自动端口,因为测试是在 Jenkins 服务器上运行的,我不能只使用端口 443 或 8443,因为它们已经被占用了。

我知道使用端口0作为标准端口将导致tomcat使用自动端口分配,但我无法设法将其与两个连接器一起使用。

预期行为

我想对我的自定义 SSL 连接器也使用自动端口分配。

是否有可能以某种方式这样做?

示例代码

这是我的tomcat实例的代码:

@Before
public void setup() throws Throwable {

    File tomcatWorkingDir = new File(mWorkingDir);

    //Empty the target/tomcat-working-dir directory if it exist
    //Create the directory otherwise
    if(tomcatWorkingDir.exists() && tomcatWorkingDir.isDirectory()){
        LOGGER.info("cleaning tomcat-working-dir directory");
        FileUtils.cleanDirectory(new File(mWorkingDir)); 
    } else {
        LOGGER.info("create tomcat-working-dir directory");
        tomcatWorkingDir.mkdir();
    }

    LOGGER.info("disabling ssl certification validation");
    //Disable JVM ssl sockets connection
    disableJVMCertificate();

    //Add server certificate
    createServerCertificate();

    //Custom SSL Connector
    Connector SSLConnector = getSSLConnector();

    mTomcat = new Tomcat();

    //Standard http startup port
    mTomcat.setPort(0);

    //Set up base directory 
    //Otherwise, tomcat would use the current directory
    mTomcat.setBaseDir(mWorkingDir);

    LOGGER.info("setting the ssl connector in TOMCAT");
    Service service = mTomcat.getService();
    service.addConnector(SSLConnector);

    //Redirect current port
    Connector defaultConnector = mTomcat.getConnector();
    defaultConnector.setRedirectPort(SERVER_HTTPS_PORT);

    //Configure the way WAR are managed by the engine
    mTomcat.getHost().setAutoDeploy(true);
    mTomcat.getHost().setDeployOnStartup(true);

    //Add mock server into our webApp
    String servletName = "/server";
    File webApp = new File(mWorkingDir,"../../../ws-mock-server/src/main/webapp");

    mTomcat.addWebapp(mTomcat.getHost(), servletName, webApp.getAbsolutePath());

    //start tomcat
    LOGGER.info("starting TOMCAT");

    mTomcat.start();
  }

和这里为我的自定义 ssl 连接器。

    private static Connector getSSLConnector(){
    Connector connector = new Connector();
    connector.setPort(SERVER_HTTPS_PORT);
    connector.setSecure(true);

    //Http protocol Http11AprProtocol
    connector.setAttribute("protocol", "org.apache.coyote.http11.Http11AprProtocol");

    //Maximum threads allowedd on this instance of tomcat
    connector.setAttribute("maxThreads","200");
    connector.setAttribute("SSLEnabled", true);

    //No client Authentification is required in order to connect
    connector.setAttribute("clientAuth", false);

    //SSL TLSv1 protocol
    connector.setAttribute("sslProtocol","TLS");

    //Ciphers configuration describing how server will encrypt his messages
    //A common cipher suite need to exist between server and client in an ssl
    //communication in order for the handshake to succeed
    connector.setAttribute("ciphers","TLS_ECDHE_RSA_WITH_AES_128_CBC_SHA");

    LOGGER.info("setting keystore file");
    //Here an absolute file path is needed in order to properly set up the keystore attribute
    connector.setAttribute("keystoreFile",new File(".").getAbsolutePath().replace("\\", "/")+"/"+mWorkingDir+"/server.jks");

    LOGGER.info("setting keystore pass");
    connector.setAttribute("keystorePass","changeit");

    return connector;
}

答案 1

对于这个问题,我有两种解决方案:

手动选择 SSL 端口

ServerSocket(0) 构造函数自动选择一个空闲端口。雄猫也使用这种方法。

try (ServerSocket testSocket = new ServerSocket(0)) {
    int randomFreePort = testSocket.getLocalPort(); 
    sslConnector.setPort(randomFreePort);
    defaultConnector.setRedirectPort( randomFreePort);
} // At this point the testSocket.close() called
tomcat.start();

我知道,有一种可能性,另一个进程在 和 之间分配相同的端口,但你可以通过测试来检测这种情况。testSocket.close()tomcat.start()LifecycleState.FAILED.equals(sslConnector.getState())

使用生命周期侦听器

Tomcat 连接器具有生命周期感知能力,因此您将收到有关“before_init”和“after_init”事件的通知。Tomcat 会按照您添加到服务中的顺序初始化连接器。

  1. 添加 ssl 连接器。
  2. 添加 http 连接器。(这将是“默认”连接器。不要调用 ,因为它获取第一个连接器或创建新的连接器。)mTomcat.getConnector()
  3. SSL 连接器初始化完成后,可以通过 getLocalPort() 调用获取所选端口。
  4. 在 http 连接器初始化之前,调用 setRedirectPort

完整示例:

    Tomcat mTomcat = new Tomcat();
    Connector sslConnector = getSSLConnector(); 
    mTomcat.getService().addConnector(sslConnector);    
    Connector defaultConnector = new Connector();
    defaultConnector.setPort(0);
    mTomcat.getService().addConnector(defaultConnector);

    // Do the rest of the Tomcat setup

    AtomicInteger sslPort = new AtomicInteger();
    sslConnector.addLifecycleListener(event->{
        if( "after_init".equals(event.getType()) )
            sslPort.set(sslConnector.getLocalPort());
    });
    defaultConnector.addLifecycleListener(event->{
        if( "before_init".equals(event.getType()) )
            defaultConnector.setRedirectPort(sslPort.get());
    });

    mTomcat.start();

答案 2

我还没有尝试过,但从代码来看,它看起来像

  1. 您可以在服务器启动后setRedirectPort

  2. 您可以使用 来获取实际端口Connector.getLocalPort

所以我认为你可以尝试添加类似的东西

mTomcat.start(); // <-- your existing code
defaultConnector.setRedirectPort(SSLConnector.getLocalPort())

推荐