Michael McGuinness
7001ca854c
Queuing Changes and Cfg Testing * staarting * staarting * startedpush * note * save * mocking * removederrs * fixtests * cleanuperrs * newenvsetup * preppingtests * queue * mmovetocfgpassunittests * sortoutconfig * passinginteg * deps * fixtests
80 lines
1.6 KiB
Go
80 lines
1.6 KiB
Go
package database_test
|
|
|
|
import (
|
|
"queryorchestration/internal/serviceconfig/database"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestGetDBOpts(t *testing.T) {
|
|
cfg := database.DBConfig{}
|
|
|
|
opts := cfg.GetDBOpts()
|
|
assert.Equal(t, map[string]string{}, opts)
|
|
|
|
cfg.DBNoSSL = true
|
|
opts = cfg.GetDBOpts()
|
|
assert.Equal(t, map[string]string{"sslmode": "disable"}, opts)
|
|
}
|
|
|
|
func TestGetDBOptsString(t *testing.T) {
|
|
cfg := database.DBConfig{}
|
|
|
|
opts := cfg.GetDBOptsString()
|
|
assert.Equal(t, "", opts)
|
|
|
|
cfg.DBNoSSL = true
|
|
opts = cfg.GetDBOptsString()
|
|
assert.Equal(t, "sslmode=disable", opts)
|
|
}
|
|
|
|
func TestGetDBDriver(t *testing.T) {
|
|
cfg := database.DBConfig{}
|
|
|
|
driver := cfg.GetDBDriver()
|
|
assert.Equal(t, "postgres", driver)
|
|
}
|
|
|
|
func TestGetDBBaseURI(t *testing.T) {
|
|
cfg := database.DBConfig{}
|
|
|
|
uri := cfg.GetDBBaseURI()
|
|
assert.Equal(t, "postgres://:@:0/", uri)
|
|
|
|
cfg.DBHost = "host"
|
|
cfg.DBPort = 123
|
|
cfg.DBUser = "user"
|
|
cfg.DBSecret = "pass"
|
|
uri = cfg.GetDBBaseURI()
|
|
assert.Equal(t, "postgres://user:pass@host:123/", uri)
|
|
}
|
|
|
|
func TestGetDBURI(t *testing.T) {
|
|
cfg := database.DBConfig{}
|
|
|
|
uri := cfg.GetDBURI()
|
|
assert.Equal(t, "postgres://:@:0/?", uri)
|
|
|
|
cfg.DBHost = "host"
|
|
cfg.DBPort = 123
|
|
cfg.DBUser = "user"
|
|
cfg.DBSecret = "pass"
|
|
cfg.DBName = "name"
|
|
cfg.DBNoSSL = true
|
|
uri = cfg.GetDBURI()
|
|
assert.Equal(t, "postgres://user:pass@host:123/name?sslmode=disable", uri)
|
|
}
|
|
|
|
func TestGetDBName(t *testing.T) {
|
|
cfg := database.DBConfig{}
|
|
|
|
name := cfg.GetDBName()
|
|
assert.Equal(t, "", name)
|
|
|
|
cfg.DBName = "name"
|
|
name = cfg.GetDBName()
|
|
assert.Equal(t, "name", name)
|
|
assert.Equal(t, cfg.DBName, name)
|
|
}
|