java - HtmlUnit 的 WebClient 在第二次相同的 loadWebResponse() 调用中失败

Question

我正在尝试为我正在为我的工作开发的 web 应用程序中的一个非常长且笨拙的“getPost”方法编写测试。我正在使用 JUnit、HtmlUnit 和 Jetty 的 ServletTester 来近似向 servlet 发送请求和接收响应。我设法让它大部分都能正常工作，但我遇到了问题。我正在尝试测试登录功能。如果用户登录成功，服务器应该将一些 JSON 与用户信息一起发送回客户端。如果用户已经登录，服务器应该返回“result”:“failure”和一条错误信息。

当我尝试测试第二个要求时，我的问题来了。我可以成功登录，并获得正确的数据。但是，当我尝试再次发送请求时，它返回 404:未找到。我尝试将代码分解为不同的测试，但我必须能够调用登录两次才能测试第二个要求。 JUnit 文件中的后续测试运行良好，同时 servlet 保持连接状态。我尝试提出第二个相同的请求，但也失败了。我在互联网上搜索无济于事。简而言之，我被难住了。

这是我正在使用的(不必要的代码已被编辑掉):

//In MyFunServlet class:
private final static String USERID_ATTRIBUTENAME = "userid";
void doPost(HttpServletRequest request, HttpServletResponse response) {
    String action = request.getParameter("opt");
    final HttpSession session = request.getSession();

    if(action != null){
       Long userId = (Long)session.getAttribute(USERID_ATTRIBUTENAME);
       if(userId != null){
           //do stuffz
       } else {
           if(action.equals("login")) {
               User user = LoginUser(request, response);
               try{
                   JSONObject json = new JSONObject();
                   if(request.getAttribute("result") == "success"){
                       json.put("result", "success");
                       json.put("id", user.getId());
                       json.put("name", user.getName());
                   } else {
                       json.put("result", "failure");
                       json.put("message", request.getAttribute("message"));
                   }
                   SendJSONResponse(json, request, response);
               }catch(Exception e){
               }
            } else {
                System.out.print("Unknown opt: " + action);
                response.setStatus(HttpServletResponse.SC_NOT_FOUND);
            }
        }
    }
}

private void LoginUser(HttpServletRequest request, HttpServletResponse response){
    final HttpSession session = request.getSession();
    User user = null;
    Long userId = (Long)session.getAttribute(USERID_ATTRIBUTENAME);
    if(userId != null){
        request.setAttribute("result", "failure");
        request.setAttribute("message", "The user is already logged in.");
    } else {
        final String email = request.getParameter("accountEmail");
        final String password = request.getParameter("accountPassword");
        if(email != null) {
            user = helperClass.magicallyGetUserByEmail(email);
            if(user != null){
                if(user.getPassword().equals(password)){
                    session.setAttribute(USERID_ATTRIBUTENAME, user.getId();
                    request.setAttribute("result", "success");
                }
            }
        } else {
           request.setAttribute("result", "failure");
        }
    }
    return user;
}

private void SendJSONResponse(JSONObject json, HttpServletRequest request,
                                HttpServletResponse response) {
    String contentStr = json.toString();
    response.setContentType("application/json");
    response.setStatus( HttpServletResponse.SC_OK);
    response.setContentLength(contentStr.length());
    response.getWriter().print(contentStr);
    response.flushBuffer();
}

出于引用目的，此文件长 1084 行。 doPost 方法大约有 900 个。免责声明:这不是我的代码。我没有写它。我只需要测试一下。 现在进行测试:

//In MyFunServletTest.java:
//using JUnit 4

public class MyFunServletTest {
    static ServletTester tester;
    static String baseUrl;

    WebClient webClient = new WebClient();

    User user;
    WebRequest loginRequest;

    @BeforeClass
    public static void initClass(){
        tester = new ServletTester;
        tester.setContextPath("/");
        tester.addServlet(MyFunServlet.class, "/fun.service");
        baseUrl = tester.createSocketConnector(true);
        tester.start();
    }

    @AfterClass
    public static void cleanClass() {
        tester.stop();
    }

    @Before
    public void preTest(){
        //taking values from our magical test user
        user = new User();
        user.setEmail("blah@blah.com");
        user.setPassword("secure");

        loginRequest = new WebRequest(baseUrl + "/fun.service", HttpMethod.POST);
        List<NameValuePair> params = new ArrayList<NameValuePair>();
        params.add(new NameValuePair("opt","login"));
        params.add(new NameValuePair("accountEmail", user.getEmail());
        params.add(new NameValuePair("accountPassword", user.getPassword());
        loginRequest.setRequestParameters(params);

    }

    @Test
    public void testLogin() {
        WebResponse response = webClient.loadWebResponse(loginRequest);
        JSONObject responseJSON = new JSONObject(response.getContentAsString());

        //this test passes
        assertEquals("success", responseJSON.getString("result"));

        response = webClient.loadWebResponse(loginRequest);

        //this test fails
        assertTrue(404 != response.getStatusCode());

        //this then causes an error, as response.getContentAsString() is null.
        esponseJSON = new JSONObject(response.getContentAsString());
    }
}

帮助？我哪里遗漏了什么？

谢谢。

Answer 1

没有自己跑测试的能力，只能提供一些方法:

• 尝试创建两个 JSONObject 对象来分别存储两个响应，并比较两者(打印它们或使用调试器)，看看是否有任何东西看起来那里很奇怪。

• 如果这没有告诉您任何信息，请创建两个单独的相同请求实例并使用每个实例。

• 然后尝试跟踪对 loadWebResponse 的调用，以准确查看请求的 URL(提高日志级别也可能告诉您这一点)。

如果 404 是正确的，那么第二个请求会以某种方式被破坏，但问题是 WHERE。