/*
    * The MIT License (MIT)
    *
    * Copyright (c) 2013 - 2023, Tapio Rautonen
    *
    * Permission is hereby granted, free of charge, to any person obtaining a copy
    * of this software and associated documentation files (the "Software"), to deal
    * in the Software without restriction, including without limitation the rights
    * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
   * copies of the Software, and to permit persons to whom the Software is
   * furnished to do so, subject to the following conditions:
   *
   * The above copyright notice and this permission notice shall be included in
   * all copies or substantial portions of the Software.
   *
   * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
   * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
   * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
   * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
   * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
   * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
   * THE SOFTWARE.
   */
  package org.eluder.coveralls.maven.plugin.httpclient;
  
  import com.github.tomakehurst.wiremock.junit.WireMockRule;
  import org.apache.http.client.HttpClient;
  import org.apache.http.client.methods.HttpGet;
  import org.apache.http.util.EntityUtils;
  import org.apache.maven.settings.Proxy;
  import org.junit.Rule;
  import org.junit.Test;
  import org.junit.jupiter.api.Assertions;
  
  import static com.github.tomakehurst.wiremock.client.WireMock.*;
  
  public class HttpClientFactoryTest {
  
      private final int PROXY_PORT = 9797;
      private final int TARGET_PORT = 9696;
      private final String TARGET_URL = "http://localhost:" + TARGET_PORT;
  
      @Rule
      public WireMockRule targetServer = new WireMockRule(TARGET_PORT);
  
      @Rule
      public WireMockRule proxyServer = new WireMockRule(PROXY_PORT);
  
  
      @Test
      public void testSimpleRequest() throws Exception {
          targetServer.stubFor(get(urlMatching(".*")).willReturn(aResponse().withBody("Hello World!")));
  
          HttpClient client = new HttpClientFactory(TARGET_URL).create();
          String body = EntityUtils.toString(client.execute(new HttpGet(TARGET_URL)).getEntity());
  
          Assertions.assertEquals("Hello World!", body);
      }
  
      @Test
      public void testUnAuthorizedProxyRequest() throws Exception {
          targetServer.stubFor(get(urlMatching(".*")).willReturn(aResponse().withBody("Hello World!")));
  
          proxyServer.stubFor(get(urlMatching(".*")).willReturn(aResponse().withBody("Hello Proxy!")));
  
          Proxy proxy = new Proxy();
          proxy.setHost("localhost");
          proxy.setPort(PROXY_PORT);
          proxy.setProtocol("http");
  
          HttpClient client = new HttpClientFactory(TARGET_URL).proxy(proxy).create();
          String body = EntityUtils.toString(client.execute(new HttpGet(TARGET_URL)).getEntity());
  
          Assertions.assertEquals("Hello Proxy!", body);
      }
  
      @Test
      public void testAuthorixedProxyRequest() throws Exception {
          targetServer.stubFor(get(urlMatching(".*")).willReturn(aResponse().withBody("Hello World!")));
  
          proxyServer.stubFor(get(urlMatching(".*")).withHeader("Proxy-Authorization", matching("Basic Zm9vOmJhcg=="))
                  .willReturn(aResponse().withBody("Hello Proxy!"))
                  .atPriority(1));
          proxyServer.stubFor(any(urlMatching(".*"))
                  .willReturn(aResponse().withStatus(407).withHeader("Proxy-Authenticate", "Basic"))
                  .atPriority(2));
  
          Proxy proxy = new Proxy();
          proxy.setHost("localhost");
          proxy.setPort(PROXY_PORT);
          proxy.setProtocol("http");
          proxy.setUsername("foo");
          proxy.setPassword("bar");
  
          HttpClient client = new HttpClientFactory(TARGET_URL).proxy(proxy).create();
          String body = EntityUtils.toString(client.execute(new HttpGet(TARGET_URL)).getEntity());
  
          Assertions.assertEquals("Hello Proxy!", body);
      }
 
     @Test
     public void testNonProxiedHostRequest() throws Exception {
         targetServer.stubFor(get(urlMatching(".*")).willReturn(aResponse().withBody("Hello World!")));
 
         proxyServer.stubFor(get(urlMatching(".*")).willReturn(aResponse().withBody("Hello Proxy!")));
 
         Proxy proxy = new Proxy();
         proxy.setHost("localhost");
         proxy.setPort(PROXY_PORT);
         proxy.setProtocol("http");
         proxy.setNonProxyHosts("localhost|example.com");
 
         HttpClient client = new HttpClientFactory(TARGET_URL).proxy(proxy).create();
         String body = EntityUtils.toString(client.execute(new HttpGet(TARGET_URL)).getEntity());
 
         Assertions.assertEquals("Hello World!", body);
     }
 }