package android.webkit;

import static com.google.common.truth.Truth.assertThat;

import androidx.test.ext.junit.runners.AndroidJUnit4;
import org.junit.After;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.robolectric.annotation.internal.DoNotInstrument;

/** Compatibility test for {@link CookieManager} */
@DoNotInstrument
@RunWith(AndroidJUnit4.class)
public class CookieManagerTest {

  @After
  public void tearDown() {
    CookieManager.getInstance().removeAllCookie();
  }

  @Test
  public void setCookie_doesNotAllowMultipleCookies() {
    final String httpsUrl = "https://robolectric.org/";
    final CookieManager cookieManager = CookieManager.getInstance();
    cookieManager.setCookie(httpsUrl, "A=100; B=200");
    String cookie = cookieManager.getCookie(httpsUrl);
    assertThat(cookie).isEqualTo("A=100");
  }

  @Test
  public void setCookie_multipleCookies() {
    final String httpsUrl = "https://robolectric.org/";
    final CookieManager cookieManager = CookieManager.getInstance();
    cookieManager.setCookie(httpsUrl, "A=100;");
    cookieManager.setCookie(httpsUrl, "B=100;");
    String cookie = cookieManager.getCookie(httpsUrl);
    assertThat(cookie).isEqualTo("A=100; B=100");
  }

  @Test
  public void setCookie_overrideCookieHasTheSameKey() {
    final String httpsUrl = "https://robolectric.org/";
    final CookieManager cookieManager = CookieManager.getInstance();
    cookieManager.setCookie(httpsUrl, "A=100;");
    cookieManager.setCookie(httpsUrl, "A=200;");
    String cookie = cookieManager.getCookie(httpsUrl);
    assertThat(cookie).isEqualTo("A=200");
  }

  @Test
  public void getCookie_doesNotReturnAttributes() {
    final String httpsUrl = "https://robolectric.org/";
    final CookieManager cookieManager = CookieManager.getInstance();
    cookieManager.setCookie(httpsUrl, "ID=test-id; Path=/; Domain=.robolectric.org");
    String cookie = cookieManager.getCookie(httpsUrl);
    assertThat(cookie).isEqualTo("ID=test-id");
  }

  @Test
  public void shouldSetAndGetCookieWithWhitespacesInUrlParameters() {
    CookieManager cookieManager = CookieManager.getInstance();
    String url = "http://www.google.com/?q=This is a test query";
    String value = "my cookie";
    cookieManager.setCookie(url, value);
    assertThat(cookieManager.getCookie(url)).isEqualTo(value);
  }

  @Test
  public void shouldSetAndGetCookieWithEncodedWhitespacesInUrlParameters() {
    CookieManager cookieManager = CookieManager.getInstance();
    String url = "http://www.google.com/?q=This%20is%20a%20test%20query";
    String value = "my cookie";
    cookieManager.setCookie(url, value);
    assertThat(cookieManager.getCookie(url)).isEqualTo(value);
  }
}