55 lines
1.8 KiB
Go
55 lines
1.8 KiB
Go
package dbimpl
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/grafana/grafana/pkg/setting"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestGetEnginePostgresFromConfig(t *testing.T) {
|
|
cfg := setting.NewCfg()
|
|
s, err := cfg.Raw.NewSection("entity_api")
|
|
require.NoError(t, err)
|
|
s.Key("db_type").SetValue("mysql")
|
|
s.Key("db_host").SetValue("localhost")
|
|
s.Key("db_name").SetValue("grafana")
|
|
s.Key("db_user").SetValue("user")
|
|
s.Key("db_password").SetValue("password")
|
|
|
|
engine, err := getEnginePostgres(cfg.SectionWithEnvOverrides("entity_api"), nil)
|
|
|
|
assert.NotNil(t, engine)
|
|
assert.NoError(t, err)
|
|
assert.True(t, strings.Contains(engine.DataSourceName(), "dbname=grafana"))
|
|
}
|
|
|
|
func TestGetEngineMySQLFromConfig(t *testing.T) {
|
|
cfg := setting.NewCfg()
|
|
s, err := cfg.Raw.NewSection("entity_api")
|
|
require.NoError(t, err)
|
|
s.Key("db_type").SetValue("mysql")
|
|
s.Key("db_host").SetValue("localhost")
|
|
s.Key("db_name").SetValue("grafana")
|
|
s.Key("db_user").SetValue("user")
|
|
s.Key("db_password").SetValue("password")
|
|
|
|
engine, err := getEngineMySQL(cfg.SectionWithEnvOverrides("entity_api"), nil)
|
|
|
|
assert.NotNil(t, engine)
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
func TestGetConnectionStrings(t *testing.T) {
|
|
t.Run("generate mysql connection string", func(t *testing.T) {
|
|
expected := "user:password@tcp(localhost)/grafana?collation=utf8mb4_unicode_ci&allowNativePasswords=true&clientFoundRows=true"
|
|
assert.Equal(t, expected, connectionStringMySQL("user", "password", "tcp", "localhost", "grafana"))
|
|
})
|
|
t.Run("generate postgres connection string", func(t *testing.T) {
|
|
expected := "user=user password=password host=localhost port=5432 dbname=grafana sslmode=disable"
|
|
assert.Equal(t, expected, connectionStringPostgres("user", "password", "localhost", "5432", "grafana", "disable"))
|
|
})
|
|
}
|