com.hp.autonomy.frontend.find.core.parametricfields.AbstractParametricValuesControllerTest.java Source code

Introduction

Here is the source code for com.hp.autonomy.frontend.find.core.parametricfields.AbstractParametricValuesControllerTest.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.frontend.find.core.parametricfields;

import com.hp.autonomy.frontend.find.core.search.QueryRestrictionsBuilderFactory;
import com.hp.autonomy.searchcomponents.core.parametricvalues.BucketingParams;
import com.hp.autonomy.searchcomponents.core.parametricvalues.ParametricRequest;
import com.hp.autonomy.searchcomponents.core.parametricvalues.ParametricValuesService;
import com.hp.autonomy.searchcomponents.core.search.QueryRestrictions;
import com.hp.autonomy.types.requests.idol.actions.tags.RangeInfo;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Matchers;
import org.mockito.Mock;
import org.mockito.invocation.InvocationOnMock;
import org.mockito.runners.MockitoJUnitRunner;
import org.mockito.stubbing.Answer;
import org.springframework.beans.factory.ObjectFactory;

import java.io.Serializable;
import java.io.UnsupportedEncodingException;
import java.net.URLEncoder;
import java.util.Collections;
import java.util.List;
import java.util.Map;

import static org.hamcrest.core.Is.is;
import static org.junit.Assert.assertThat;
import static org.mockito.Mockito.*;

@RunWith(MockitoJUnitRunner.class)
public abstract class AbstractParametricValuesControllerTest<C extends ParametricValuesController<Q, R, S, E>, Q extends QueryRestrictions<S>, R extends ParametricRequest<S>, S extends Serializable, E extends Exception> {
    @Mock
    protected ParametricValuesService<R, S, E> parametricValuesService;

    @Mock
    protected QueryRestrictionsBuilderFactory<Q, S> queryRestrictionsBuilderFactory;

    @Mock
    protected ObjectFactory<ParametricRequest.Builder<R, S>> parametricRequestBuilderFactory;

    protected C parametricValuesController;

    protected abstract C newControllerInstance();

    @Before
    public void setUp() {
        parametricValuesController = newControllerInstance();
    }

    @Test
    public void getDependentParametricValues() throws E {
        parametricValuesController.getDependentParametricValues(Collections.singletonList("SomeParametricField"),
                "Some query text", null, Collections.<S>emptyList(), null, null, 0, null);
        verify(parametricValuesService).getDependentParametricValues(Matchers.<R>any());
    }

    @Test
    public void getParametricValuesInBuckets() throws UnsupportedEncodingException, E {
        final String fieldName = "birth&death";

        final RangeInfo rangeInfo = mock(RangeInfo.class);
        final BucketingParams expectedBucketingParams = new BucketingParams(5, -0.5, 0.5);

        when(parametricValuesService.getNumericParametricValuesInBuckets(Matchers.<R>any(),
                Matchers.<Map<String, BucketingParams>>any())).thenAnswer(new Answer<List<RangeInfo>>() {
                    @Override
                    public List<RangeInfo> answer(final InvocationOnMock invocation) {
                        @SuppressWarnings("unchecked")
                        final Map<String, BucketingParams> bucketingParamsPerField = invocation.getArgumentAt(1,
                                Map.class);

                        final BucketingParams bucketingParams = bucketingParamsPerField.get(fieldName);
                        return expectedBucketingParams.equals(bucketingParams)
                                ? Collections.singletonList(rangeInfo)
                                : Collections.<RangeInfo>emptyList();
                    }
                });

        final RangeInfo output = parametricValuesController.getNumericParametricValuesInBucketsForField(
                URLEncoder.encode(fieldName, "UTF-8"), expectedBucketingParams.getTargetNumberOfBuckets(),
                expectedBucketingParams.getMin(), expectedBucketingParams.getMax(), "*", "",
                Collections.<S>emptyList(), null, null, 0);

        assertThat(output, is(rangeInfo));
    }
}