mirror of
https://github.com/d0zingcat/gocryptotrader.git
synced 2026-05-13 15:09:42 +00:00
e8e0ff3a35
* CI: Add macOS, Windows and Linux arm64 support * Modify arm64 job name to be inline with the others * linter: Prevent cache from causing "Cannot open: File exists" issues * Use setup-go and setup-node's inbuilt caching
92 lines
2.5 KiB
Go
92 lines
2.5 KiB
Go
package testhelpers
|
|
|
|
import (
|
|
"database/sql"
|
|
"path/filepath"
|
|
"reflect"
|
|
|
|
"github.com/thrasher-corp/gocryptotrader/database"
|
|
"github.com/thrasher-corp/gocryptotrader/database/drivers"
|
|
psqlConn "github.com/thrasher-corp/gocryptotrader/database/drivers/postgres"
|
|
sqliteConn "github.com/thrasher-corp/gocryptotrader/database/drivers/sqlite3"
|
|
"github.com/thrasher-corp/gocryptotrader/database/repository"
|
|
"github.com/thrasher-corp/goose"
|
|
"github.com/thrasher-corp/sqlboiler/boil"
|
|
)
|
|
|
|
var (
|
|
// TempDir temp folder for sqlite database
|
|
TempDir string
|
|
// PostgresTestDatabase postgresql database config details
|
|
PostgresTestDatabase *database.Config
|
|
// MigrationDir default folder for migration's
|
|
MigrationDir = filepath.Join("..", "..", "migrations")
|
|
)
|
|
|
|
// GetConnectionDetails returns connection details for CI or test db instances
|
|
func GetConnectionDetails() *database.Config {
|
|
return &database.Config{
|
|
Enabled: true,
|
|
Driver: "postgres",
|
|
ConnectionDetails: drivers.ConnectionDetails{
|
|
// Host: "",
|
|
// Port: 5432,
|
|
// Username: "",
|
|
// Password: "",
|
|
// Database: "",
|
|
// SSLMode: "",
|
|
},
|
|
}
|
|
}
|
|
|
|
// ConnectToDatabase opens connection to database and returns pointer to instance of database.DB
|
|
func ConnectToDatabase(conn *database.Config) (dbConn *database.Instance, err error) {
|
|
err = database.DB.SetConfig(conn)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if conn.Driver == database.DBPostgreSQL {
|
|
dbConn, err = psqlConn.Connect(conn)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
} else if conn.Driver == database.DBSQLite3 || conn.Driver == database.DBSQLite {
|
|
database.DB.DataPath = TempDir
|
|
dbConn, err = sqliteConn.Connect(conn.Database)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
err = migrateDB(database.DB.SQL)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
database.DB.SetConnected(true)
|
|
return
|
|
}
|
|
|
|
// CloseDatabase closes database connection
|
|
func CloseDatabase(conn *database.Instance) (err error) {
|
|
if conn != nil {
|
|
return conn.SQL.Close()
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// CheckValidConfig checks if database connection details are empty
|
|
func CheckValidConfig(config *drivers.ConnectionDetails) bool {
|
|
return !reflect.DeepEqual(drivers.ConnectionDetails{}, *config)
|
|
}
|
|
|
|
func migrateDB(db *sql.DB) error {
|
|
return goose.Run("up", db, repository.GetSQLDialect(), MigrationDir, "")
|
|
}
|
|
|
|
// EnableVerboseTestOutput enables debug output for SQL queries
|
|
func EnableVerboseTestOutput() error {
|
|
boil.DebugMode = true
|
|
boil.DebugWriter = database.Logger{}
|
|
return nil
|
|
}
|