From 6a36d4e2e3d4b8d96d26f8d7be3fdb5c93a6acb1 Mon Sep 17 00:00:00 2001
From: twthorn <thomaswilliamthornton@gmail.com>
Date: Wed, 30 Oct 2024 17:15:09 -0400
Subject: [PATCH] DBZ-8325 Fix snapshot recovery itest

---
 .../io/debezium/connector/vitess/VitessValueConverter.java     | 2 +-
 .../java/io/debezium/connector/vitess/VitessConnectorIT.java   | 3 ++-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/main/java/io/debezium/connector/vitess/VitessValueConverter.java b/src/main/java/io/debezium/connector/vitess/VitessValueConverter.java
index ccab7b42..7a23f913 100644
--- 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
index e670a4c0..aaed7deb 100644
--- 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++) {