/*
    * Copyright (c) 2002-2025 Gargoyle Software Inc.
    *
    * Licensed under the Apache License, Version 2.0 (the "License");
    * you may not use this file except in compliance with the License.
    * You may obtain a copy of the License at
    * https://www.apache.org/licenses/LICENSE-2.0
    *
    * Unless required by applicable law or agreed to in writing, software
   * distributed under the License is distributed on an "AS IS" BASIS,
   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   * See the License for the specific language governing permissions and
   * limitations under the License.
   */
  package org.htmlunit;
  
  import static org.junit.jupiter.api.Assertions.fail;
  
  import java.io.IOException;
  import java.io.Writer;
  import java.net.SocketTimeoutException;
  import java.util.HashMap;
  import java.util.Map;
  
  import javax.servlet.Servlet;
  import javax.servlet.http.HttpServlet;
  import javax.servlet.http.HttpServletRequest;
  import javax.servlet.http.HttpServletResponse;
  
  import org.htmlunit.html.HtmlPage;
  import org.htmlunit.util.Cookie;
  import org.htmlunit.util.MimeType;
  import org.junit.jupiter.api.Test;
  
  /**
   * Tests for {@link WebClient} that run with BrowserRunner.
   *
   * @author Ahmed Ashour
   * @author Ronald Brill
   */
  public class WebClient4Test extends WebServerTestCase {
  
      /**
       * Verifies that a WebClient can be serialized and deserialized after it has been used.
       * @throws Exception if an error occurs
       */
      @Test
      public void serialization_afterUse() throws Exception {
          startWebServer("./");
  
          try (WebClient client = getWebClient()) {
              TextPage textPage = client.getPage(URL_FIRST + "LICENSE.txt");
              assertTrue(textPage.getContent().contains("Apache License"));
  
              try (WebClient copy = clone(client)) {
                  assertNotNull(copy);
  
                  final WebWindow window = copy.getCurrentWindow();
                  assertNotNull(window);
  
                  final WebWindow topWindow = window.getTopWindow();
                  assertNotNull(topWindow);
  
                  final Page page = topWindow.getEnclosedPage();
                  assertNotNull(page);
  
                  final WebResponse response = page.getWebResponse();
                  assertNotNull(response);
  
                  final String content = response.getContentAsString();
                  assertNotNull(content);
                  assertTrue(content.contains("Apache License"));
  
                  textPage = copy.getPage(URL_FIRST + "LICENSE.txt");
                  assertTrue(textPage.getContent().contains("Apache License"));
              }
          }
      }
  
      /**
       * Verifies that a redirect limit kicks in even if the redirects aren't for the same URL
       * and don't use the same redirect HTTP status code (see bug 2915453).
       * @throws Exception if an error occurs
       */
      @Test
      public void redirectInfinite303And307() throws Exception {
          final Map<String, Class<? extends Servlet>> servlets = new HashMap<>();
          servlets.put(RedirectServlet307.URL, RedirectServlet307.class);
          servlets.put(RedirectServlet303.URL, RedirectServlet303.class);
          startWebServer("./", new String[0], servlets);
  
          final WebClient client = getWebClient();
  
          try {
              client.getPage("http://localhost:" + PORT + RedirectServlet307.URL);
          }
          catch (final Exception e) {
              assertTrue(e.getMessage(), e.getMessage().contains("Too much redirect"));
          }
     }
 
     /**
      * Helper class for {@link #redirectInfinite303And307}.
      */
     public static class RedirectServlet extends HttpServlet {
         private int count_;
         private int status_;
         private String location_;
 
         /**
          * Creates a new instance.
          * @param status the HTTP status to return
          * @param location the location to redirect to
          */
         public RedirectServlet(final int status, final String location) {
             count_ = 0;
             status_ = status;
             location_ = location;
         }
 
         /**
          * {@inheritDoc}
          */
         @Override
         protected void doGet(final HttpServletRequest req, final HttpServletResponse resp) throws IOException {
             count_++;
             resp.setStatus(status_);
             resp.setHeader("Location", location_);
             resp.getWriter().write(status_ + " " + req.getContextPath() + " " + count_);
         }
     }
 
     /**
      * Helper class for {@link #redirectInfinite303And307}.
      */
     public static class RedirectServlet303 extends RedirectServlet {
         static final String URL = "/test";
         /** Creates a new instance. */
         public RedirectServlet303() {
             super(303, RedirectServlet307.URL);
         }
     }
 
     /**
      * Helper class for {@link #redirectInfinite303And307}.
      */
     public static class RedirectServlet307 extends RedirectServlet {
         static final String URL = "/test2";
         /** Creates a new instance. */
         public RedirectServlet307() {
             super(307, RedirectServlet303.URL);
         }
     }
 
     /**
      * Regression test for bug 2903223: body download time was not taken into account
      * in {@link WebResponse#getLoadTime()}.
      * @throws Exception if an error occurs
      */
     @Test
     public void bodyDowloadTime() throws Exception {
         final Map<String, Class<? extends Servlet>> servlets = new HashMap<>();
         servlets.put("/*", ServeBodySlowlyServlet.class);
         startWebServer("./", new String[0], servlets);
 
         final Page page = getWebClient().getPage(URL_FIRST);
         final long loadTime = page.getWebResponse().getLoadTime();
         assertTrue("Load time: " + loadTime + ", last request time: " + ServeBodySlowlyServlet.LastRequestTime_,
                 loadTime >= ServeBodySlowlyServlet.LastRequestTime_);
     }
 
     /**
      * Helper class for {@link #bodyDowloadTime}.
      */
     public static class ServeBodySlowlyServlet extends HttpServlet {
         private static volatile long LastRequestTime_ = -1;
 
         @Override
         protected void doGet(final HttpServletRequest req, final HttpServletResponse resp) throws IOException {
             final long before = System.currentTimeMillis();
             final Writer writer = resp.getWriter();
             writeSomeContent(writer); // some content quickly
             writer.flush();
             try {
                 Thread.sleep(500);
             }
             catch (final InterruptedException e) {
                 throw new RuntimeException(e);
             }
             writeSomeContent(writer); // and some content later
             LastRequestTime_ = System.currentTimeMillis() - before;
         }
 
         private static void writeSomeContent(final Writer writer) throws IOException {
             for (int i = 0; i < 1000; i++) {
                 writer.append((char) ('a' + (i % 26)));
             }
         }
     }
 
     /**
      * @throws Exception if the test fails
      */
     @Test
     public void useProxy() throws Exception {
         final Map<String, Class<? extends Servlet>> servlets = new HashMap<>();
         servlets.put("/test", UseProxyHeaderServlet.class);
         startWebServer("./", null, servlets);
 
         final WebClient client = getWebClient();
         final HtmlPage page = client.getPage(URL_FIRST + "test");
         assertEquals("Going anywhere?", page.asNormalizedText());
     }
 
     /**
      * Servlet for {@link #useProxy()}.
      */
     public static class UseProxyHeaderServlet extends HttpServlet {
         /**
          * {@inheritDoc}
          */
         @Override
         protected void doGet(final HttpServletRequest request, final HttpServletResponse response) throws IOException {
             response.setStatus(HttpServletResponse.SC_USE_PROXY);
             //Won't matter!
             //response.setHeader("Location", "http://www.google.com");
             response.setContentType(MimeType.TEXT_HTML);
             final Writer writer = response.getWriter();
             writer.write(DOCTYPE_HTML + "<html><body>Going anywhere?</body></html>");
         }
     }
 
     /**
      * Regression test for bug 2803378: GET or POST to a URL that returns HTTP 204 (No Content).
      * @throws Exception if an error occurs
      */
     @Test
     public void noContent() throws Exception {
         final Map<String, Class<? extends Servlet>> servlets = new HashMap<>();
         servlets.put("/test1", NoContentServlet1.class);
         servlets.put("/test2", NoContentServlet2.class);
         startWebServer("./", null, servlets);
         final WebClient client = getWebClient();
         final HtmlPage page = client.getPage(URL_FIRST + "test1");
         final HtmlPage page2 = page.getHtmlElementById("submit").click();
         assertEquals(page, page2);
     }
 
     /**
      * First servlet for {@link #noContent()}.
      */
     public static class NoContentServlet1 extends HttpServlet {
         /**
          * {@inheritDoc}
          */
         @Override
         protected void doGet(final HttpServletRequest req, final HttpServletResponse res) throws IOException {
             res.setContentType(MimeType.TEXT_HTML);
             final Writer writer = res.getWriter();
             writer.write(DOCTYPE_HTML
                     + "<html><body><form action='test2'>\n"
                     + "<input id='submit' type='submit' value='submit'></input>\n"
                     + "</form></body></html>");
         }
     }
 
     /**
      * Second servlet for {@link #noContent()}.
      */
     public static class NoContentServlet2 extends HttpServlet {
         /**
          * {@inheritDoc}
          */
         @Override
         protected void doGet(final HttpServletRequest req, final HttpServletResponse res) {
             res.setStatus(HttpServletResponse.SC_NO_CONTENT);
         }
     }
 
     /**
      * Regression test for bug 2821888: HTTP 304 (Not Modified) was being treated as a redirect. Note that a 304
      * response doesn't really make sense because we're not sending any If-Modified-Since headers, but we want to
      * at least make sure that we're not throwing exceptions when we receive one of these responses.
      * @throws Exception if an error occurs
      */
     @Test
     public void notModified() throws Exception {
         final Map<String, Class<? extends Servlet>> servlets = new HashMap<>();
         servlets.put("/test", NotModifiedServlet.class);
         startWebServer("./", null, servlets);
         final WebClient client = getWebClient();
         final HtmlPage page = client.getPage(URL_FIRST + "test");
         final TextPage page2 = client.getPage(URL_FIRST + "test");
         assertNotNull(page);
         assertNotNull(page2);
     }
 
     /**
      * Servlet for {@link #notModified()}.
      */
     public static class NotModifiedServlet extends HttpServlet {
         private boolean first_ = true;
 
         /**
          * {@inheritDoc}
          */
         @Override
         protected void doGet(final HttpServletRequest req, final HttpServletResponse res) throws IOException {
             if (first_) {
                 first_ = false;
                 res.setContentType(MimeType.TEXT_HTML);
                 final Writer writer = res.getWriter();
                 writer.write(DOCTYPE_HTML + "<html><body>foo</body></html>");
             }
             else {
                 res.setStatus(HttpServletResponse.SC_NOT_MODIFIED);
             }
         }
     }
 
     /**
      * @throws Exception if an error occurs
      */
     @Test
     public void timeout() throws Exception {
         final Map<String, Class<? extends Servlet>> servlets = new HashMap<>();
         servlets.put("/*", DelayDeliverServlet.class);
         startWebServer("./", null, servlets);
 
         final WebClient client = getWebClient();
         client.getOptions().setTimeout(500);
 
         try {
             client.getPage(URL_FIRST);
             fail("timeout expected!");
         }
         catch (final SocketTimeoutException e) {
             // as expected
         }
 
         // now configure higher timeout allowing to get the page
         client.getOptions().setTimeout(5000);
         client.getPage(URL_FIRST);
     }
 
     /**
      * Make sure cookies set for the request are overwriting the cookieManager.
      *
      * @throws Exception if something goes wrong
      */
     @Test
     public void requestHeaderCookieFromRequest() throws Exception {
         final String content = DOCTYPE_HTML + "<html></html>";
         final MockWebConnection webConnection = new MockWebConnection();
         webConnection.setDefaultResponse(content);
 
         startWebServer(webConnection);
 
         final WebClient client = getWebClient();
 
         // no cookie sent
         client.getPage(URL_FIRST);
         assertNull(webConnection.getLastAdditionalHeaders().get(HttpHeader.COOKIE));
 
         // process web request with cookie
         WebRequest wr = new WebRequest(URL_FIRST, HttpMethod.GET);
         wr.setAdditionalHeader(HttpHeader.COOKIE, "yummy_cookie=choco");
         client.getPage(wr);
         assertEquals("yummy_cookie=choco", webConnection.getLastAdditionalHeaders().get(HttpHeader.COOKIE));
 
         // add cookie to the cookie manager and test if sent
         final CookieManager mgr = client.getCookieManager();
         mgr.addCookie(new Cookie(URL_FIRST.getHost(), "my_key", "my_value", "/", null, false));
         wr = new WebRequest(URL_FIRST, HttpMethod.GET);
         client.getPage(wr);
         assertEquals("my_key=my_value", webConnection.getLastAdditionalHeaders().get(HttpHeader.COOKIE));
 
         // request page again, now the the request provides his own cookies
         wr = new WebRequest(URL_FIRST, HttpMethod.GET);
         wr.setAdditionalHeader(HttpHeader.COOKIE, "yummy_cookie=choco");
         client.getPage(wr);
         assertEquals("yummy_cookie=choco", webConnection.getLastAdditionalHeaders().get(HttpHeader.COOKIE));
 
         // request page again, now the the request provides his own cookies as part of a map
         wr = new WebRequest(URL_FIRST, HttpMethod.GET);
         final Map<String, String> headers = new HashMap<>();
         headers.put("accept-language", "es-ES,es;q=0.9");
         headers.put(HttpHeader.COOKIE, "tasty_cookie=strawberry");
         wr.setAdditionalHeaders(headers);
         client.getPage(wr);
         assertEquals("tasty_cookie=strawberry", webConnection.getLastAdditionalHeaders().get(HttpHeader.COOKIE));
 
         // and finally to make sure the cookies from the store are still there
         wr = new WebRequest(URL_FIRST, HttpMethod.GET);
         client.getPage(wr);
         assertEquals("my_key=my_value", webConnection.getLastAdditionalHeaders().get(HttpHeader.COOKIE));
     }
 
     /**
      * Servlet for {@link #timeout()}.
      */
     public static class DelayDeliverServlet extends HttpServlet {
         /**
          * {@inheritDoc}
          */
         @Override
         protected void doGet(final HttpServletRequest req, final HttpServletResponse res) throws IOException {
             try {
                 Thread.sleep(1000);
             }
             catch (final InterruptedException e) {
                 throw new RuntimeException(e);
             }
             res.setContentType(MimeType.TEXT_HTML);
             final Writer writer = res.getWriter();
             writer.write(DOCTYPE_HTML + "<html><head><title>hello</title></head><body>foo</body></html>");
         }
     }
 
     /**
      * @throws Exception if an error occurs
      */
     @Test
     public void redirectInfiniteMeta() throws Exception {
         final Map<String, Class<? extends Servlet>> servlets = new HashMap<>();
         servlets.put("/test1", RedirectMetaServlet1.class);
         servlets.put("/test2", RedirectMetaServlet2.class);
         startWebServer("./", new String[0], servlets);
 
         final WebClient client = getWebClient();
 
         try {
             client.getPage(URL_FIRST + "test1");
         }
         catch (final Exception e) {
             assertTrue(e.getMessage(), e.getMessage().contains("Too much redirect"));
         }
     }
 
     /**
      * Servlet for {@link #redirectInfiniteMeta()}.
      */
     public static class RedirectMetaServlet1 extends HttpServlet {
         /**
          * {@inheritDoc}
          */
         @Override
         protected void doGet(final HttpServletRequest req, final HttpServletResponse res) throws IOException {
             res.setContentType(MimeType.TEXT_HTML);
             final Writer writer = res.getWriter();
             writer.write(DOCTYPE_HTML
                     + "<html><head>\n"
                     + "  <meta http-equiv='refresh' content='0;URL=test2'>\n"
                     + "</head><body>foo</body></html>");
         }
     }
 
     /**
      * Servlet for {@link #redirectInfiniteMeta()}.
      */
     public static class RedirectMetaServlet2 extends HttpServlet {
         /**
          * {@inheritDoc}
          */
         @Override
         protected void doGet(final HttpServletRequest req, final HttpServletResponse res) throws IOException {
             res.setContentType(MimeType.TEXT_HTML);
             final Writer writer = res.getWriter();
             writer.write(DOCTYPE_HTML
                     + "<html><head>\n"
                     + "  <meta http-equiv='refresh' content='0;URL=test1'>\n"
                     + "</head><body>foo</body></html>");
         }
     }
 }