138 lines
3.4 KiB
Go
138 lines
3.4 KiB
Go
package config
|
|
|
|
import (
|
|
"os"
|
|
"path"
|
|
"testing"
|
|
|
|
"github.com/sirupsen/logrus"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
var defaultConfig = AppConfig{
|
|
LogLevel: defaultLogLevel,
|
|
ServerMode: defaultServerMode,
|
|
Host: defaultServerHost,
|
|
Port: defaultServerPort,
|
|
SockPath: defaultServerSocket,
|
|
StorageType: string(defaultStorageType),
|
|
StorageConfig: &StorageConfig{
|
|
File: defaultStorageFile,
|
|
Host: defaultStorageHost,
|
|
Port: defaultStoragePort,
|
|
Database: defaultStorageDB,
|
|
User: defaultStorageUser,
|
|
Password: defaultStoragePassword,
|
|
Ssl: struct {
|
|
Mode string
|
|
CaFile string
|
|
}{Mode: defaultStorageSSLMode, CaFile: defaultStorageSSLCaFile},
|
|
},
|
|
}
|
|
|
|
func initJson(t *testing.T, content string) string {
|
|
tmpPath := t.TempDir()
|
|
confPath := path.Join(tmpPath, "config.json")
|
|
err := os.WriteFile(confPath, []byte(content), 0o644)
|
|
require.NoError(t, err)
|
|
return confPath
|
|
}
|
|
|
|
func setEnv(t *testing.T, envVars map[string]string) {
|
|
for key, val := range envVars {
|
|
t.Setenv(key, val)
|
|
}
|
|
}
|
|
|
|
func TestDefault(t *testing.T) {
|
|
t.Run("no file", func(t *testing.T) {
|
|
conf, err := New("/this/path/does/not/exist")
|
|
require.NoError(t, err)
|
|
assert.Equal(t, defaultConfig, *conf)
|
|
})
|
|
|
|
t.Run("empty config", func(t *testing.T) {
|
|
confPath := initJson(t, `{}`)
|
|
conf, err := New(confPath)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, defaultConfig, *conf)
|
|
})
|
|
}
|
|
|
|
// Since we still use a JSON conf for the OIDC config, we still need to check this for now
|
|
// But as soon as the config file is not necessary, this will probably disappear
|
|
func TestInvalidJSON(t *testing.T) {
|
|
confPath := initJson(t, "toto")
|
|
errMsg := "failed to parse config file: invalid character 'o' in literal true (expecting 'r')"
|
|
_, err := New(confPath)
|
|
assert.ErrorContains(t, err, errMsg)
|
|
}
|
|
|
|
func TestHostNetMode(t *testing.T) {
|
|
envVars := map[string]string{
|
|
string(varServerMode): string(ModeNet),
|
|
string(varServerHost): "127.0.0.1",
|
|
string(varServerPort): "8888",
|
|
}
|
|
setEnv(t, envVars)
|
|
|
|
conf, err := New("")
|
|
require.NoError(t, err)
|
|
|
|
assert.Equal(t, ModeNet, conf.ServerMode)
|
|
assert.Equal(t, "127.0.0.1", conf.Host)
|
|
assert.Equal(t, 8888, conf.Port)
|
|
}
|
|
|
|
func TestHostSocketMode(t *testing.T) {
|
|
envVars := map[string]string{
|
|
string(varServerMode): string(ModeUnix),
|
|
string(varServerSocket): "/run/polyculeconnect.sock",
|
|
}
|
|
setEnv(t, envVars)
|
|
|
|
conf, err := New("")
|
|
require.NoError(t, err)
|
|
|
|
assert.Equal(t, ModeUnix, conf.ServerMode)
|
|
assert.Equal(t, "/run/polyculeconnect.sock", conf.SockPath)
|
|
}
|
|
|
|
func TestLogLevel(t *testing.T) {
|
|
envVars := map[string]string{
|
|
string(varLogLevel): "error",
|
|
}
|
|
setEnv(t, envVars)
|
|
|
|
conf, err := New("")
|
|
require.NoError(t, err)
|
|
|
|
assert.Equal(t, logrus.ErrorLevel, conf.LogLevel)
|
|
}
|
|
|
|
func TestLogLevelInvalidValue(t *testing.T) {
|
|
envVars := map[string]string{
|
|
string(varLogLevel): "toto",
|
|
}
|
|
setEnv(t, envVars)
|
|
|
|
conf, err := New("")
|
|
require.NoError(t, err)
|
|
|
|
assert.Equal(t, logrus.InfoLevel, conf.LogLevel) // if invalid, no error should occur, but info level should be used
|
|
}
|
|
|
|
func TestSqliteConfig(t *testing.T) {
|
|
envVars := map[string]string{
|
|
string(varStorageType): "sqlite",
|
|
string(varStorageFile): "/data/polyculeconnect.db",
|
|
}
|
|
setEnv(t, envVars)
|
|
|
|
conf, err := New("")
|
|
require.NoError(t, err)
|
|
|
|
assert.Equal(t, string(SQLite), conf.StorageType)
|
|
assert.Equal(t, "/data/polyculeconnect.db", conf.StorageConfig.File)
|
|
}
|