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.htmlunit.junit.BrowserRunner;
21  import org.junit.Test;
22  import org.junit.runner.RunWith;
23  
24  /**
25   * Tests for {@link NiceRefreshHandler}.
26   *
27   * @author Marc Guillemot
28   */
29  @RunWith(BrowserRunner.class)
30  public final class NiceRefreshHandlerTest extends SimpleWebTestCase {
31  
32      /**
33       * @throws Exception if the test fails
34       */
35      @Test
36      public void refreshImmediatelyForDelaySmallerThanMax() throws Exception {
37          doTest(2, URL_THIRD);
38      }
39  
40      /**
41       * @throws Exception if the test fails
42       */
43      @Test
44      public void noRefreshForDelayLargerThanMax() throws Exception {
45          doTest(1, URL_FIRST);
46      }
47  
48      private void doTest(final int handlerMaxDelay, final URL expectedUrl) throws Exception {
49          final String html = DOCTYPE_HTML
50              + "<html>\n"
51              + "<head><title>Meta Redirect Page</title>\n"
52              + "<meta http-equiv='Refresh' content='2; URL=" + URL_THIRD + "'>\n"
53              + "</head>\n"
54              + "<body>redirect page body</body>\n"
55              + "</html>";
56  
57          final WebClient client = getWebClientWithMockWebConnection();
58          final MockWebConnection conn = getMockWebConnection();
59          conn.setResponse(URL_FIRST, html);
60          conn.setDefaultResponse("");
61          client.setRefreshHandler(new NiceRefreshHandler(handlerMaxDelay));
62  
63          final HtmlPage page = client.getPage(URL_FIRST);
64          assertEquals(expectedUrl, page.getUrl());
65      }
66  }