add test

Author: chenhuiyeh

replication/replication_test.go

@@ -464,3 +464,74 @@ func (t *testSyncerSuite) TestMysqlBinlogCodec() {
 		require.NoError(t.T(), err)
 	}
 }
+
+func (t *testSyncerSuite) TestFloatWithTrailingZeros() {
+	t.setupTest(mysql.MySQLFlavor)
+
+	str := `DROP TABLE IF EXISTS test_float_zeros`
+	t.testExecute(str)
+
+	// Create table with both float and double columns
+	str = `CREATE TABLE test_float_zeros (
+		id INT PRIMARY KEY,
+		f_val FLOAT,
+		d_val DOUBLE
+	)`
+	t.testExecute(str)
+
+	// Test with useFloatWithTrailingZero = true
+	t.b.cfg.UseFloatWithTrailingZero = true
+	t.testFloatWithTrailingZerosCase(true)
+
+	// Test with useFloatWithTrailingZero = false
+	t.b.cfg.UseFloatWithTrailingZero = false
+	t.testFloatWithTrailingZerosCase(false)
+}
+
+func (t *testSyncerSuite) testFloatWithTrailingZerosCase(useTrailingZero bool) {
+	// Insert values with trailing zeros
+	t.testExecute(`INSERT INTO test_float_zeros VALUES (1, 5.1, 5.1)`)
+	t.testExecute(`INSERT INTO test_float_zeros VALUES (2, 1.100, 1.100)`)
+
+	// Get current position
+	r, err := t.c.Execute("SHOW MASTER STATUS")
+	require.NoError(t.T(), err)
+	binFile, _ := r.GetString(0, 0)
+	binPos, _ := r.GetInt(0, 1)
+
+	// Start syncing from current position
+	s, err := t.b.StartSync(mysql.Position{Name: binFile, Pos: uint32(binPos)})
+	require.NoError(t.T(), err)
+
+	// Insert another row to trigger binlog events
+	t.testExecute(`INSERT INTO test_float_zeros VALUES (3, 3.0, 3.0)`)
+
+	// Read events and verify
+	ctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)
+	defer cancel()
+
+	for {
+		evt, err := s.GetEvent(ctx)
+		require.NoError(t.T(), err)
+
+		// We're interested in RowsEvent
+		if evt.Header.EventType != WRITE_ROWS_EVENTv2 {
+			continue
+		}
+
+		// Type assert to RowsEvent
+		rowsEvent := evt.Event.(*RowsEvent)
+		for _, row := range rowsEvent.Rows {
+			// The third row should contain our test values
+			if row[0].(int32) == 3 {
+				// Float should preserve trailing zeros if useFloatWithTrailingZero is true
+				fVal := row[1].(float32)
+				if useTrailingZero {
+					require.Equal(t.T(), "3.0", fmt.Sprintf("%.1f", fVal))
+				} else {
+					require.Equal(t.T(), "3", fmt.Sprintf("%.1f", fVal))
+				}
+			}
+		}
+	}
+}