Function getDriver

backend/tests/transaction_mode_test.go:196–252

Helper functions

(ctx context.Context, engine storepb.Engine, container *testcontainer.Container, testDBName string)

// Helper functions
func getDriver(ctx context.Context, engine storepb.Engine, container *testcontainer.Container, testDBName string) (db.Driver, error) {
	var dbDatabase string
	switch engine {
	case storepb.Engine_MYSQL:
		dbDatabase = testDBName
	case storepb.Engine_POSTGRES:
		dbDatabase = "postgres"
	case storepb.Engine_ORACLE:
		dbDatabase = "FREEPDB1"
	case storepb.Engine_MSSQL:
		dbDatabase = "master"
	default:
		dbDatabase = ""
	}

	config := db.ConnectionConfig{
		DataSource: &storepb.DataSource{
			Host:     container.GetHost(),
			Port:     container.GetPort(),
			Username: "root",
			Database: dbDatabase,
		},
		ConnectionContext: db.ConnectionContext{
			DatabaseName: dbDatabase,
		},
		Password: "root",
	}

	// Special handling for different engines based on testcontainer setup
	switch engine {
	case storepb.Engine_MYSQL:
		config.DataSource.Username = "root"
		config.Password = "root-password"
	case storepb.Engine_POSTGRES:
		config.DataSource.Username = "postgres"
		config.Password = "root-password"
	case storepb.Engine_ORACLE:
		config.DataSource.Username = "testuser"
		config.Password = "testpass"
		config.DataSource.Database = ""
		config.ConnectionContext.DatabaseName = ""
		// Set service name for Oracle
		config.DataSource.ServiceName = "FREEPDB1"
	case storepb.Engine_MSSQL:
		config.DataSource.Username = "sa"
		config.Password = "Test123!"
	default:
		// Use default config for other engines
	}

	driver, err := db.Open(ctx, engine, config)
	if err != nil {
		return nil, err
	}

	return driver, nil
}