View Javadoc
1   /*
2    * Copyright (c) 2002-2025 Gargoyle Software Inc.
3    *
4    * Licensed under the Apache License, Version 2.0 (the "License");
5    * you may not use this file except in compliance with the License.
6    * You may obtain a copy of the License at
7    * https://www.apache.org/licenses/LICENSE-2.0
8    *
9    * Unless required by applicable law or agreed to in writing, software
10   * distributed under the License is distributed on an "AS IS" BASIS,
11   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12   * See the License for the specific language governing permissions and
13   * limitations under the License.
14   */
15  package org.htmlunit;
16  
17  import java.net.URL;
18  
19  import org.htmlunit.html.HtmlPage;
20  import org.junit.jupiter.api.Test;
21  
22  /**
23   * Tests for {@link NiceRefreshHandler}.
24   *
25   * @author Marc Guillemot
26   * @author Ronald Brill
27   */
28  public final class NiceRefreshHandlerTest extends SimpleWebTestCase {
29  
30      /**
31       * @throws Exception if the test fails
32       */
33      @Test
34      public void refreshImmediatelyForDelaySmallerThanMax() throws Exception {
35          doTest(2, URL_THIRD);
36      }
37  
38      /**
39       * @throws Exception if the test fails
40       */
41      @Test
42      public void noRefreshForDelayLargerThanMax() throws Exception {
43          doTest(1, URL_FIRST);
44      }
45  
46      private void doTest(final int handlerMaxDelay, final URL expectedUrl) throws Exception {
47          final String html = DOCTYPE_HTML
48              + "<html>\n"
49              + "<head><title>Meta Redirect Page</title>\n"
50              + "<meta http-equiv='Refresh' content='2; URL=" + URL_THIRD + "'>\n"
51              + "</head>\n"
52              + "<body>redirect page body</body>\n"
53              + "</html>";
54  
55          final WebClient client = getWebClientWithMockWebConnection();
56          final MockWebConnection conn = getMockWebConnection();
57          conn.setResponse(URL_FIRST, html);
58          conn.setDefaultResponse("");
59          client.setRefreshHandler(new NiceRefreshHandler(handlerMaxDelay));
60  
61          final HtmlPage page = client.getPage(URL_FIRST);
62          assertEquals(expectedUrl, page.getUrl());
63      }
64  }