ClientUtilsTest.java

/*
 * Copyright (C) 2015 Square, Inc.
 *
 * 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 keywhiz.client;

import com.google.common.collect.ImmutableList;
import com.google.common.io.Resources;
import java.io.File;
import java.io.IOException;
import java.net.CookieManager;
import java.net.CookieStore;
import java.net.HttpCookie;
import java.nio.file.Files;
import java.nio.file.NoSuchFileException;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.nio.file.attribute.PosixFilePermission;
import java.util.List;
import keywhiz.cli.ClientUtils;
import keywhiz.cli.configs.CliConfiguration;
import okhttp3.OkHttpClient;
import org.junit.Before;
import org.junit.Rule;
import org.junit.Test;
import org.junit.rules.TemporaryFolder;
import org.mockito.Mock;
import org.mockito.junit.MockitoJUnit;
import org.mockito.junit.MockitoRule;

import static org.assertj.core.api.Assertions.assertThat;
import static org.mockito.Mockito.when;

public class ClientUtilsTest {
  @Rule public MockitoRule mockito = MockitoJUnit.rule();
  @Rule public TemporaryFolder tempFolder = new TemporaryFolder();

  @Mock CookieManager cookieManager;
  @Mock CookieStore cookieStore;
  @Mock CliConfiguration config;

  private HttpCookie xsrfCookie = new HttpCookie("XSRF-TOKEN", "xsrf-contents");
  {
    xsrfCookie.setPath("/");
    xsrfCookie.setDomain("localhost");
    xsrfCookie.setVersion(1);
    xsrfCookie.setHttpOnly(false);
    xsrfCookie.setSecure(true);
  }

  private HttpCookie sessionCookie = new HttpCookie("session", "session-contents");
  {
    sessionCookie.setPath("/admin");
    sessionCookie.setDomain("localhost");
    sessionCookie.setVersion(1);
    sessionCookie.setHttpOnly(true);
    sessionCookie.setSecure(true);
  }

  private ImmutableList<HttpCookie> cookieList = ImmutableList.of(sessionCookie, xsrfCookie);

  private Path cookiePath;

  @Before public void setup() throws IOException {
    tempFolder.create();
    cookiePath = Paths.get(tempFolder.getRoot().getPath(), "/.keywhiz.cookies");
    ClientUtils.getCookieManager().getCookieStore().removeAll();
  }

  @Test public void testSslOkHttpClientCreation() throws Exception {
    OkHttpClient sslClient = ClientUtils.sslOkHttpClient(config.getDevTrustStore(),
        ImmutableList.of());

    assertThat(sslClient.followSslRedirects()).isFalse();
    assertThat(sslClient.sslSocketFactory()).isNotNull();
    assertThat(sslClient.networkInterceptors()).isNotEmpty();

    assertThat(sslClient.cookieJar()).isNotNull();
    java.util.List<HttpCookie> cookieList = ClientUtils.getCookieManager().getCookieStore().getCookies();
    assertThat(cookieList).isEmpty();
  }

  @Test public void testSslOkHttpClientCreationWithCookies() throws Exception {
    OkHttpClient sslClient = ClientUtils.sslOkHttpClient(config.getDevTrustStore(), cookieList);

    assertThat(sslClient.followSslRedirects()).isFalse();
    assertThat(sslClient.cookieJar()).isNotNull();
    assertThat(sslClient.sslSocketFactory()).isNotNull();
    assertThat(sslClient.networkInterceptors()).isNotEmpty();

    java.util.List<HttpCookie> cookieList = ClientUtils.getCookieManager().getCookieStore().getCookies();
    assertThat(cookieList).contains(xsrfCookie);
    assertThat(cookieList).contains(sessionCookie);
  }

  @Test public void testSaveCookies() throws Exception {
    when(cookieManager.getCookieStore()).thenReturn(cookieStore);
    when(cookieStore.getCookies()).thenReturn(ImmutableList.of(xsrfCookie, sessionCookie));

    ClientUtils.saveCookies(cookieManager, cookiePath);

    File cookieFile = cookiePath.toFile();

    assertThat(cookieFile.exists()).isTrue();
    assertThat(Files.getPosixFilePermissions(cookieFile.toPath())).containsOnly(
        PosixFilePermission.OWNER_READ, PosixFilePermission.OWNER_WRITE);
  }

  @Test(expected = NoSuchFileException.class)
  public void testLoadCookiesFailWithoutFile() throws Exception {
    ClientUtils.loadCookies(cookiePath);
  }


  @Test public void testLoadCookiesWithFile() throws Exception {
    Path savedCookies = Paths.get(Resources.getResource("fixtures/cookies.json").getPath());
    List<HttpCookie> loadedCookies = ClientUtils.loadCookies(savedCookies);

    assertThat(loadedCookies).hasSameElementsAs(cookieList);
  }

  @Test public void testSaveAndLoadCookies() throws Exception {
    when(cookieManager.getCookieStore()).thenReturn(cookieStore);
    when(cookieStore.getCookies()).thenReturn(ImmutableList.of(xsrfCookie, sessionCookie));
    ClientUtils.saveCookies(cookieManager, cookiePath);

    assertThat(ClientUtils.loadCookies(cookiePath)).hasSameElementsAs(cookieList);
  }
}