Implement ExecuteSqls() for each test container type

.github/workflows/go.yml

@@ -8,7 +8,7 @@ on:
 
 jobs:
 
-  build:
+  build-and-test:
     runs-on: ubuntu-22.04
     steps:
     - uses: actions/checkout@v3

yb-voyager/test/containers/mysql_container.go

@@ -2,6 +2,7 @@ package testcontainers
 
 import (
 	"context"
+	"database/sql"
 	"fmt"
 	"os"
 	"time"
@@ -16,6 +17,7 @@ import (
 type MysqlContainer struct {
 	ContainerConfig
 	container testcontainers.Container
+	db        *sql.DB
 }
 
 func (ms *MysqlContainer) Start(ctx context.Context) (err error) {
@@ -59,14 +61,39 @@ func (ms *MysqlContainer) Start(ctx context.Context) (err error) {
 		ContainerRequest: req,
 		Started:          true,
 	})
-	return err
+	if err != nil {
+		return err
+	}
+
+	dsn := ms.GetConnectionString()
+	db, err := sql.Open("mysql", dsn)
+	if err != nil {
+		return fmt.Errorf("failed to open mysql connection: %w", err)
+	}
+
+	if err = db.Ping(); err != nil {
+		db.Close()
+		return fmt.Errorf("failed to ping mysql after connection: %w", err)
+	}
+
+	// Store the DB connection for reuse
+	ms.db = db
+
+	return nil
 }
 
 func (ms *MysqlContainer) Terminate(ctx context.Context) {
 	if ms == nil {
 		return
 	}
 
+	// Close the DB connection if it exists
+	if ms.db != nil {
+		if err := ms.db.Close(); err != nil {
+			log.Errorf("failed to close mysql db connection: %v", err)
+		}
+	}
+
 	err := ms.container.Terminate(ctx)
 	if err != nil {
 		log.Errorf("failed to terminate mysql container: %v", err)
@@ -96,6 +123,29 @@ func (ms *MysqlContainer) GetConfig() ContainerConfig {
 	return ms.ContainerConfig
 }
 
+// GetConnectionString constructs and returns the MySQL DSN
 func (ms *MysqlContainer) GetConnectionString() string {
-	panic("GetConnectionString() not implemented yet for mysql")
+	host, port, err := ms.GetHostPort()
+	if err != nil {
+		utils.ErrExit("failed to get host port for mysql connection string: %v", err)
+	}
+
+	// DSN format: user:password@tcp(host:port)/dbname
+	return fmt.Sprintf("%s:%s@tcp(%s:%d)/%s",
+		ms.User, ms.Password, host, port, ms.DBName)
+}
+
+// ExecuteSqls executes a list of SQL statements using the persistent DB connection
+func (ms *MysqlContainer) ExecuteSqls(sqls []string) error {
+	if ms.db == nil {
+		return fmt.Errorf("db connection not initialized for mysql container")
+	}
+
+	for _, sqlStmt := range sqls {
+		_, err := ms.db.Exec(sqlStmt)
+		if err != nil {
+			return fmt.Errorf("failed to execute sql '%s': %w", sqlStmt, err)
+		}
+	}
+	return nil
 }

yb-voyager/test/containers/oracle_container.go

@@ -100,3 +100,7 @@ func (ora *OracleContainer) GetConfig() ContainerConfig {
 func (ora *OracleContainer) GetConnectionString() string {
 	panic("GetConnectionString() not implemented yet for oracle")
 }
+
+func (ora *OracleContainer) ExecuteSqls(sqls []string) error {
+	return nil
+}

yb-voyager/test/containers/postgres_container.go

@@ -8,6 +8,7 @@ import (
 	"time"
 
 	"github.com/docker/go-connections/nat"
+	"github.com/jackc/pgx/v5"
 	log "github.com/sirupsen/logrus"
 	"github.com/testcontainers/testcontainers-go"
 	"github.com/testcontainers/testcontainers-go/wait"
@@ -125,3 +126,20 @@ func (pg *PostgresContainer) GetConnectionString() string {
 
 	return fmt.Sprintf("postgresql://%s:%s@%s:%d/%s", config.User, config.Password, host, port, config.DBName)
 }
+
+func (pg *PostgresContainer) ExecuteSqls(sqls []string) error {
+	connStr := pg.GetConnectionString()
+	conn, err := pgx.Connect(context.Background(), connStr)
+	if err != nil {
+		return fmt.Errorf("failed to connect postgres for executing sqls: %w", err)
+	}
+	defer conn.Close(context.Background())
+
+	for _, sql := range sqls {
+		_, err := conn.Exec(context.Background(), sql)
+		if err != nil {
+			return fmt.Errorf("failed to execute sql '%s': %w", sql, err)
+		}
+	}
+	return nil
+}

yb-voyager/test/containers/testcontainers.go

@@ -28,6 +28,7 @@ type TestContainer interface {
 
 			// Add Capability to run multiple versions of a dbtype parallely
 	*/
+	ExecuteSqls(sqls []string) error
 }
 
 type ContainerConfig struct {

yb-voyager/test/containers/yugabytedb_container.go

@@ -7,6 +7,7 @@ import (
 	"time"
 
 	"github.com/docker/go-connections/nat"
+	"github.com/jackc/pgx/v5"
 	log "github.com/sirupsen/logrus"
 	"github.com/testcontainers/testcontainers-go"
 	"github.com/testcontainers/testcontainers-go/wait"
@@ -109,3 +110,20 @@ func (yb *YugabyteDBContainer) GetConnectionString() string {
 
 	return fmt.Sprintf("postgresql://%s:%s@%s:%d/%s", config.User, config.Password, host, port, config.DBName)
 }
+
+func (yb *YugabyteDBContainer) ExecuteSqls(sqls []string) error {
+	connStr := yb.GetConnectionString()
+	conn, err := pgx.Connect(context.Background(), connStr)
+	if err != nil {
+		return fmt.Errorf("failed to connect postgres for executing sqls: %w", err)
+	}
+	defer conn.Close(context.Background())
+
+	for _, sql := range sqls {
+		_, err := conn.Exec(context.Background(), sql)
+		if err != nil {
+			return fmt.Errorf("failed to execute sql '%s': %w", sql, err)
+		}
+	}
+	return nil
+}