@Test(expectedExceptions = ExceededMemoryLimitException.class, expectedExceptionsMessageRegExp = "Query exceeded local memory limit of 10B") public void testMemoryLimit() throws Exception { List<Page> input = rowPagesBuilder(BIGINT, DOUBLE) .row(1, 0.1) .row(2, 0.2) .pageBreak() .row(-1, -0.1) .row(4, 0.4) .build(); DriverContext driverContext = createTaskContext(executor, TEST_SESSION, new DataSize(10, Unit.BYTE)) .addPipelineContext(true, true) .addDriverContext(); OrderByOperatorFactory operatorFactory = new OrderByOperatorFactory( 0, new PlanNodeId("test"), ImmutableList.of(BIGINT, DOUBLE), ImmutableList.of(1), 10, ImmutableList.of(0), ImmutableList.of(ASC_NULLS_LAST)); Operator operator = operatorFactory.createOperator(driverContext); toPages(operator, input); } }
@Test public void testMultiFieldKey() throws Exception { List<Page> input = rowPagesBuilder(VARCHAR, BIGINT) .row("a", 1) .row("b", 2) .pageBreak() .row("b", 3) .row("a", 4) .build(); OrderByOperatorFactory operatorFactory = new OrderByOperatorFactory( 0, new PlanNodeId("test"), ImmutableList.of(VARCHAR, BIGINT), ImmutableList.of(0, 1), 10, ImmutableList.of(0, 1), ImmutableList.of(ASC_NULLS_LAST, DESC_NULLS_LAST)); Operator operator = operatorFactory.createOperator(driverContext); MaterializedResult expected = MaterializedResult.resultBuilder(driverContext.getSession(), VARCHAR, BIGINT) .row("a", 4) .row("a", 1) .row("b", 3) .row("b", 2) .build(); assertOperatorEquals(operator, input, expected); }
@Test public void testSingleFieldKey() throws Exception { List<Page> input = rowPagesBuilder(BIGINT, DOUBLE) .row(1, 0.1) .row(2, 0.2) .pageBreak() .row(-1, -0.1) .row(4, 0.4) .build(); OrderByOperatorFactory operatorFactory = new OrderByOperatorFactory( 0, new PlanNodeId("test"), ImmutableList.of(BIGINT, DOUBLE), ImmutableList.of(1), 10, ImmutableList.of(0), ImmutableList.of(ASC_NULLS_LAST)); Operator operator = operatorFactory.createOperator(driverContext); MaterializedResult expected = resultBuilder(driverContext.getSession(), DOUBLE) .row(-0.1) .row(0.1) .row(0.2) .row(0.4) .build(); assertOperatorEquals(operator, input, expected); }
@Test public void testReverseOrder() throws Exception { List<Page> input = rowPagesBuilder(BIGINT, DOUBLE) .row(1, 0.1) .row(2, 0.2) .pageBreak() .row(-1, -0.1) .row(4, 0.4) .build(); OrderByOperatorFactory operatorFactory = new OrderByOperatorFactory( 0, new PlanNodeId("test"), ImmutableList.of(BIGINT, DOUBLE), ImmutableList.of(0), 10, ImmutableList.of(0), ImmutableList.of(DESC_NULLS_LAST)); Operator operator = operatorFactory.createOperator(driverContext); MaterializedResult expected = resultBuilder(driverContext.getSession(), BIGINT) .row(4) .row(2) .row(1) .row(-1) .build(); assertOperatorEquals(operator, input, expected); }