com.hp.autonomy.iod.client.api.search.FieldNamesTest.java Source code

Introduction

Here is the source code for com.hp.autonomy.iod.client.api.search.FieldNamesTest.java

Source

/*
 * Copyright 2015 Hewlett-Packard Development Company, L.P.
 * Licensed under the MIT License (the "License"); you may not use this file except in compliance with the License.
 */

package com.hp.autonomy.iod.client.api.search;

import com.fasterxml.jackson.databind.ObjectMapper;
import lombok.extern.slf4j.Slf4j;
import org.junit.Before;
import org.junit.Test;

import java.io.IOException;
import java.io.StringWriter;
import java.io.Writer;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;

import static org.hamcrest.CoreMatchers.hasItems;
import static org.hamcrest.CoreMatchers.is;
import static org.junit.Assert.assertThat;

@Slf4j
public class FieldNamesTest {

    private FieldNames fieldNames;

    @Before
    public void initialize() {
        final Map<String, Integer> fieldValue0 = new LinkedHashMap<>();
        fieldValue0.put("1", 1);
        fieldValue0.put("2", 2);

        final Map<String, Integer> fieldValue1 = new LinkedHashMap<>();
        fieldValue1.put("3", 3);
        fieldValue1.put("4", 4);
        fieldValue1.put("5", 5);

        fieldNames = new FieldNames.Builder().addParametricValue("zero", fieldValue0)
                .addParametricValue("one", fieldValue1).build();
    }

    @Test
    public void testGetFieldSet() {
        final Set<String> fieldSet = fieldNames.getFieldNames();

        assertThat(fieldSet, hasItems("zero", "one"));
    }

    @Test
    public void testGetFieldValueCount() {
        assertThat(fieldNames.getFieldValueCount("zero", "1"), is(1));
        assertThat(fieldNames.getFieldValueCount("one", "4"), is(4));
    }

    @Test
    public void testGetValuesForFieldName() {
        final Set<String> setZero = fieldNames.getValuesForFieldName("zero");
        assertThat(setZero, hasItems("1", "2"));

        final Set<String> setOne = fieldNames.getValuesForFieldName("one");
        assertThat(setOne, hasItems("3", "4", "5"));
    }

    @Test
    public void testGetValuesAndCountsForFieldName() {
        final List<FieldNames.ValueAndCount> parametricValues = fieldNames.getValuesAndCountsForFieldName("zero");
        final FieldNames.ValueAndCount one = parametricValues.get(0);
        assertThat(one.getValue(), is("1"));
        assertThat(one.getCount(), is(1));
    }

    @Test
    public void testIterator() {
        int x = 0;

        for (final FieldNames.ParametricValue value : fieldNames) {
            x++;

            assertThat(value.getValue(), is(String.valueOf(x)));
            assertThat(value.getCount(), is(x));

            if (x < 3) {
                assertThat(value.getFieldName(), is("zero"));
            } else {
                assertThat(value.getFieldName(), is("one"));
            }

        }

    }

    @Test
    public void testGetJson() throws IOException {
        final ObjectMapper mapper = new ObjectMapper();
        final Writer output = new StringWriter();
        mapper.writeValue(output, fieldNames);
        final String json = output.toString();

        assertThat(json, is(
                "{\"zero\":[{\"value\":\"1\",\"count\":1},{\"value\":\"2\",\"count\":2}],\"one\":[{\"value\":\"3\",\"count\":3},{\"value\":\"4\",\"count\":4},{\"value\":\"5\",\"count\":5}]}"));
    }
}