/**
* Copyright (C) 2012 - present by OpenGamma Inc. and the OpenGamma group of companies
*
* Please see distribution for license.
*/
package com.opengamma.web.analytics;
import static org.testng.AssertJUnit.assertTrue;
import org.json.JSONArray;
import org.json.JSONException;
import org.testng.annotations.Test;
import com.google.common.collect.ImmutableList;
import com.opengamma.util.test.TestGroup;
import com.opengamma.web.analytics.formatting.ResultsFormatter;
import com.opengamma.web.analytics.formatting.TypeFormatter;
/**
* Test.
*/
@Test(groups = TestGroup.UNIT)
public class GridColumnsJsonWriterTest {
@Test
public void getJson() throws JSONException {
TestCellRenderer renderer = new TestCellRenderer();
GridColumn col1 = new GridColumn("col1", "col1 desc", Double.class, renderer);
GridColumn col2 = new GridColumn("col2", "col2 desc", String.class, renderer);
GridColumn col3 = new GridColumn("col3", "col3 desc", Double.class, renderer);
GridColumn col4 = new GridColumn("col4", "col4 desc", String.class, renderer);
GridColumnGroup group1 = new GridColumnGroup("group1", ImmutableList.of(col1, col2), true);
GridColumnGroup group2 = new GridColumnGroup("group2", ImmutableList.of(col3, col4), false);
ImmutableList<GridColumnGroup> groups = ImmutableList.of(group1, group2);
GridColumnsJsonWriter writer = new GridColumnsJsonWriter(new ResultsFormatter());
String json = writer.getJson(groups);
String expectedJson =
"[{\"name\":\"group1\",\"dependencyGraphsAvailable\":true,\"columns\":[" +
"{\"header\":\"col1\",\"description\":\"col1 desc\",\"type\":\"DOUBLE\"}," +
"{\"header\":\"col2\",\"description\":\"col2 desc\",\"type\":\"STRING\"}]}," +
"{\"name\":\"group2\",\"dependencyGraphsAvailable\":false,\"columns\":[" +
"{\"header\":\"col3\",\"description\":\"col3 desc\",\"type\":\"DOUBLE\"}," +
"{\"header\":\"col4\",\"description\":\"col4 desc\",\"type\":\"STRING\"}]}]";
assertTrue(JsonTestUtils.equal(new JSONArray(expectedJson), new JSONArray(json)));
}
private static class TestCellRenderer implements GridColumn.CellRenderer {
@Override
public ResultsCell getResults(int rowIndex,
TypeFormatter.Format format,
ResultsCache cache,
Class<?> columnType,
Object inlineKey) {
return null;
}
}
}