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

时间:2017-02-17 13:09:14

标签: java http tomcat ssl jenkins

描述

我做了一个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;
}

2 个答案:

答案 0 :(得分:2)

我有两个解决这个问题的方法:

手动选择SSL端口

ServerSocket(0) constructor会自动选择一个空闲端口。 Tomcat也使用这种方法。

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连接器具有生命周期识别功能,因此您将收到关于&_ 39; before_init'和' after_init'事件。当您将连接器添加到服务时,Tomcat会按顺序初始化连接器。

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

        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();
    

答案 1 :(得分:1)

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

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

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

  3. 所以我认为你可以尝试添加像

    这样的东西
    mTomcat.start(); // <-- your existing code
    defaultConnector.setRedirectPort(SSLConnector.getLocalPort())