Diego Augusto Molina
93 lines
2.1 KiB
Go
93 lines
2.1 KiB
Go
package dbimpl
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestGetEngineMySQLFromConfig(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
t.Run("happy path", func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
getter := newTestSectionGetter(map[string]string{
|
|
"db_type": "mysql",
|
|
"db_host": "/var/run/mysql.socket",
|
|
"db_name": "grafana",
|
|
"db_user": "user",
|
|
"db_password": "password",
|
|
})
|
|
engine, err := getEngineMySQL(getter, nil)
|
|
assert.NotNil(t, engine)
|
|
assert.NoError(t, err)
|
|
})
|
|
|
|
t.Run("invalid string", func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
getter := newTestSectionGetter(map[string]string{
|
|
"db_type": "mysql",
|
|
"db_host": "/var/run/mysql.socket",
|
|
"db_name": string(invalidUTF8ByteSequence),
|
|
"db_user": "user",
|
|
"db_password": "password",
|
|
})
|
|
engine, err := getEngineMySQL(getter, nil)
|
|
assert.Nil(t, engine)
|
|
assert.Error(t, err)
|
|
assert.ErrorIs(t, err, ErrInvalidUTF8Sequence)
|
|
})
|
|
}
|
|
|
|
func TestGetEnginePostgresFromConfig(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
t.Run("happy path", func(t *testing.T) {
|
|
t.Parallel()
|
|
getter := newTestSectionGetter(map[string]string{
|
|
"db_type": "mysql",
|
|
"db_host": "localhost",
|
|
"db_name": "grafana",
|
|
"db_user": "user",
|
|
"db_password": "password",
|
|
})
|
|
engine, err := getEnginePostgres(getter, nil)
|
|
|
|
assert.NotNil(t, engine)
|
|
assert.NoError(t, err)
|
|
})
|
|
|
|
t.Run("invalid string", func(t *testing.T) {
|
|
t.Parallel()
|
|
getter := newTestSectionGetter(map[string]string{
|
|
"db_type": "mysql",
|
|
"db_host": string(invalidUTF8ByteSequence),
|
|
"db_name": "grafana",
|
|
"db_user": "user",
|
|
"db_password": "password",
|
|
})
|
|
engine, err := getEnginePostgres(getter, nil)
|
|
|
|
assert.Nil(t, engine)
|
|
assert.Error(t, err)
|
|
assert.ErrorIs(t, err, ErrInvalidUTF8Sequence)
|
|
})
|
|
|
|
t.Run("invalid hostport", func(t *testing.T) {
|
|
t.Parallel()
|
|
getter := newTestSectionGetter(map[string]string{
|
|
"db_type": "mysql",
|
|
"db_host": "1:1:1",
|
|
"db_name": "grafana",
|
|
"db_user": "user",
|
|
"db_password": "password",
|
|
})
|
|
engine, err := getEnginePostgres(getter, nil)
|
|
|
|
assert.Nil(t, engine)
|
|
assert.Error(t, err)
|
|
})
|
|
}
|