NEOS-1791: Add Mysql init schema set and enum test (#3417)

Author: alishakawaguchi

internal/integration-tests/worker/workflow/mysql_test.go

@@ -580,6 +580,7 @@ func test_mysql_schema_reconciliation(
 		{schema: schema, table: "cyclic_table", rowCount: 3},
 		{schema: schema, table: "astronaut", rowCount: 2},
 		{schema: schema, table: "astronaut_log", rowCount: 2},
+		{schema: schema, table: "plants", rowCount: 3},
 	}
 
 	for _, expected := range expectedResults {
@@ -767,7 +768,7 @@ func test_mysql_schema_reconciliation_column_values(
 	testutil_testdata.VerifySQLTableColumnValues(t, ctx, mysql.Source.DB, mysql.Target.DB, schema, "multi_col_parent", sqlmanager_shared.MysqlDriver, []string{"mcp_a", "mcp_b"})
 	testutil_testdata.VerifySQLTableColumnValues(t, ctx, mysql.Source.DB, mysql.Target.DB, schema, "multi_col_child", sqlmanager_shared.MysqlDriver, []string{"mc_child_id"})
 	testutil_testdata.VerifySQLTableColumnValues(t, ctx, mysql.Source.DB, mysql.Target.DB, schema, "cyclic_table", sqlmanager_shared.MysqlDriver, []string{"cycle_id"})
-
+	testutil_testdata.VerifySQLTableColumnValues(t, ctx, mysql.Source.DB, mysql.Target.DB, schema, "plants", sqlmanager_shared.MysqlDriver, []string{"plant_id"})
 }
 
 func cleanupMysqlDatabases(ctx context.Context, mysql *tcmysql.MysqlTestSyncContainer, databases []string) error {

internal/integration-tests/worker/workflow/workflow-integration_test.go

@@ -167,7 +167,7 @@ func Test_Workflow(t *testing.T) {
 				t.Parallel()
 				test_mysql_schema_reconciliation(t, ctx, mysql, neosyncApi, dbManagers, accountId, sourceConn, destConn, true)
 			})
-			t.Run("no_truncate", func(t *testing.T) {
+			t.Run("retain_data", func(t *testing.T) {
 				t.Parallel()
 				test_mysql_schema_reconciliation(t, ctx, mysql, neosyncApi, dbManagers, accountId, sourceConn, destConn, false)
 			})

internal/testutil/testdata/mysql/schema-init/alter-job-mappings.go

@@ -126,5 +126,29 @@ func GetAlterSyncJobMappings(schema string) []*mgmtv1alpha1.JobMapping {
 				},
 			},
 		},
+		{
+			Schema: schema,
+			Table:  "plants",
+			Column: "climate",
+			Transformer: &mgmtv1alpha1.JobMappingTransformer{
+				Config: &mgmtv1alpha1.TransformerConfig{
+					Config: &mgmtv1alpha1.TransformerConfig_PassthroughConfig{
+						PassthroughConfig: &mgmtv1alpha1.Passthrough{},
+					},
+				},
+			},
+		},
+		{
+			Schema: schema,
+			Table:  "plants",
+			Column: "soil_types",
+			Transformer: &mgmtv1alpha1.JobMappingTransformer{
+				Config: &mgmtv1alpha1.TransformerConfig{
+					Config: &mgmtv1alpha1.TransformerConfig_PassthroughConfig{
+						PassthroughConfig: &mgmtv1alpha1.Passthrough{},
+					},
+				},
+			},
+		},
 	}
 }

internal/testutil/testdata/mysql/schema-init/alter-statements.sql

@@ -269,3 +269,10 @@ ALTER TABLE astronaut_log
 
 ALTER TABLE astronaut
   MODIFY COLUMN first_name VARCHAR (20) COMMENT 'I\'m an astronaut';
+
+
+ALTER TABLE plants
+ADD COLUMN climate ENUM('Tropical', 'Desert', 'Temperate') NOT NULL DEFAULT 'Temperate';
+
+ALTER TABLE plants
+ADD COLUMN soil_types SET('Sandy', 'Clay', 'Loamy', 'Silty', 'Peaty') NOT NULL DEFAULT 'Loamy';

internal/testutil/testdata/mysql/schema-init/create-tables.sql

@@ -472,3 +472,19 @@ INSERT INTO astronaut_log (
 VALUES
     (1, 'Neil Armstrong', 'INSERTED', NOW()),
     (2, 'Buzz Aldrin', 'INSERTED', NOW());
+
+
+CREATE TABLE IF NOT EXISTS plants (
+    plant_id INT PRIMARY KEY AUTO_INCREMENT,
+    plant_name VARCHAR(100) NOT NULL,
+    plant_type ENUM('Alive', 'Dormant', 'Dead') NOT NULL,
+    water_types SET('Rainwater', 'Tapwater', 'Wellwater') NOT NULL
+);
+
+INSERT INTO plants (plant_name, plant_type, water_types) 
+VALUES 
+    ('Palm Tree', 'Alive', 'Rainwater'),
+    ('Cactus', 'Dormant', 'Rainwater'),
+    ('Flower', 'Dead', 'Wellwater');
+
+