org.hawkular.component.pinger.TraitsTest.java Source code

Introduction

Here is the source code for org.hawkular.component.pinger.TraitsTest.java

Source

/*
 * Copyright 2015 Red Hat, Inc. and/or its affiliates
 * and other contributors as indicated by the @author tags.
 *
 * 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 org.hawkular.component.pinger;

import java.util.Map;

import org.apache.http.HttpResponse;
import org.apache.http.HttpStatus;
import org.apache.http.HttpVersion;
import org.apache.http.message.BasicHttpResponse;
import org.hawkular.component.pinger.Traits.TraitHeader;
import org.junit.Assert;
import org.junit.Test;

import com.google.common.collect.ImmutableMap;

/**
 * @author <a href="https://github.com/ppalaga">Peter Palaga</a>
 */
public class TraitsTest {

    @Test
    public void testCollect() {

        HttpResponse response = new BasicHttpResponse(HttpVersion.HTTP_1_0, HttpStatus.SC_OK, "OK");
        /* Real headers from google.cz */
        response.addHeader("Date", "Thu, 14 May 2015 16:23:11 GMT");
        response.addHeader("Expires", "-1");
        response.addHeader("Cache-Control", "private, max-age=0");
        response.addHeader("Content-Type", "text/html; charset=ISO-8859-2");
        response.addHeader("Set-Cookie", "PREF=ID=1cf542d95de677ce:FF=0:TM=1431620591: ... etc");
        response.addHeader("Set-Cookie", "NID=67=UwDo4W6rmaylwdjaMgnLW6N_Mtd5X4lFth36C ... etc");
        response.addHeader("Server", "gws");
        response.addHeader("X-XSS-Protection", "1; mode=block");
        response.addHeader("X-Frame-Options", "SAMEORIGIN");
        response.addHeader("Alternate-Protocol", "80:quic,p=1");
        response.addHeader("Accept-Ranges", "none");
        response.addHeader("Vary", "Accept-Encoding");
        response.addHeader("Transfer-Encoding", "chunked");

        Map<TraitHeader, String> found = Traits.collect(response, 0).getItems();

        Map<TraitHeader, String> expected = new ImmutableMap.Builder<TraitHeader, String>()
                .put(TraitHeader.SERVER, "gws").build();

        Assert.assertEquals(expected, found);

    }

    @Test
    public void testCollectEmpty() {

        HttpResponse response = new BasicHttpResponse(HttpVersion.HTTP_1_0, HttpStatus.SC_OK, "OK");
        Map<TraitHeader, String> found = Traits.collect(response, 0).getItems();
        Map<TraitHeader, String> expected = new ImmutableMap.Builder<TraitHeader, String>().build();
        Assert.assertEquals(expected, found);

    }

    @Test
    public void testCollectCase() {

        HttpResponse response = new BasicHttpResponse(HttpVersion.HTTP_1_0, HttpStatus.SC_OK, "OK");
        response.addHeader("sErVeR", "whatever");
        response.addHeader("X-POWERED-BY", "hawkular");

        Map<TraitHeader, String> found = Traits.collect(response, 0).getItems();

        Map<TraitHeader, String> expected = new ImmutableMap.Builder<TraitHeader, String>()
                .put(TraitHeader.X_POWERED_BY, "hawkular").put(TraitHeader.SERVER, "whatever").build();

        Assert.assertEquals(expected, found);

    }

}