DBZ-8160 Updated tests to reflect engine's signal API changes
This commit is contained in:
parent
f20d05fab9
commit
bec109308d
@ -31,8 +31,8 @@
|
||||
import io.debezium.connector.postgresql.PostgresConnectorConfig.SnapshotMode;
|
||||
import io.debezium.connector.postgresql.spi.CustomActionProvider;
|
||||
import io.debezium.embedded.async.AbstractAsyncEngineConnectorTest;
|
||||
import io.debezium.engine.DebeziumEngine;
|
||||
import io.debezium.junit.logging.LogInterceptor;
|
||||
import io.debezium.pipeline.signal.SignalRecord;
|
||||
import io.debezium.pipeline.signal.actions.Log;
|
||||
|
||||
public class SignalsIT extends AbstractAsyncEngineConnectorTest {
|
||||
@ -114,7 +114,7 @@ private void signalLog(boolean includingEscapedCharacter, boolean useSource) thr
|
||||
}
|
||||
else {
|
||||
expectedNumRecords = 1;
|
||||
signaler.signal(new SignalRecord("1", "log", "{\"message\": \"Signal message at offset ''{}''\"}", null));
|
||||
getSignaler().signal(new DebeziumEngine.Signal("1", "log", "{\"message\": \"Signal message at offset ''{}''\"}", null));
|
||||
}
|
||||
|
||||
waitForAvailableRecords(800, TimeUnit.MILLISECONDS);
|
||||
|
@ -8,7 +8,6 @@
|
||||
import io.debezium.embedded.AbstractConnectorTest;
|
||||
import io.debezium.embedded.TestingDebeziumEngine;
|
||||
import io.debezium.engine.DebeziumEngine;
|
||||
import io.debezium.pipeline.signal.SignalRecord;
|
||||
|
||||
/**
|
||||
* Base class for testing connectors using {@link AsyncEmbeddedEngine}.
|
||||
@ -17,18 +16,17 @@
|
||||
*/
|
||||
public class AbstractAsyncEngineConnectorTest extends AbstractConnectorTest {
|
||||
|
||||
protected DebeziumEngine.Signaler<SignalRecord> signaler;
|
||||
|
||||
@Override
|
||||
protected DebeziumEngine.Builder createEngineBuilder() {
|
||||
this.signaler = new AsyncEngineSignaler();
|
||||
|
||||
return new AsyncEmbeddedEngine.AsyncEngineBuilder()
|
||||
.using(signaler);
|
||||
return new AsyncEmbeddedEngine.AsyncEngineBuilder();
|
||||
}
|
||||
|
||||
@Override
|
||||
protected TestingDebeziumEngine createEngine(DebeziumEngine.Builder builder) {
|
||||
return new TestingAsyncEmbeddedEngine((AsyncEmbeddedEngine) builder.build());
|
||||
}
|
||||
|
||||
protected DebeziumEngine.Signaler getSignaler() {
|
||||
return engine.getSignaler();
|
||||
}
|
||||
}
|
||||
|
@ -36,4 +36,9 @@ public void close() throws IOException {
|
||||
public void runWithTask(Consumer consumer) {
|
||||
engine.runWithTask(consumer);
|
||||
}
|
||||
|
||||
@Override
|
||||
public Signaler getSignaler() {
|
||||
return engine.getSignaler();
|
||||
}
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user