photoprism/internal/config/db_test.go
2021-02-20 13:13:59 +01:00

110 lines
3.1 KiB
Go

package config
import (
"testing"
"github.com/stretchr/testify/assert"
)
func TestConfig_DatabaseDriver(t *testing.T) {
c := NewConfig(CliTestContext())
driver := c.DatabaseDriver()
assert.Equal(t, SQLite, driver)
}
func TestConfig_ParseDatabaseDsn(t *testing.T) {
c := NewConfig(CliTestContext())
c.options.DatabaseDsn = "foo:b@r@tcp(honeypot:1234)/baz?charset=utf8mb4,utf8&parseTime=true"
assert.Equal(t, "honeypot:1234", c.DatabaseServer())
assert.Equal(t, "honeypot", c.DatabaseHost())
assert.Equal(t, 1234, c.DatabasePort())
assert.Equal(t, "baz", c.DatabaseName())
assert.Equal(t, "foo", c.DatabaseUser())
assert.Equal(t, "b@r", c.DatabasePassword())
}
func TestConfig_DatabaseServer(t *testing.T) {
c := NewConfig(CliTestContext())
assert.Equal(t, "localhost", c.DatabaseServer())
c.options.DatabaseServer = "test"
assert.Equal(t, "test", c.DatabaseServer())
}
func TestConfig_DatabaseHost(t *testing.T) {
c := NewConfig(CliTestContext())
assert.Equal(t, "localhost", c.DatabaseHost())
}
func TestConfig_DatabasePort(t *testing.T) {
c := NewConfig(CliTestContext())
assert.Equal(t, 3306, c.DatabasePort())
}
func TestConfig_DatabasePortString(t *testing.T) {
c := NewConfig(CliTestContext())
assert.Equal(t, "3306", c.DatabasePortString())
}
func TestConfig_DatabaseName(t *testing.T) {
c := NewConfig(CliTestContext())
assert.Equal(t, "photoprism", c.DatabaseName())
}
func TestConfig_DatabaseUser(t *testing.T) {
c := NewConfig(CliTestContext())
assert.Equal(t, "photoprism", c.DatabaseUser())
}
func TestConfig_DatabasePassword(t *testing.T) {
c := NewConfig(CliTestContext())
assert.Equal(t, "", c.DatabasePassword())
}
func TestConfig_DatabaseDsn(t *testing.T) {
c := NewConfig(CliTestContext())
dsn := c.DatabaseDriver()
assert.Equal(t, SQLite, dsn)
c.options.DatabaseDsn = ""
c.options.DatabaseDriver = "MariaDB"
assert.Equal(t, "photoprism:@tcp(localhost)/photoprism?charset=utf8mb4,utf8&collation=utf8mb4_unicode_ci&parseTime=true", c.DatabaseDsn())
c.options.DatabaseDriver = "tidb"
assert.Equal(t, "/go/src/github.com/photoprism/photoprism/storage/testdata/index.db", c.DatabaseDsn())
c.options.DatabaseDriver = "Postgres"
assert.Equal(t, "/go/src/github.com/photoprism/photoprism/storage/testdata/index.db", c.DatabaseDsn())
c.options.DatabaseDriver = "SQLite"
assert.Equal(t, "/go/src/github.com/photoprism/photoprism/storage/testdata/index.db", c.DatabaseDsn())
c.options.DatabaseDriver = ""
assert.Equal(t, "/go/src/github.com/photoprism/photoprism/storage/testdata/index.db", c.DatabaseDsn())
}
func TestConfig_DatabaseConns(t *testing.T) {
c := NewConfig(CliTestContext())
c.options.DatabaseConns = 28
assert.Equal(t, 28, c.DatabaseConns())
c.options.DatabaseConns = 3000
assert.Equal(t, 1024, c.DatabaseConns())
}
func TestConfig_DatabaseConnsIdle(t *testing.T) {
c := NewConfig(CliTestContext())
c.options.DatabaseConnsIdle = 14
c.options.DatabaseConns = 28
assert.Equal(t, 14, c.DatabaseConnsIdle())
c.options.DatabaseConnsIdle = -55
assert.Greater(t, c.DatabaseConnsIdle(), 8)
c.options.DatabaseConnsIdle = 35
assert.Equal(t, 28, c.DatabaseConnsIdle())
}