mirror of
https://github.com/SigNoz/signoz.git
synced 2026-04-26 22:00:23 +01:00
3e241944e7
* chore(sqlstore): added connection max lifetime param support * chore(sqlstore): fix test * chore(sqlstore): default to 0 for conn_max_lifetime * chore(sqlstore): rename the config * chore(sqlstore): rename the config
59 lines
1.4 KiB
Go
59 lines
1.4 KiB
Go
package sqlstore
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/SigNoz/signoz/pkg/config"
|
|
"github.com/SigNoz/signoz/pkg/config/envprovider"
|
|
"github.com/SigNoz/signoz/pkg/factory"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestNewWithEnvProvider(t *testing.T) {
|
|
t.Setenv("SIGNOZ_SQLSTORE_PROVIDER", "sqlite")
|
|
t.Setenv("SIGNOZ_SQLSTORE_SQLITE_PATH", "/tmp/test.db")
|
|
t.Setenv("SIGNOZ_SQLSTORE_SQLITE_MODE", "wal")
|
|
t.Setenv("SIGNOZ_SQLSTORE_SQLITE_BUSY__TIMEOUT", "5s")
|
|
t.Setenv("SIGNOZ_SQLSTORE_SQLITE_TRANSACTION__MODE", "immediate")
|
|
t.Setenv("SIGNOZ_SQLSTORE_MAX__OPEN__CONNS", "50")
|
|
t.Setenv("SIGNOZ_SQLSTORE_MAX__CONN__LIFETIME", "3h")
|
|
|
|
conf, err := config.New(
|
|
context.Background(),
|
|
config.ResolverConfig{
|
|
Uris: []string{"env:"},
|
|
ProviderFactories: []config.ProviderFactory{
|
|
envprovider.NewFactory(),
|
|
},
|
|
},
|
|
[]factory.ConfigFactory{
|
|
NewConfigFactory(),
|
|
},
|
|
)
|
|
require.NoError(t, err)
|
|
|
|
actual := &Config{}
|
|
err = conf.Unmarshal("sqlstore", actual)
|
|
require.NoError(t, err)
|
|
|
|
expected := &Config{
|
|
Provider: "sqlite",
|
|
Connection: ConnectionConfig{
|
|
MaxOpenConns: 50,
|
|
MaxConnLifetime: time.Hour * 3,
|
|
},
|
|
Sqlite: SqliteConfig{
|
|
Path: "/tmp/test.db",
|
|
Mode: "wal",
|
|
BusyTimeout: 5 * time.Second,
|
|
TransactionMode: "immediate",
|
|
},
|
|
}
|
|
|
|
assert.Equal(t, expected, actual)
|
|
assert.NoError(t, actual.Validate())
|
|
}
|