org.apache.flink.runtime.operators.testutils.ExpectedTestException Java Examples
The following examples show how to use
org.apache.flink.runtime.operators.testutils.ExpectedTestException.
You can vote up the ones you like or vote down the ones you don't like,
and go to the original project or source file by following the links above each example. You may check out the related API usage on the sidebar.
Example #1
Source Project: Flink-CEPplus Author: ljygz File: CombineTaskExternalITCase.java License: Apache License 2.0 | 6 votes |
@Override public void combine(Iterable<Record> records, Collector<Record> out) { Record element = null; int sum = 0; for (Record next : records) { element = next; element.getField(1, this.combineValue); sum += this.combineValue.getValue(); } if (++this.cnt >= 10) { throw new ExpectedTestException(); } this.combineValue.setValue(sum); element.setField(1, this.combineValue); out.collect(element); }
Example #2
Source Project: Flink-CEPplus Author: ljygz File: LeftOuterJoinTaskTest.java License: Apache License 2.0 | 6 votes |
@Test(expected = ExpectedTestException.class) public void testFailingHashLeftOuterJoinTask() throws Exception { int keyCnt1 = 20; int valCnt1 = 20; int keyCnt2 = 20; int valCnt2 = 20; setOutput(new DiscardingOutputCollector<Tuple2<Integer, Integer>>()); addDriverComparator(this.comparator1); addDriverComparator(this.comparator2); getTaskConfig().setDriverPairComparator(new RuntimePairComparatorFactory()); getTaskConfig().setDriverStrategy(DriverStrategy.LEFT_HYBRIDHASH_BUILD_SECOND); getTaskConfig().setRelativeMemoryDriver(this.hash_frac); final AbstractOuterJoinDriver<Tuple2<Integer, Integer>, Tuple2<Integer, Integer>, Tuple2<Integer, Integer>> testTask = getOuterJoinDriver(); addInput(new UniformIntTupleGenerator(keyCnt1, valCnt1, true), this.serializer); addInput(new UniformIntTupleGenerator(keyCnt2, valCnt2, true), this.serializer); testDriver(testTask, MockFailingJoinStub.class); }
Example #3
Source Project: Flink-CEPplus Author: ljygz File: AbstractOuterJoinTaskTest.java License: Apache License 2.0 | 6 votes |
@Test(expected = ExpectedTestException.class) public void testFailingOuterJoinTask() throws Exception { int keyCnt1 = 20; int valCnt1 = 20; int keyCnt2 = 20; int valCnt2 = 20; setOutput(new DiscardingOutputCollector<Tuple2<Integer, Integer>>()); addDriverComparator(this.comparator1); addDriverComparator(this.comparator2); getTaskConfig().setDriverPairComparator(new RuntimePairComparatorFactory()); getTaskConfig().setDriverStrategy(this.getSortDriverStrategy()); getTaskConfig().setRelativeMemoryDriver(this.bnljn_frac); setNumFileHandlesForSort(4); final AbstractOuterJoinDriver<Tuple2<Integer, Integer>, Tuple2<Integer, Integer>, Tuple2<Integer, Integer>> testTask = getOuterJoinDriver(); addInput(new UniformIntTupleGenerator(keyCnt1, valCnt1, true), this.serializer); addInput(new UniformIntTupleGenerator(keyCnt2, valCnt2, true), this.serializer); testDriver(testTask, MockFailingJoinStub.class); }
Example #4
Source Project: Flink-CEPplus Author: ljygz File: RightOuterJoinTaskTest.java License: Apache License 2.0 | 6 votes |
@Test(expected = ExpectedTestException.class) public void testFailingHashRightOuterJoinTask() throws Exception { int keyCnt1 = 20; int valCnt1 = 20; int keyCnt2 = 20; int valCnt2 = 20; setOutput(new DiscardingOutputCollector<Tuple2<Integer, Integer>>()); addDriverComparator(this.comparator1); addDriverComparator(this.comparator2); getTaskConfig().setDriverPairComparator(new RuntimePairComparatorFactory()); getTaskConfig().setDriverStrategy(DriverStrategy.RIGHT_HYBRIDHASH_BUILD_FIRST); getTaskConfig().setRelativeMemoryDriver(this.hash_frac); setNumFileHandlesForSort(4); final AbstractOuterJoinDriver<Tuple2<Integer, Integer>, Tuple2<Integer, Integer>, Tuple2<Integer, Integer>> testTask = getOuterJoinDriver(); addInput(new UniformIntTupleGenerator(keyCnt1, valCnt1, true), this.serializer); addInput(new UniformIntTupleGenerator(keyCnt2, valCnt2, true), this.serializer); testDriver(testTask, MockFailingJoinStub.class); }
Example #5
Source Project: Flink-CEPplus Author: ljygz File: ReduceTaskTest.java License: Apache License 2.0 | 6 votes |
@Test public void testFailingReduceTask() { final int keyCnt = 100; final int valCnt = 20; addInput(new UniformRecordGenerator(keyCnt, valCnt, true)); addDriverComparator(this.comparator); setOutput(this.outList); getTaskConfig().setDriverStrategy(DriverStrategy.SORTED_GROUP_REDUCE); GroupReduceDriver<Record, Record> testTask = new GroupReduceDriver<>(); try { testDriver(testTask, MockFailingReduceStub.class); Assert.fail("Function exception was not forwarded."); } catch (ExpectedTestException eetex) { // Good! } catch (Exception e) { LOG.info("Exception which was not the ExpectedTestException while running the test task.", e); Assert.fail("Test caused exception: " + e.getMessage()); } this.outList.clear(); }
Example #6
Source Project: Flink-CEPplus Author: ljygz File: ReduceTaskTest.java License: Apache License 2.0 | 6 votes |
@Override public void reduce(Iterable<Record> records, Collector<Record> out) { Record element = null; int valCnt = 0; for (Record next : records) { element = next; valCnt++; } if (++this.cnt >= 10) { throw new ExpectedTestException(); } element.getField(0, this.key); this.value.setValue(valCnt - this.key.getValue()); element.setField(1, this.value); out.collect(element); }
Example #7
Source Project: flink Author: flink-tpc-ds File: CombineTaskExternalITCase.java License: Apache License 2.0 | 6 votes |
@Override public void combine(Iterable<Record> records, Collector<Record> out) { Record element = null; int sum = 0; for (Record next : records) { element = next; element.getField(1, this.combineValue); sum += this.combineValue.getValue(); } if (++this.cnt >= 10) { throw new ExpectedTestException(); } this.combineValue.setValue(sum); element.setField(1, this.combineValue); out.collect(element); }
Example #8
Source Project: flink Author: flink-tpc-ds File: CombineTaskTest.java License: Apache License 2.0 | 6 votes |
@Override public void combine(Iterable<Tuple2<Integer, Integer>> records, Collector<Tuple2<Integer, Integer>> out) { int key = 0; int sum = 0; for (Tuple2<Integer, Integer> next : records) { key = next.f0; sum += next.f1; } if (++this.cnt >= 10) { throw new ExpectedTestException(); } int resultValue = sum - key; out.collect(new Tuple2<>(key, resultValue)); }
Example #9
Source Project: flink Author: flink-tpc-ds File: LeftOuterJoinTaskTest.java License: Apache License 2.0 | 6 votes |
@Test(expected = ExpectedTestException.class) public void testFailingHashLeftOuterJoinTask() throws Exception { int keyCnt1 = 20; int valCnt1 = 20; int keyCnt2 = 20; int valCnt2 = 20; setOutput(new DiscardingOutputCollector<Tuple2<Integer, Integer>>()); addDriverComparator(this.comparator1); addDriverComparator(this.comparator2); getTaskConfig().setDriverPairComparator(new RuntimePairComparatorFactory()); getTaskConfig().setDriverStrategy(DriverStrategy.LEFT_HYBRIDHASH_BUILD_SECOND); getTaskConfig().setRelativeMemoryDriver(this.hash_frac); final AbstractOuterJoinDriver<Tuple2<Integer, Integer>, Tuple2<Integer, Integer>, Tuple2<Integer, Integer>> testTask = getOuterJoinDriver(); addInput(new UniformIntTupleGenerator(keyCnt1, valCnt1, true), this.serializer); addInput(new UniformIntTupleGenerator(keyCnt2, valCnt2, true), this.serializer); testDriver(testTask, MockFailingJoinStub.class); }
Example #10
Source Project: flink Author: flink-tpc-ds File: AbstractOuterJoinTaskTest.java License: Apache License 2.0 | 6 votes |
@Test(expected = ExpectedTestException.class) public void testFailingOuterJoinTask() throws Exception { int keyCnt1 = 20; int valCnt1 = 20; int keyCnt2 = 20; int valCnt2 = 20; setOutput(new DiscardingOutputCollector<Tuple2<Integer, Integer>>()); addDriverComparator(this.comparator1); addDriverComparator(this.comparator2); getTaskConfig().setDriverPairComparator(new RuntimePairComparatorFactory()); getTaskConfig().setDriverStrategy(this.getSortDriverStrategy()); getTaskConfig().setRelativeMemoryDriver(this.bnljn_frac); setNumFileHandlesForSort(4); final AbstractOuterJoinDriver<Tuple2<Integer, Integer>, Tuple2<Integer, Integer>, Tuple2<Integer, Integer>> testTask = getOuterJoinDriver(); addInput(new UniformIntTupleGenerator(keyCnt1, valCnt1, true), this.serializer); addInput(new UniformIntTupleGenerator(keyCnt2, valCnt2, true), this.serializer); testDriver(testTask, MockFailingJoinStub.class); }
Example #11
Source Project: flink Author: flink-tpc-ds File: RightOuterJoinTaskTest.java License: Apache License 2.0 | 6 votes |
@Test(expected = ExpectedTestException.class) public void testFailingHashRightOuterJoinTask() throws Exception { int keyCnt1 = 20; int valCnt1 = 20; int keyCnt2 = 20; int valCnt2 = 20; setOutput(new DiscardingOutputCollector<Tuple2<Integer, Integer>>()); addDriverComparator(this.comparator1); addDriverComparator(this.comparator2); getTaskConfig().setDriverPairComparator(new RuntimePairComparatorFactory()); getTaskConfig().setDriverStrategy(DriverStrategy.RIGHT_HYBRIDHASH_BUILD_FIRST); getTaskConfig().setRelativeMemoryDriver(this.hash_frac); setNumFileHandlesForSort(4); final AbstractOuterJoinDriver<Tuple2<Integer, Integer>, Tuple2<Integer, Integer>, Tuple2<Integer, Integer>> testTask = getOuterJoinDriver(); addInput(new UniformIntTupleGenerator(keyCnt1, valCnt1, true), this.serializer); addInput(new UniformIntTupleGenerator(keyCnt2, valCnt2, true), this.serializer); testDriver(testTask, MockFailingJoinStub.class); }
Example #12
Source Project: flink Author: flink-tpc-ds File: ReduceTaskTest.java License: Apache License 2.0 | 6 votes |
@Test public void testFailingReduceTask() { final int keyCnt = 100; final int valCnt = 20; addInput(new UniformRecordGenerator(keyCnt, valCnt, true)); addDriverComparator(this.comparator); setOutput(this.outList); getTaskConfig().setDriverStrategy(DriverStrategy.SORTED_GROUP_REDUCE); GroupReduceDriver<Record, Record> testTask = new GroupReduceDriver<>(); try { testDriver(testTask, MockFailingReduceStub.class); Assert.fail("Function exception was not forwarded."); } catch (ExpectedTestException eetex) { // Good! } catch (Exception e) { LOG.info("Exception which was not the ExpectedTestException while running the test task.", e); Assert.fail("Test caused exception: " + e.getMessage()); } this.outList.clear(); }
Example #13
Source Project: flink Author: flink-tpc-ds File: ReduceTaskTest.java License: Apache License 2.0 | 6 votes |
@Override public void reduce(Iterable<Record> records, Collector<Record> out) { Record element = null; int valCnt = 0; for (Record next : records) { element = next; valCnt++; } if (++this.cnt >= 10) { throw new ExpectedTestException(); } element.getField(0, this.key); this.value.setValue(valCnt - this.key.getValue()); element.setField(1, this.value); out.collect(element); }
Example #14
Source Project: flink Author: apache File: RocksDBInitTest.java License: Apache License 2.0 | 6 votes |
@Test public void testTempLibFolderDeletedOnFail() throws Exception { PowerMockito.spy(RocksDB.class); PowerMockito.when(RocksDB.class, "loadLibrary").thenThrow(new ExpectedTestException()); File tempFolder = temporaryFolder.newFolder(); try { RocksDBStateBackend.ensureRocksDBIsLoaded(tempFolder.getAbsolutePath()); fail("Not throwing expected exception."); } catch (IOException ignored) { // ignored } File[] files = tempFolder.listFiles(); Assert.assertNotNull(files); Assert.assertEquals(0, files.length); }
Example #15
Source Project: flink Author: apache File: CombineTaskExternalITCase.java License: Apache License 2.0 | 6 votes |
@Override public void combine(Iterable<Record> records, Collector<Record> out) { Record element = null; int sum = 0; for (Record next : records) { element = next; element.getField(1, this.combineValue); sum += this.combineValue.getValue(); } if (++this.cnt >= 10) { throw new ExpectedTestException(); } this.combineValue.setValue(sum); element.setField(1, this.combineValue); out.collect(element); }
Example #16
Source Project: flink Author: apache File: StreamTaskTest.java License: Apache License 2.0 | 6 votes |
private void testFailToConfirmCheckpointMessage(Consumer<StreamTask<?, ?>> consumer) throws Exception { StreamMap<Integer, Integer> streamMap = new StreamMap<>(new FailOnNotifyCheckpointMapper<>()); MultipleInputStreamTaskTestHarnessBuilder<Integer> builder = new MultipleInputStreamTaskTestHarnessBuilder<>(OneInputStreamTask::new, BasicTypeInfo.INT_TYPE_INFO) .addInput(BasicTypeInfo.INT_TYPE_INFO); StreamTaskMailboxTestHarness<Integer> harness = builder .setupOutputForSingletonOperatorChain(streamMap) .build(); try { consumer.accept(harness.streamTask); harness.streamTask.runMailboxStep(); fail(); } catch (ExpectedTestException expected) { // expected exception } }
Example #17
Source Project: flink Author: apache File: CombineTaskTest.java License: Apache License 2.0 | 6 votes |
@Override public void combine(Iterable<Tuple2<Integer, Integer>> records, Collector<Tuple2<Integer, Integer>> out) { int key = 0; int sum = 0; for (Tuple2<Integer, Integer> next : records) { key = next.f0; sum += next.f1; } if (++this.cnt >= 10) { throw new ExpectedTestException(); } int resultValue = sum - key; out.collect(new Tuple2<>(key, resultValue)); }
Example #18
Source Project: flink Author: apache File: LeftOuterJoinTaskTest.java License: Apache License 2.0 | 6 votes |
@Test(expected = ExpectedTestException.class) public void testFailingHashLeftOuterJoinTask() throws Exception { int keyCnt1 = 20; int valCnt1 = 20; int keyCnt2 = 20; int valCnt2 = 20; setOutput(new DiscardingOutputCollector<Tuple2<Integer, Integer>>()); addDriverComparator(this.comparator1); addDriverComparator(this.comparator2); getTaskConfig().setDriverPairComparator(new RuntimePairComparatorFactory()); getTaskConfig().setDriverStrategy(DriverStrategy.LEFT_HYBRIDHASH_BUILD_SECOND); getTaskConfig().setRelativeMemoryDriver(this.hash_frac); final AbstractOuterJoinDriver<Tuple2<Integer, Integer>, Tuple2<Integer, Integer>, Tuple2<Integer, Integer>> testTask = getOuterJoinDriver(); addInput(new UniformIntTupleGenerator(keyCnt1, valCnt1, true), this.serializer); addInput(new UniformIntTupleGenerator(keyCnt2, valCnt2, true), this.serializer); testDriver(testTask, MockFailingJoinStub.class); }
Example #19
Source Project: flink Author: apache File: AbstractOuterJoinTaskTest.java License: Apache License 2.0 | 6 votes |
@Test(expected = ExpectedTestException.class) public void testFailingOuterJoinTask() throws Exception { int keyCnt1 = 20; int valCnt1 = 20; int keyCnt2 = 20; int valCnt2 = 20; setOutput(new DiscardingOutputCollector<Tuple2<Integer, Integer>>()); addDriverComparator(this.comparator1); addDriverComparator(this.comparator2); getTaskConfig().setDriverPairComparator(new RuntimePairComparatorFactory()); getTaskConfig().setDriverStrategy(this.getSortDriverStrategy()); getTaskConfig().setRelativeMemoryDriver(this.bnljn_frac); setNumFileHandlesForSort(4); final AbstractOuterJoinDriver<Tuple2<Integer, Integer>, Tuple2<Integer, Integer>, Tuple2<Integer, Integer>> testTask = getOuterJoinDriver(); addInput(new UniformIntTupleGenerator(keyCnt1, valCnt1, true), this.serializer); addInput(new UniformIntTupleGenerator(keyCnt2, valCnt2, true), this.serializer); testDriver(testTask, MockFailingJoinStub.class); }
Example #20
Source Project: flink Author: apache File: RightOuterJoinTaskTest.java License: Apache License 2.0 | 6 votes |
@Test(expected = ExpectedTestException.class) public void testFailingHashRightOuterJoinTask() throws Exception { int keyCnt1 = 20; int valCnt1 = 20; int keyCnt2 = 20; int valCnt2 = 20; setOutput(new DiscardingOutputCollector<Tuple2<Integer, Integer>>()); addDriverComparator(this.comparator1); addDriverComparator(this.comparator2); getTaskConfig().setDriverPairComparator(new RuntimePairComparatorFactory()); getTaskConfig().setDriverStrategy(DriverStrategy.RIGHT_HYBRIDHASH_BUILD_FIRST); getTaskConfig().setRelativeMemoryDriver(this.hash_frac); setNumFileHandlesForSort(4); final AbstractOuterJoinDriver<Tuple2<Integer, Integer>, Tuple2<Integer, Integer>, Tuple2<Integer, Integer>> testTask = getOuterJoinDriver(); addInput(new UniformIntTupleGenerator(keyCnt1, valCnt1, true), this.serializer); addInput(new UniformIntTupleGenerator(keyCnt2, valCnt2, true), this.serializer); testDriver(testTask, MockFailingJoinStub.class); }
Example #21
Source Project: flink Author: apache File: ReduceTaskTest.java License: Apache License 2.0 | 6 votes |
@Test public void testFailingReduceTask() { final int keyCnt = 100; final int valCnt = 20; addInput(new UniformRecordGenerator(keyCnt, valCnt, true)); addDriverComparator(this.comparator); setOutput(this.outList); getTaskConfig().setDriverStrategy(DriverStrategy.SORTED_GROUP_REDUCE); GroupReduceDriver<Record, Record> testTask = new GroupReduceDriver<>(); try { testDriver(testTask, MockFailingReduceStub.class); Assert.fail("Function exception was not forwarded."); } catch (ExpectedTestException eetex) { // Good! } catch (Exception e) { LOG.info("Exception which was not the ExpectedTestException while running the test task.", e); Assert.fail("Test caused exception: " + e.getMessage()); } this.outList.clear(); }
Example #22
Source Project: flink Author: apache File: ReduceTaskTest.java License: Apache License 2.0 | 6 votes |
@Override public void reduce(Iterable<Record> records, Collector<Record> out) { Record element = null; int valCnt = 0; for (Record next : records) { element = next; valCnt++; } if (++this.cnt >= 10) { throw new ExpectedTestException(); } element.getField(0, this.key); this.value.setValue(valCnt - this.key.getValue()); element.setField(1, this.value); out.collect(element); }
Example #23
Source Project: Flink-CEPplus Author: ljygz File: CoGroupTaskTest.java License: Apache License 2.0 | 5 votes |
@Test public void testFailingSortCoGroupTask() { int keyCnt1 = 100; int valCnt1 = 2; int keyCnt2 = 200; int valCnt2 = 1; setOutput(this.output); addInput(new UniformRecordGenerator(keyCnt1, valCnt1, true)); addInput(new UniformRecordGenerator(keyCnt2, valCnt2, true)); addDriverComparator(this.comparator1); addDriverComparator(this.comparator2); getTaskConfig().setDriverPairComparator(RecordPairComparatorFactory.get()); getTaskConfig().setDriverStrategy(DriverStrategy.CO_GROUP); final CoGroupDriver<Record, Record, Record> testTask = new CoGroupDriver<Record, Record, Record>(); try { testDriver(testTask, MockFailingCoGroupStub.class); Assert.fail("Function exception was not forwarded."); } catch (ExpectedTestException etex) { // good! } catch (Exception e) { e.printStackTrace(); Assert.fail("The test caused an exception."); } }
Example #24
Source Project: Flink-CEPplus Author: ljygz File: CoGroupTaskTest.java License: Apache License 2.0 | 5 votes |
@Override public void coGroup(Iterable<Record> records1, Iterable<Record> records2, Collector<Record> out) { int val1Cnt = 0; for (@SuppressWarnings("unused") Record r : records1) { val1Cnt++; } for (Record record2 : records2) { if (val1Cnt == 0) { if(++this.cnt>=10) { throw new ExpectedTestException(); } out.collect(record2); } else { for (int i=0; i<val1Cnt; i++) { if(++this.cnt>=10) { throw new ExpectedTestException(); } out.collect(record2); } } } }
Example #25
Source Project: Flink-CEPplus Author: ljygz File: CachedMatchTaskTest.java License: Apache License 2.0 | 5 votes |
@Test public void testFailingHashFirstMatchTask() { int keyCnt1 = 20; int valCnt1 = 20; int keyCnt2 = 20; int valCnt2 = 20; addInput(new UniformRecordGenerator(keyCnt1, valCnt1, false)); addInput(new UniformRecordGenerator(keyCnt2, valCnt2, false)); addDriverComparator(this.comparator1); addDriverComparator(this.comparator2); getTaskConfig().setDriverPairComparator(RecordPairComparatorFactory.get()); setOutput(new NirvanaOutputList()); getTaskConfig().setDriverStrategy(DriverStrategy.HYBRIDHASH_BUILD_FIRST_CACHED); getTaskConfig().setRelativeMemoryDriver(1.0f); BuildFirstCachedJoinDriver<Record, Record, Record> testTask = new BuildFirstCachedJoinDriver<Record, Record, Record>(); try { testResettableDriver(testTask, MockFailingMatchStub.class, 3); Assert.fail("Function exception was not forwarded."); } catch (ExpectedTestException etex) { // good! } catch (Exception e) { e.printStackTrace(); Assert.fail("Test caused an exception."); } }
Example #26
Source Project: Flink-CEPplus Author: ljygz File: CachedMatchTaskTest.java License: Apache License 2.0 | 5 votes |
@Test public void testFailingHashSecondMatchTask() { int keyCnt1 = 20; int valCnt1 = 20; int keyCnt2 = 20; int valCnt2 = 20; addInput(new UniformRecordGenerator(keyCnt1, valCnt1, false)); addInput(new UniformRecordGenerator(keyCnt2, valCnt2, false)); addDriverComparator(this.comparator1); addDriverComparator(this.comparator2); getTaskConfig().setDriverPairComparator(RecordPairComparatorFactory.get()); setOutput(new NirvanaOutputList()); getTaskConfig().setDriverStrategy(DriverStrategy.HYBRIDHASH_BUILD_SECOND_CACHED); getTaskConfig().setRelativeMemoryDriver(1.0f); BuildSecondCachedJoinDriver<Record, Record, Record> testTask = new BuildSecondCachedJoinDriver<Record, Record, Record>(); try { testResettableDriver(testTask, MockFailingMatchStub.class, 3); Assert.fail("Function exception was not forwarded."); } catch (ExpectedTestException etex) { // good! } catch (Exception e) { e.printStackTrace(); Assert.fail("Test caused an exception."); } }
Example #27
Source Project: Flink-CEPplus Author: ljygz File: CachedMatchTaskTest.java License: Apache License 2.0 | 5 votes |
@Override public void join(Record record1, Record record2, Collector<Record> out) throws Exception { if (++this.cnt >= 10) { throw new ExpectedTestException(); } out.collect(record1); }
Example #28
Source Project: Flink-CEPplus Author: ljygz File: JoinTaskTest.java License: Apache License 2.0 | 5 votes |
@Test public void testFailingHashFirstMatchTask() { int keyCnt1 = 20; int valCnt1 = 20; int keyCnt2 = 20; int valCnt2 = 20; addInput(new UniformRecordGenerator(keyCnt1, valCnt1, false)); addInput(new UniformRecordGenerator(keyCnt2, valCnt2, false)); addDriverComparator(this.comparator1); addDriverComparator(this.comparator2); getTaskConfig().setDriverPairComparator(RecordPairComparatorFactory.get()); setOutput(new NirvanaOutputList()); getTaskConfig().setDriverStrategy(DriverStrategy.HYBRIDHASH_BUILD_FIRST); getTaskConfig().setRelativeMemoryDriver(hash_frac); JoinDriver<Record, Record, Record> testTask = new JoinDriver<>(); try { testDriver(testTask, MockFailingMatchStub.class); Assert.fail("Function exception was not forwarded."); } catch (ExpectedTestException etex) { // good! } catch (Exception e) { e.printStackTrace(); Assert.fail("Test caused an exception."); } }
Example #29
Source Project: Flink-CEPplus Author: ljygz File: JoinTaskTest.java License: Apache License 2.0 | 5 votes |
@Test public void testFailingHashSecondMatchTask() { int keyCnt1 = 20; int valCnt1 = 20; int keyCnt2 = 20; int valCnt2 = 20; addInput(new UniformRecordGenerator(keyCnt1, valCnt1, false)); addInput(new UniformRecordGenerator(keyCnt2, valCnt2, false)); addDriverComparator(this.comparator1); addDriverComparator(this.comparator2); getTaskConfig().setDriverPairComparator(RecordPairComparatorFactory.get()); setOutput(new NirvanaOutputList()); getTaskConfig().setDriverStrategy(DriverStrategy.HYBRIDHASH_BUILD_SECOND); getTaskConfig().setRelativeMemoryDriver(hash_frac); JoinDriver<Record, Record, Record> testTask = new JoinDriver<>(); try { testDriver(testTask, MockFailingMatchStub.class); Assert.fail("Function exception was not forwarded."); } catch (ExpectedTestException etex) { // good! } catch (Exception e) { e.printStackTrace(); Assert.fail("Test caused an exception."); } }
Example #30
Source Project: Flink-CEPplus Author: ljygz File: JoinTaskTest.java License: Apache License 2.0 | 5 votes |
@Override public void join(Record record1, Record record2, Collector<Record> out) throws Exception { if (++this.cnt >= 10) { throw new ExpectedTestException(); } out.collect(record1); }