Function newTestDatabase

main_test.go:99–146

newTestDatabase creates a new temporary test database. When a test database connection is returned, it will have created a new database and initialized it with tables from a reference database.

(base *sql.DB, dbUser, dbPassword, dbName, dbHost string)

// database connection is returned, it will have created a new database and
// initialized it with tables from a reference database.
func newTestDatabase(base *sql.DB, dbUser, dbPassword, dbName, dbHost string) (*sql.DB, func() error, error) {
	var err error
	var baseName = dbName

	if baseName == "" {
		row := base.QueryRow("SELECT DATABASE()")
		err := row.Scan(&baseName)
		if err != nil {
			return nil, nil, err
		}
	}
	tUUID, _ := uuid.NewV4()
	suffix := strings.Replace(tUUID.String(), "-", "_", -1)
	newDBName := baseName + suffix
	_, err = base.Exec("CREATE DATABASE " + newDBName)
	if err != nil {
		return nil, nil, err
	}
	newDB, err := initMySQL(dbUser, dbPassword, newDBName, dbHost)
	if err != nil {
		return nil, nil, err
	}

	rows, err := base.Query("SHOW TABLES IN " + baseName)
	if err != nil {
		return nil, nil, err
	}
	for rows.Next() {
		var tableName string
		if err := rows.Scan(&tableName); err != nil {
			return nil, nil, err
		}
		query := fmt.Sprintf("CREATE TABLE %s LIKE %s.%s", tableName, baseName, tableName)
		if _, err := newDB.Exec(query); err != nil {
			return nil, nil, err
		}
	}

	cleanup := func() error {
		if closeErr := newDB.Close(); closeErr != nil {
			fmt.Println(closeErr)
		}

		_, err = base.Exec("DROP DATABASE " + newDBName)
		return err
	}
	return newDB, cleanup, nil
}

func countRows(t *testing.T, ctx context.Context, db *sql.DB, count int, query string, args ...interface{}) {
	var returned int