/*
    * 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.util;
  
  import static java.nio.charset.StandardCharsets.UTF_8;
  
  import java.io.ByteArrayOutputStream;
  import java.util.ArrayList;
  import java.util.Arrays;
  import java.util.Collections;
  import java.util.List;
  import java.util.zip.GZIPOutputStream;
  
  import org.apache.commons.io.FileUtils;
  import org.apache.commons.io.IOUtils;
  import org.htmlunit.MockWebConnection;
  import org.htmlunit.SimpleWebTestCase;
  import org.htmlunit.WebRequest;
  import org.htmlunit.WebResponse;
  import org.junit.Test;
  
  /**
   * Tests for {@link DebuggingWebConnection}.
   *
   * @author Marc Guillemot
   */
  public class DebuggingWebConnectionTest extends SimpleWebTestCase {
  
      /**
       * @throws Exception if the test fails
       */
      @Test
      public void nameValueListToJsMap() throws Exception {
          assertEquals("{}", DebuggingWebConnection.nameValueListToJsMap(null));
          assertEquals("{}", DebuggingWebConnection.nameValueListToJsMap(Collections.emptyList()));
  
          List<NameValuePair> list = Collections.singletonList(new NameValuePair("name", "value"));
          assertEquals("{'name': 'value'}", DebuggingWebConnection.nameValueListToJsMap(list));
  
          list = Collections.singletonList(new NameValuePair("na me", "value"));
          assertEquals("{'na me': 'value'}", DebuggingWebConnection.nameValueListToJsMap(list));
  
          list = new ArrayList<>();
          list.add(new NameValuePair("na me", "value1"));
          list.add(new NameValuePair("key", "value 2"));
          list.add(new NameValuePair("key 2", "value 3"));
          list.add(new NameValuePair("key 4", "with ' quote")); // can it really happen in header?
          final String expected = "{'na me': 'value1', 'key': 'value 2', 'key 2': 'value 3', 'key 4': 'with \\' quote'}";
          assertEquals(expected, DebuggingWebConnection.nameValueListToJsMap(list));
      }
  
      /**
       * @throws Exception if the test fails
       */
      @Test
      public void escapeJSString() throws Exception {
          assertEquals("", DebuggingWebConnection.escapeJSString(""));
          assertEquals("hello", DebuggingWebConnection.escapeJSString("hello"));
          assertEquals("I\\'m here", DebuggingWebConnection.escapeJSString("I'm here"));
      }
  
      /**
       * @throws Exception if the test fails
       */
      @Test
      public void chooseExtension() throws Exception {
          assertEquals(".html", DebuggingWebConnection.chooseExtension(MimeType.TEXT_HTML));
  
          assertEquals(".js", DebuggingWebConnection.chooseExtension("text/javascript"));
  
          assertEquals(".css", DebuggingWebConnection.chooseExtension(MimeType.TEXT_CSS));
  
          assertEquals(".xml", DebuggingWebConnection.chooseExtension(MimeType.TEXT_XML));
  
          assertEquals(".txt", DebuggingWebConnection.chooseExtension(MimeType.TEXT_PLAIN));
      }
  
      /**
       * Ensures that Content-Encoding headers are removed when JavaScript is uncompressed.
       * (was causing java.io.IOException: Not in GZIP format).
       * @throws Exception if the test fails
       */
      @Test
      public void gzip() throws Exception {
          final ByteArrayOutputStream baos = new ByteArrayOutputStream();
          try (GZIPOutputStream gzipOutputStream = new GZIPOutputStream(baos)) {
              IOUtils.write("alert(1)", gzipOutputStream, UTF_8);
         }
 
         final MockWebConnection mockConnection = new MockWebConnection();
         final List<NameValuePair> responseHeaders = Arrays.asList(
             new NameValuePair("Content-Encoding", "gzip"));
         mockConnection.setResponse(URL_FIRST, baos.toByteArray(), 200, "OK", MimeType.TEXT_JAVASCRIPT,
             responseHeaders);
 
         final String dirName = "test-" + getClass().getSimpleName();
         try (DebuggingWebConnection dwc = new DebuggingWebConnection(mockConnection, dirName)) {
 
             final WebRequest request = new WebRequest(URL_FIRST);
             final WebResponse response = dwc.getResponse(request); // was throwing here
             assertNull(response.getResponseHeaderValue("Content-Encoding"));
 
             FileUtils.deleteDirectory(dwc.getReportFolder());
         }
     }
 }