Tests: Adding Reverse Replication Tests (GoogleCloudPlatform#1951)

* changes * changes * changes * change * change * change * change * change * changes
ollionorg · Nov 18, 2024 · b04de34 · b04de34
1 parent b7e6db7
commit b04de34
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 10 deletions.
diff --git a/...to-sourcedb/src/test/java/com/google/cloud/teleport/v2/templates/SpannerToSourceDbIT.java b/...to-sourcedb/src/test/java/com/google/cloud/teleport/v2/templates/SpannerToSourceDbIT.java
@@ -49,7 +49,10 @@
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
-/** Integration test for {@link SpannerToSourceDb} Flex template. */
+/**
+ * Integration test for {@link SpannerToSourceDb} Flex template for basic run including new spanner
+ * tables and column rename use-case.
+ */
 @Category({TemplateIntegrationTest.class, SkipDirectRunnerTest.class})
 @TemplateIntegrationTest(SpannerToSourceDb.class)
 @RunWith(JUnit4.class)
@@ -134,7 +137,7 @@ public static void cleanUp() throws IOException {
   }
 
   @Test
-  public void spannerToSourceDbBasic() throws InterruptedException {
+  public void spannerToSourceDbBasic() throws InterruptedException, IOException {
     assertThatPipeline(jobInfo).isRunning();
     // Write row in Spanner
     writeRowInSpanner();
@@ -144,16 +147,20 @@ public void spannerToSourceDbBasic() throws InterruptedException {
 
   private void writeRowInSpanner() {
     // Write a single record to Spanner
-    Mutation m =
+    Mutation m1 =
         Mutation.newInsertOrUpdateBuilder("Users")
             .set("id")
             .to(1)
-            .set("name")
+            .set("full_name")
             .to("FF")
             .set("from")
             .to("AA")
             .build();
-    spannerResourceManager.write(m);
+    spannerResourceManager.write(m1);
+
+    Mutation m2 =
+        Mutation.newInsertOrUpdateBuilder("Users2").set("id").to(2).set("name").to("B").build();
+    spannerResourceManager.write(m2);
 
     // Write a single record to Spanner for the given logical shard
     // Add the record with the transaction tag as txBy=
@@ -171,14 +178,14 @@ private void writeRowInSpanner() {
         .run(
             (TransactionCallable<Void>)
                 transaction -> {
-                  Mutation m2 =
+                  Mutation m3 =
                       Mutation.newInsertOrUpdateBuilder("Users")
                           .set("id")
                           .to(2)
-                          .set("name")
+                          .set("full_name")
                           .to("GG")
                           .build();
-                  transaction.buffer(m2);
+                  transaction.buffer(m3);
                   return null;
                 });
   }

diff --git a/v2/spanner-to-sourcedb/src/test/resources/SpannerToSourceDbIT/session.json b/v2/spanner-to-sourcedb/src/test/resources/SpannerToSourceDbIT/session.json
@@ -28,7 +28,7 @@
           "Id": "c142"
         },
         "c143": {
-          "Name": "name",
+          "Name": "full_name",
           "T": {
             "Name": "STRING",
             "Len": 25,

diff --git a/v2/spanner-to-sourcedb/src/test/resources/SpannerToSourceDbIT/spanner-schema.sql b/v2/spanner-to-sourcedb/src/test/resources/SpannerToSourceDbIT/spanner-schema.sql
@@ -1,9 +1,14 @@
 CREATE TABLE IF NOT EXISTS Users (
     id INT64 NOT NULL,
-    name STRING(25),
+    full_name STRING(25),
     `from` STRING(25)
 ) PRIMARY KEY(id);
 
+CREATE TABLE IF NOT EXISTS Users2 (
+    id INT64 NOT NULL,
+    name STRING(25),
+    ) PRIMARY KEY(id);
+
 CREATE CHANGE STREAM allstream
   FOR ALL OPTIONS (
   value_capture_type = 'NEW_ROW',