@Test
public void testCollectWithPartitionedColumns() throws Exception {
Routing routing = docSchemaInfo.getTableInfo(PARTITIONED_TABLE_NAME).getRouting(WhereClause.MATCH_ALL);
TableIdent tableIdent = new TableIdent(DocSchemaInfo.NAME, PARTITIONED_TABLE_NAME);
CollectNode collectNode = new CollectNode("docCollect", routing);
collectNode.toCollect(Arrays.<Symbol>asList(
new Reference(new ReferenceInfo(
new ReferenceIdent(tableIdent, "id"),
RowGranularity.DOC, DataTypes.INTEGER)),
new Reference(new ReferenceInfo(
new ReferenceIdent(tableIdent, "date"),