Handle Flaky Test SpannerToSourceDbCustomTransformationIT timeout failure.

v2/spanner-to-sourcedb/src/test/java/com/google/cloud/teleport/v2/templates/SpannerToSourceDbCustomTransformationIT.java

@@ -334,19 +334,30 @@ private void writeRowInSpanner() {
     spannerResourceManager.write(m);
   }
 
-  private void assertRowInMySQL() {
+  private void assertRowInMySQL() throws InterruptedException {
     PipelineOperator.Result result =
         pipelineOperator()
             .waitForCondition(
                 createConfig(jobInfo, Duration.ofMinutes(15)),
                 () -> jdbcResourceManager.getRowCount(TABLE) == 1);
+    /*
+     * Added to handle updates.
+     * TODO(khajanchi@), explore if this sleep be replaced with something more definite.
+     */
+    Thread.sleep(Duration.ofMinutes(1L).toMillis());
+
     assertThatResult(result).meetsConditions();
 
     result =
         pipelineOperator()
             .waitForCondition(
                 createConfig(jobInfo, Duration.ofMinutes(15)),
                 () -> jdbcResourceManager.getRowCount(TABLE2) == 2);
+    /*
+     * Added to handle updates.
+     * TODO(khajanchi@), explore if this sleep be replaced with something more definite.
+     */
+    Thread.sleep(Duration.ofMinutes(1L).toMillis());
     assertThatResult(result).meetsConditions();
 
     List<Map<String, Object>> rows = jdbcResourceManager.readTable(TABLE);

v2/spanner-to-sourcedb/src/test/resources/SpannerToSourceDbCustomTransformationIT/mysql-schema.sql

@@ -1,3 +1,6 @@
+DROP TABLE IF EXISTS Users1;
+DROP TABLE IF EXISTS AllDatatypeTransformation;
+
 CREATE TABLE Users1 (
     id INT NOT NULL,
     first_name VARCHAR(25),

v2/spanner-to-sourcedb/src/test/resources/SpannerToSourceDbCustomTransformationIT/spanner-schema.sql

@@ -1,4 +1,7 @@
-CREATE TABLE IF NOT EXISTS Users1 (
+DROP TABLE IF EXISTS Users1;
+DROP TABLE IF EXISTS AllDatatypeTransformation;
+
+CREATE TABLE Users1 (
     id INT64 NOT NULL,
     name STRING(25),
 ) PRIMARY KEY(id);