com.gooddata.md.report.AttributeInGridTest.java Source code

Introduction

Here is the source code for com.gooddata.md.report.AttributeInGridTest.java

Source

/*
 * Copyright (C) 2004-2017, GoodData(R) Corporation. All rights reserved.
 * This source code is licensed under the BSD-style license found in the
 * LICENSE.txt file in the root directory of this source tree.
 */
package com.gooddata.md.report;

import com.gooddata.md.Attribute;
import com.gooddata.md.DisplayForm;
import org.apache.commons.lang3.SerializationUtils;
import org.testng.annotations.Test;

import java.util.Iterator;
import java.util.List;

import static com.gooddata.util.ResourceUtils.readObjectFromResource;
import static java.util.Arrays.asList;
import static net.javacrumbs.jsonunit.JsonMatchers.jsonEquals;
import static net.javacrumbs.jsonunit.core.util.ResourceUtils.resource;
import static org.hamcrest.CoreMatchers.is;
import static org.hamcrest.CoreMatchers.notNullValue;
import static org.hamcrest.MatcherAssert.assertThat;
import static org.hamcrest.Matchers.hasSize;
import static org.hamcrest.text.MatchesPattern.matchesPattern;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;

public class AttributeInGridTest {

    private static final String URI = "/URI";
    private static final String ALIAS = "ALIAS";
    private static final String ATTR_TITLE = "attrTitle";

    @Test
    public void testDeserialization() throws Exception {
        final AttributeInGrid attr = readObjectFromResource("/md/report/attributeInGrid.json",
                AttributeInGrid.class);

        assertThat(attr, is(notNullValue()));
        assertThat(attr.getUri(), is(URI));
        assertThat(attr.getAlias(), is(ALIAS));

        assertThat(attr.getStringTotals(), is(notNullValue()));
        assertThat(attr.getStringTotals(), hasSize(2));
        final Iterator<List<Total>> i = attr.getTotals().iterator();
        final List<Total> subTotal1 = i.next();
        assertThat(subTotal1, hasSize(1));
        assertThat(subTotal1.iterator().next(), is(Total.SUM));
        final List<Total> subTotal2 = i.next();
        assertThat(subTotal2, hasSize(2));
        assertThat(subTotal2.get(0), is(Total.AVG));
        assertThat(subTotal2.get(1), is(Total.MED));
    }

    @Test
    public void testSerialization() throws Exception {
        final List<List<Total>> totals = asList(asList(Total.SUM), asList(Total.AVG, Total.MED));
        final AttributeInGrid attr = new AttributeInGrid(URI, ALIAS, totals);
        assertThat(attr, jsonEquals(resource("md/report/attributeInGrid.json")));
    }

    @Test
    public void testCreateFromAttribute() throws Exception {
        final Attribute attr = mock(Attribute.class);
        final DisplayForm displayForm = mock(DisplayForm.class);
        when(attr.getDefaultDisplayForm()).thenReturn(displayForm);
        when(attr.getTitle()).thenReturn(ATTR_TITLE);
        when(displayForm.getUri()).thenReturn(URI);
        when(displayForm.getTitle()).thenReturn(ALIAS);

        final AttributeInGrid attrInGrid = new AttributeInGrid(attr);
        assertThat(attrInGrid.getUri(), is(URI));
        assertThat(attrInGrid.getAlias(), is(ATTR_TITLE));
    }

    @Test
    public void testCreateFromDisplayForm() throws Exception {
        final DisplayForm displayForm = mock(DisplayForm.class);
        when(displayForm.getUri()).thenReturn(URI);
        when(displayForm.getTitle()).thenReturn(ALIAS);

        final AttributeInGrid attrInGrid = new AttributeInGrid(displayForm);
        assertThat(attrInGrid.getUri(), is(URI));
        assertThat(attrInGrid.getAlias(), is(ALIAS));
    }

    @Test
    public void testToStringFormat() throws Exception {
        final AttributeInGrid attr = readObjectFromResource("/md/report/attributeInGrid.json",
                AttributeInGrid.class);

        assertThat(attr.toString(), matchesPattern(AttributeInGrid.class.getSimpleName() + "\\[.*\\]"));
    }

    @Test
    public void testSerializable() throws Exception {
        final AttributeInGrid attr = readObjectFromResource("/md/report/attributeInGrid.json",
                AttributeInGrid.class);
        final AttributeInGrid deserialized = SerializationUtils.roundtrip(attr);

        assertThat(deserialized, jsonEquals(attr));
    }
}