/**
 * Copyright (C) 2013-2015 all@code-story.net
 *
 * 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 net.codestory.http;

import static java.nio.charset.StandardCharsets.*;

import java.io.*;
import java.util.*;

import net.codestory.http.annotations.*;
import net.codestory.http.payload.*;
import net.codestory.http.testhelpers.*;

import org.junit.*;

public class GetTest extends AbstractProdWebServerTest {
  @Test
  public void content_types() {
    configure(routes -> routes
        .get("/index", "Index")
        .get("/text", new Payload("text/plain", "TEXT"))
        .get("/html", new Payload("text/html", "<body>HTML</body>"))
        .get("/raw", "RAW".getBytes(UTF_8))
        .get("/json", new Person("NAME", 42))
        .get("/optionalIndex", Optional.of("Index"))
    );

    get("/index").should().haveType("text/html").contain("Index");
    get("/text").should().haveType("text/plain").contain("TEXT");
    get("/html").should().haveType("text/html").contain("HTML");
    get("/raw").should().haveType("application/octet-stream").contain("RAW");
    get("/json").should().haveType("application/json").contain("{\"name\":\"NAME\",\"age\":42}");
    get("/optionalIndex").should().haveType("text/html").contain("Index");
  }

  @Test
  public void request_params() {
    configure(routes -> routes
        .get("/hello/:name", (context, name) -> "Hello " + name)
        .get("/say/:what/how/:loud", (context, what, loud) -> what + " " + loud)
        .get("/:one/:two/:three", (context, one, two, three) -> one + " " + two + " " + three)
    );

    get("/hello/Dave").should().contain("Hello Dave");
    get("/hello/John%20Doe").should().contain("Hello John Doe");
    get("/say/HI/how/LOUD").should().contain("HI LOUD");
    get("/ONE/TWO/THREE").should().contain("ONE TWO THREE");
  }

  @Test
  public void query_params() {
    configure(routes -> routes
        .get("/index", "Hello")
        .get("/hello?name=:name", (context, name) -> "Hello " + name)
        .add(new Object() {
          @Get("/keyValues")
          public String keyValues(Map<String, String> keyValues) {
            return keyValues.toString();
          }
        })
        .add(new Object() {
          @Get("/queryBoolean")
          public boolean query(Query query) {
            return query.getBoolean("name");
          }
        })
        .add(new Object() {
          @Get("/queryInt")
          public int query(Query query) {
            return query.getInteger("name");
          }
        })
    );

    get("/index?query=useless").should().contain("Hello");
    get("/hello?name=Dave").should().contain("Hello Dave");
    get("/keyValues?key1=value1&key2=value2").should().contain("key2=value2");
    get("/queryBoolean?name=true").should().contain("true");
    get("/queryInt?name=42").should().contain("42");
  }

  @Test
  public void io_streams() {
    configure(routes -> routes
        .get("/", () -> new Payload("text/html", new ByteArrayInputStream("Hello".getBytes())))
    );

    get("/").should().haveType("text/html").contain("Hello");
  }

  static class Person {
    String name;
    int age;

    Person(String name, int age) {
      this.name = name;
      this.age = age;
    }
  }
}