/*
    * Licensed to the Apache Software Foundation (ASF) under one or more
    * contributor license agreements.  See the NOTICE file distributed with
    * this work for additional information regarding copyright ownership.
    * The ASF licenses this file to You 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 org.apache.juneau.rest;
  
  import org.apache.juneau.*;
  import org.apache.juneau.http.annotation.*;
  import org.apache.juneau.plaintext.*;
  import org.apache.juneau.rest.annotation.*;
  import org.apache.juneau.rest.mock.*;
  import org.apache.juneau.utest.utils.*;
  import org.junit.jupiter.api.*;
  
  class Header_AcceptCharset_Test extends TestBase {
  
  	//------------------------------------------------------------------------------------------------------------------
  	// Test that Q-values are being resolved correctly.
  	//------------------------------------------------------------------------------------------------------------------
  
  	@Rest(defaultCharset="utf-8",serializers=PlainTextSerializer.class)
  	public static class A {
  		@RestGet
  		public String a() {
  			return "foo";
  		}
  	}
  
  	@Test void a01_qValues() throws Exception {
  		var a = MockRestClient.build(A.class);
  		a.get("/a").accept("text/plain").acceptCharset("utf-8").run().assertCharset().is("utf-8");
  		a.get("/a").accept("text/plain").acceptCharset("iso-8859-1").run().assertCharset().is("ISO-8859-1");
  		a.get("/a").accept("text/plain").acceptCharset("bad,utf-8").run().assertCharset().is("utf-8");
  		a.get("/a").accept("text/plain").acceptCharset("utf-8,bad").run().assertCharset().is("utf-8");
  		a.get("/a").accept("text/plain").acceptCharset("bad;q=0.9,utf-8;q=0.1").run().assertCharset().is("utf-8");
  		a.get("/a").accept("text/plain").acceptCharset("bad;q=0.1,utf-8;q=0.9").run().assertCharset().is("utf-8");
  		a.get("/a").accept("text/plain").acceptCharset("utf-8;q=0.9,iso-8859-1;q=0.1").run().assertCharset().is("utf-8");
  		a.get("/a").accept("text/plain").acceptCharset("utf-8;q=0.1,iso-8859-1;q=0.9").run().assertCharset().is("ISO-8859-1");
  		a.get("/a").accept("text/plain").acceptCharset("*").run().assertCharset().is("utf-8");
  		a.get("/a").accept("text/plain").acceptCharset("bad,iso-8859-1;q=0.5,*;q=0.1").run().assertCharset().is("ISO-8859-1");
  		a.get("/a").accept("text/plain").acceptCharset("bad,iso-8859-1;q=0.1,*;q=0.5").run().assertCharset().is("utf-8");
  	}
  
  	//------------------------------------------------------------------------------------------------------------------
  	// Validate various Accept-Charset variations.
  	//------------------------------------------------------------------------------------------------------------------
  
  	@Rest(defaultCharset="utf-8")
  	public static class B {
  
  		@RestPut(parsers=TestParser.class, serializers=TestSerializer.class)
  		public String a(@Content String in) {
  			return in;
  		}
  
  		public static class TestParser extends FakeReaderParser {
  			public TestParser(FakeReaderParser.Builder builder) {
  				super(builder.consumes("text/plain").function((session,in,type) -> session.getStreamCharset().toString()));
  			}
  		}
  
  		public static class TestSerializer extends FakeWriterSerializer {
  			public TestSerializer(FakeWriterSerializer.Builder builder) {
  				super(builder.produces("text/plain").function((session,o) -> (o.toString() + "/" + session.getStreamCharset())));
  			}
  		}
  	}
  
  	@Test void b01_charsetOnResponse() throws Exception {
  		var b = MockRestClient.buildLax(B.class);
  		b.put("/a", null).plainText().run().assertContent("UTF-8/UTF-8");
  		b.put("/a", null).plainText().acceptCharset("Shift_JIS").run().assertContent("UTF-8/Shift_JIS");
  		b.put("/a?noTrace=true", null).plainText().acceptCharset("BAD").run().assertStatus(406).assertContent().isContains("No supported charsets in header 'Accept-Charset': 'BAD'");
  		b.put("/a", null).plainText().acceptCharset("UTF-8").run().assertContent("UTF-8/UTF-8");
  		b.put("/a", null).plainText().acceptCharset("bad,iso-8859-1").run().assertContent("UTF-8/ISO-8859-1");
  		b.put("/a", null).plainText().acceptCharset("bad;q=0.9,iso-8859-1;q=0.1").run().assertContent("UTF-8/ISO-8859-1");
  		b.put("/a", null).plainText().acceptCharset("bad;q=0.1,iso-8859-1;q=0.9").run().assertContent("UTF-8/ISO-8859-1");
  	}
  }