/*
* 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);
}
}