// Copyright 2011 Google Inc. All Rights Reserved.
//
// 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
//
//      http://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 com.google.enterprise.adaptor;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNull;
import static org.junit.Assert.assertTrue;

import org.junit.Test;

import java.util.Date;
import java.util.Locale;

/**
 * Test cases for {@link AbstractHandler}.
 */
public class HttpExchangesTest {
  private MockHttpExchange ex = new MockHttpExchange("GET", "/",
      new MockHttpContext("/"));

  @Test
  public void testRequestUriHttp10() {
    ex = new MockHttpExchange("GET", "/test",
                              new MockHttpContext("/"));
    assertEquals("http://localhost/test",
        HttpExchanges.getRequestUri(ex).toString());
  }

  @Test
  public void testRequestUriHttp11() {
    ex.getRequestHeaders().set("Host", "abc");
    assertEquals("http://abc/", HttpExchanges.getRequestUri(ex).toString());
  }

  @Test
  public void testEnableCompression() throws Exception {
    ex.getRequestHeaders().set("Accept-Encoding", "gzip,deflate");
    HttpExchanges.enableCompressionIfSupported(ex);
    assertEquals("gzip", ex.getResponseHeaders().getFirst("Content-Encoding"));
    ex.sendResponseHeaders(200, 0);
    int bytesToWrite = 1000;
    ex.getResponseBody().write(new byte[bytesToWrite]);
    ex.getResponseBody().close();
    assertTrue(ex.getResponseBytes().length < bytesToWrite);
  }

  @Test
  public void testNotEnableCompression() throws Exception {
    HttpExchanges.enableCompressionIfSupported(ex);
    assertNull(ex.getResponseHeaders().getFirst("Content-Encoding"));
  }

  @Test
  public void testNotEnableCompressionUnsupported() throws Exception {
    ex.getRequestHeaders().set("Accept-Encoding", "deflate");
    HttpExchanges.enableCompressionIfSupported(ex);
    assertNull(ex.getResponseHeaders().getFirst("Content-Encoding"));
  }

  @Test
  public void testCannedHead() throws Exception {
    ex = new MockHttpExchange("HEAD", "/test",
                              new MockHttpContext("/"));
    // Translation.HTTP_NOT_FOUND was randomly chosen.
    HttpExchanges.cannedRespond(ex, 200, Translation.HTTP_NOT_FOUND);
    assertEquals(0, ex.getResponseBytes().length);
  }

  @Test
  public void testIfModifiedSince() throws Exception {
    final Date golden = new Date(784111777L * 1000);
    ex.getRequestHeaders().set("If-Modified-Since",
                               "Sun, 06 Nov 1994 08:49:37 GMT");
    assertEquals(golden, HttpExchanges.getIfModifiedSince(ex));

    ex.getRequestHeaders().set("If-Modified-Since",
                               "Sunday, 06-Nov-94 08:49:37 GMT");
    assertEquals(golden, HttpExchanges.getIfModifiedSince(ex));

    ex.getRequestHeaders().set("If-Modified-Since",
                               "Sun Nov  6 08:49:37 1994");
    assertEquals(golden, HttpExchanges.getIfModifiedSince(ex));

    ex.getRequestHeaders().set("If-Modified-Since",
                               "Sun, 06 Nov 1994 08:49:37 GMT");
    // Java 7 added categories for Locales. The no-argument get() is DISPLAY.
    // TODO(ejona): use reflection to override CATEGORY as well for Java 7.
    Locale defaultLocalDisplay = Locale.getDefault();
    Locale.setDefault(Locale.GERMANY);
    HttpExchanges.resetThread();
    Date date;
    try {
      date = HttpExchanges.getIfModifiedSince(ex);
    } finally {
      Locale.setDefault(defaultLocalDisplay);
      HttpExchanges.resetThread();
    }
    assertEquals(golden, date);
  }

  @Test
  public void testIfModifiedSinceEmpty() throws Exception {
    assertNull(HttpExchanges.getIfModifiedSince(ex));
  }

  @Test
  public void testIfModifiedSinceInvalid() throws Exception {
    ex.getRequestHeaders().set("If-Modified-Since",
                               "Thu, 1 Jan 1970 00:0001 GMT");
    assertNull(HttpExchanges.getIfModifiedSince(ex));
  }

  @Test
  public void testSetLastModified() throws Exception {
    String golden = "Sun, 06 Nov 1994 08:49:38 GMT";
    Date date = new Date(784111778L  * 1000);
    HttpExchanges.setLastModified(ex, date);
    assertEquals(golden, ex.getResponseHeaders().getFirst("Last-Modified"));

    // Java 7 added categories for Locales. The no-argument get() is DISPLAY.
    // TODO(ejona): use reflection to override CATEGORY as well for Java 7.
    Locale defaultLocalDisplay = Locale.getDefault();
    Locale.setDefault(Locale.GERMANY);
    HttpExchanges.resetThread();
    try {
      HttpExchanges.setLastModified(ex, date);
    } finally {
      Locale.setDefault(defaultLocalDisplay);
      HttpExchanges.resetThread();
    }
    assertEquals(golden, ex.getResponseHeaders().getFirst("Last-Modified"));
  }
}