/*
 * Copyright 2002-2019 the original author or authors.
 *
 * 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.springframework.http;

import java.util.Arrays;

import org.hamcrest.Matchers;
import org.junit.Test;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertThat;

/**
 * Unit tests for {@link ResponseCookie}.
 * @author Rossen Stoyanchev
 */
public class ResponseCookieTests {

	@Test
	public void basic() {

		assertEquals("id=", ResponseCookie.from("id", null).build().toString());
		assertEquals("id=1fWa", ResponseCookie.from("id", "1fWa").build().toString());

		assertEquals(
				"id=1fWa; Path=/path; Domain=abc; " +
						"Max-Age=0; Expires=Thu, 01 Jan 1970 00:00:00 GMT; " +
						"Secure; HttpOnly; SameSite=None",
				ResponseCookie.from("id", "1fWa")
						.domain("abc").path("/path").maxAge(0).httpOnly(true).secure(true).sameSite("None")
						.build().toString());
	}

	@Test
	public void nameChecks() {

		Arrays.asList("id", "i.d.", "i-d", "+id", "i*d", "i$d", "#id")
				.forEach(name -> {
					ResponseCookie.from(name, "value").build();
					// no exception..
				});

		Arrays.asList("\"id\"", "id\t", "i\td", "i d", "i;d", "{id}", "[id]", "\"", "id\u0091")
				.forEach(name -> {
					try {
						ResponseCookie.from(name, "value").build();
					}
					catch (IllegalArgumentException ex) {
						assertThat(ex.getMessage(), Matchers.containsString("RFC2616 token"));
					}
				});
	}

	@Test
	public void valueChecks() {

		Arrays.asList("1fWa", "", null, "1f=Wa", "1f-Wa", "1f/Wa", "1.f.W.a.")
				.forEach(value -> {
					ResponseCookie.from("id", value).build();
					// no exception..
				});

		Arrays.asList("1f\tWa", "\t", "1f Wa", "1f;Wa", "\"1fWa", "1f\\Wa", "1f\"Wa", "\"", "1fWa\u0005", "1f\u0091Wa")
				.forEach(value -> {
					try {
						ResponseCookie.from("id", value).build();
					}
					catch (IllegalArgumentException ex) {
						assertThat(ex.getMessage(), Matchers.containsString("RFC2616 cookie value"));
					}
				});
	}

	@Test
	public void domainChecks() {

		Arrays.asList("abc", "abc.org", "abc-def.org", "abc3.org", ".abc.org")
				.forEach(domain -> ResponseCookie.from("n", "v").domain(domain).build());

		Arrays.asList("-abc.org", "abc.org.", "abc.org-", "-abc.org", "abc.org-")
				.forEach(domain -> {
					try {
						ResponseCookie.from("n", "v").domain(domain).build();
					}
					catch (IllegalArgumentException ex) {
						assertThat(ex.getMessage(), Matchers.containsString("Invalid first/last char"));
					}
				});

		Arrays.asList("abc..org", "abc.-org", "abc-.org")
				.forEach(domain -> {
					try {
						ResponseCookie.from("n", "v").domain(domain).build();
					}
					catch (IllegalArgumentException ex) {
						assertThat(ex.getMessage(), Matchers.containsString("invalid cookie domain char"));
					}
				});
	}

}