引用: HttpURLConnection timeout question

-> 关于如何自动化上述单元测试用例的任何想法？

更具体地说，如果 HTTP 客户端将其超时设置为 5 秒，我希望服务器在 10 秒后发送响应。这将确保我的客户端会因超时而失败，从而自动执行此场景。

我会很感激服务器端的伪代码(任何轻量级的 http 服务器，例如 jetty 或任何其他服务器都可以)。

最佳答案

您不想在单元 测试中实际连接到真实服务器。如果您想实际连接到真实服务器，从技术上讲，这就是集成测试。

由于您正在测试客户端代码，因此您应该使用单元测试，这样您就不需要连接到真实的服务器。相反，您可以使用模拟对象来模拟与服务器的连接。这真的很棒，因为您可以模拟使用真实服务器时难以实现的情况(例如在 session 过程中连接失败等)。

使用 mock 进行单元测试还可以使测试运行得更快，因为您不需要连接任何东西，因此没有 I/O 延迟。

由于您链接到另一个问题，我将使用该代码示例(为清楚起见在此处重新粘贴)我创建了一个名为 MyClass 的类，其中包含一个方法 foo()到 URL，如果连接成功则返回 true 或 false。正如链接的问题一样:

public class MyClass {

private String url = "http://example.com";

public boolean foo(){
    try {
           HttpURLConnection.setFollowRedirects(false);
           HttpURLConnection con = (HttpURLConnection) new URL(url).openConnection();
           con.setRequestMethod("HEAD");

           con.setConnectTimeout(5000); //set timeout to 5 seconds

           return (con.getResponseCode() == HttpURLConnection.HTTP_OK);
        } catch (java.net.SocketTimeoutException e) {
           return false;
        } catch (java.io.IOException e) {
           return false;
        }

    }
}

我将使用 Mockito制作模拟对象，因为这是更流行的模拟对象库之一。此外，由于代码在 foo 方法中创建了一个新的 URL 对象(这不是最佳设计)，我将使用 PowerMock可以拦截对 new 的调用的库。在实际的生产代码中，我建议使用依赖注入(inject)或至少方法提取来为工厂方法创建 URL 对象，这样您就可以覆盖它以简化测试。但由于我与您的示例保持一致，因此我不会更改任何内容。

下面是使用 Mockito 和 Powermock 测试超时的测试代码:

import java.net.HttpURLConnection;
import java.net.SocketTimeoutException;    
import java.net.URL;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Mockito;
import org.powermock.api.mockito.PowerMockito;
import org.powermock.core.classloader.annotations.PrepareForTest;
import org.powermock.modules.junit4.PowerMockRunner;

import static org.junit.Assert.*;

@RunWith(PowerMockRunner.class)
//This tells powermock that we will modify MyClass.class in this test 
//- needed for changing the call to new URL
@PrepareForTest(MyClass.class) 
public class ConnectionTimeOutTest {

String url = "http://example.com";
@Test
public void timeout() throws Exception{
    //create a mock URL and mock HttpURLConnection objects
    //that will be our simulated server
    URL mockURL = PowerMockito.mock(URL.class);
    HttpURLConnection mockConnection = PowerMockito.mock(HttpURLConnection.class);

    //powermock will intercept our call to new URL( url) 
    //and return our mockURL object instead!
    PowerMockito.whenNew(URL.class).withArguments(url).thenReturn(mockURL);
    //This tells our mockURL class to return our mockConnection object when our client
    //calls the open connection method
    PowerMockito.when(mockURL.openConnection()).thenReturn(mockConnection);



    //this is our exception to throw to simulate a timeout
    SocketTimeoutException expectedException = new SocketTimeoutException();

    //tells our mockConnection to throw the timeout exception instead of returnig a response code
    PowerMockito.when(mockConnection.getResponseCode()).thenThrow(expectedException);

    //now we are ready to actually call the client code
    // cut = Class Under Test
    MyClass cut = new MyClass();

    //our code should catch the timeoutexception and return false
    assertFalse(cut.foo());

   // tells mockito to expect the given void methods calls
   //this will fail the test if the method wasn't called with these arguments
   //(for example, if you set the timeout to a different value)
    Mockito.verify(mockConnection).setRequestMethod("HEAD");
    Mockito.verify(mockConnection).setConnectTimeout(5000);

}
}

此测试运行时间不到一秒，这比实际等待超过 5 秒才能真正超时快得多!

关于java - 为 HTTP 客户端请求模拟 HTTP 服务器超时，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/25495773/