RowGranularity.DOC, DataTypes.INTEGER)),
new Reference(new ReferenceInfo(
new ReferenceIdent(tableIdent, "date"),
RowGranularity.SHARD, DataTypes.TIMESTAMP))
));
collectNode.maxRowGranularity(RowGranularity.DOC);
collectNode.isPartitioned(true);
Object[][] result = operation.collect(collectNode).get();
assertThat(result.length, is(2));
assertThat((Integer)result[0][0], isOneOf(1,2));