org.apache.flink.connector.testutils.source.reader

Class SourceReaderTestBase<SplitT extends SourceSplit>

java.lang.Object

org.apache.flink.util.TestLogger

org.apache.flink.connector.testutils.source.reader.SourceReaderTestBase<SplitT>

Type Parameters:

SplitT - the type of the splits.

public abstract class SourceReaderTestBase<SplitT extends SourceSplit>
extends TestLogger

An abstract test class for all the unit tests of SourceReader to inherit.

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
class	SourceReaderTestBase.ValidatingSourceOutput A source output that validates the output.

Field Summary

Fields

Modifier and Type	Field and Description
protected static int	NUM_RECORDS_PER_SPLIT
protected int	numSplits
protected int	totalNumRecords

Fields inherited from class org.apache.flink.util.TestLogger

log, nameProvider, watchman

Constructor Summary

Constructors

Constructor and Description
SourceReaderTestBase()

Method Summary

Modifier and Type	Method and Description
protected abstract SourceReader<Integer,SplitT>	createReader()
void	ensureNoDangling()
protected abstract long	getNextRecordIndex(SplitT split)
protected int	getNumSplits()
protected abstract SplitT	getSplit(int splitId, int numRecords, Boundedness boundedness)
protected abstract List<SplitT>	getSplits(int numSplits, int numRecordsPerSplit, Boundedness boundedness)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Detail

numSplits

protected final int numSplits

totalNumRecords

protected final int totalNumRecords

NUM_RECORDS_PER_SPLIT

protected static final int NUM_RECORDS_PER_SPLIT

See Also:

Constant Field Values

Constructor Detail

SourceReaderTestBase

public SourceReaderTestBase()

Method Detail

getNumSplits

protected int getNumSplits()

ensureNoDangling

@AfterEach
public void ensureNoDangling()

createReader

protected abstract SourceReader<Integer,SplitT> createReader()
                                                      throws Exception

Throws:

Exception

getSplits

protected abstract List<SplitT> getSplits(int numSplits,
                                          int numRecordsPerSplit,
                                          Boundedness boundedness)

getSplit

protected abstract SplitT getSplit(int splitId,
                                   int numRecords,
                                   Boundedness boundedness)

getNextRecordIndex

protected abstract long getNextRecordIndex(SplitT split)