io.cloudslang.content.httpclient.build.auth.AuthSchemeProviderLookupBuilderTest.java Source code

Introduction

Here is the source code for io.cloudslang.content.httpclient.build.auth.AuthSchemeProviderLookupBuilderTest.java

Source

/*******************************************************************************
 * (c) Copyright 2017 Hewlett-Packard Development Company, L.P.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Apache License v2.0 which accompany this distribution.
 *
 * The Apache License is available at
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 *******************************************************************************/
package io.cloudslang.content.httpclient.build.auth;

import org.apache.http.Header;
import org.apache.http.auth.AuthSchemeProvider;
import org.apache.http.client.config.AuthSchemes;
import org.apache.http.config.Lookup;
import org.apache.http.impl.auth.BasicScheme;
import org.apache.http.impl.auth.BasicSchemeFactory;
import org.apache.http.impl.auth.DigestSchemeFactory;
import org.apache.http.impl.auth.KerberosSchemeFactory;
import org.junit.Test;

import java.util.ArrayList;

import static junit.framework.Assert.assertEquals;
import static org.hamcrest.CoreMatchers.instanceOf;
import static org.junit.Assert.assertThat;

/**
 * Created with IntelliJ IDEA.
 * User: tusaa
 * Date: 7/1/14
 */
public class AuthSchemeProviderLookupBuilderTest {

    @Test
    public void buildLookupWithBasicAuth() {
        AuthSchemeProvider provider = getAuthSchemeProvider(AuthSchemes.BASIC);
        assertThat(provider, instanceOf(BasicSchemeFactory.class));
        BasicScheme basicSchema = ((BasicScheme) provider.create(null));
        assertEquals("UTF-8", basicSchema.getCredentialsCharset().toString());
    }

    @Test
    public void buildLookupWithDigestAuth() {
        AuthSchemeProvider provider = getAuthSchemeProvider(AuthSchemes.DIGEST);
        assertThat(provider, instanceOf(DigestSchemeFactory.class));
    }

    @Test
    public void buildLookupWithKerberosAuth() {
        AuthTypes authTypes = new AuthTypes(AuthSchemes.KERBEROS);
        AuthSchemeProvider provider = new AuthSchemeProviderLookupBuilder().setAuthTypes(authTypes)
                .setHost("myweb.contoso.com").buildAuthSchemeProviderLookup().lookup(AuthSchemes.KERBEROS);
        assertThat(provider, instanceOf(KerberosSchemeFactory.class));
    }

    @Test
    public void buildLookupWithNtlmAuth() {
        AuthSchemeProvider provider = getAuthSchemeProvider(AuthSchemes.NTLM);
        assertThat(provider, instanceOf(AuthSchemeProvider.class));
    }

    private AuthSchemeProvider getAuthSchemeProvider(String authType) {
        AuthTypes authTypes = new AuthTypes(authType);
        Lookup<AuthSchemeProvider> lookup = new AuthSchemeProviderLookupBuilder()
                .setHeaders(new ArrayList<Header>()).setAuthTypes(authTypes).buildAuthSchemeProviderLookup();
        return lookup.lookup(authType);
    }
}