mirror of
https://github.com/NginxProxyManager/nginx-proxy-manager.git
synced 2025-04-29 18:42:28 +00:00
45 lines
790 B
Go
45 lines
790 B
Go
package test
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"npm/internal/config"
|
|
"npm/internal/database"
|
|
|
|
"github.com/DATA-DOG/go-sqlmock"
|
|
"gorm.io/driver/postgres"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// Setup ...
|
|
func Setup() (sqlmock.Sqlmock, error) {
|
|
db, mock, err := sqlmock.New()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
dialector := postgres.New(postgres.Config{
|
|
Conn: db,
|
|
DriverName: "postgres",
|
|
})
|
|
gormDB, err := gorm.Open(dialector, &gorm.Config{})
|
|
database.SetDB(gormDB)
|
|
return mock, err
|
|
}
|
|
|
|
// InitConfig ...
|
|
func InitConfig(t *testing.T, envs ...string) {
|
|
if len(envs) > 0 {
|
|
for _, env := range envs {
|
|
parts := strings.Split(env, "=")
|
|
if len(parts) == 2 {
|
|
t.Setenv(parts[0], parts[1])
|
|
}
|
|
}
|
|
}
|
|
|
|
version := "999.999.999"
|
|
commit := "abcd123"
|
|
config.Init(&version, &commit)
|
|
}
|