DBZ-8325 Fix snapshot recovery itest

debezium · Oct 30, 2024 · 6a36d4e · 6a36d4e
1 parent c55cc9b
commit 6a36d4e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/main/java/io/debezium/connector/vitess/VitessValueConverter.java b/src/main/java/io/debezium/connector/vitess/VitessValueConverter.java
@@ -128,7 +128,7 @@ public SchemaBuilder schemaBuilder(Column column) {
     @Override
     public ValueConverter converter(Column column, Field fieldDefn) {
         String typeName = column.typeName().toUpperCase();
-        if (matches(typeName, "JSON")) {
+        if (matches(typeName, Query.Type.JSON.name())) {
             return (data) -> convertJson(column, fieldDefn, data);
         }
         if (matches(typeName, Query.Type.ENUM.name())) {

diff --git a/src/test/java/io/debezium/connector/vitess/VitessConnectorIT.java b/src/test/java/io/debezium/connector/vitess/VitessConnectorIT.java
@@ -2136,7 +2136,8 @@ public void testMidSnapshotRecoveryLargeTable() throws Exception {
         assertThat(recordCount < expectedSnapshotRecordsCount).isTrue();
         // Assert the total snapshot records are sent after starting
         consumer = testConsumer(expectedSnapshotRecordsCount, tableTopicPrefix);
-        startConnector();
+        startConnector(Function.identity(), false, false, 1,
+                -1, -1, tableInclude, VitessConnectorConfig.SnapshotMode.INITIAL, TestHelper.TEST_SHARD);
         consumer.await(TestHelper.waitTimeForRecords(), TimeUnit.SECONDS);
 
         for (int i = 1; i <= expectedSnapshotRecordsCount; i++) {