mirror of
https://git.mirrors.martin98.com/https://github.com/SigNoz/signoz
synced 2025-07-29 17:52:01 +08:00
df5ab64c83
* fix: use common timeout middleware * fix: use apiserver factory for config * fix: add backward compatibility for old variables * fix: remove apiserver provider and use config directly * fix: remove apiserver interface * fix: address comments * fix: address minor comments * fix: address minor comments
52 lines
1.0 KiB
Go
52 lines
1.0 KiB
Go
package apiserver
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"go.signoz.io/signoz/pkg/config"
|
|
"go.signoz.io/signoz/pkg/config/envprovider"
|
|
"go.signoz.io/signoz/pkg/factory"
|
|
)
|
|
|
|
func TestNewWithEnvProvider(t *testing.T) {
|
|
t.Setenv("SIGNOZ_APISERVER_TIMEOUT_DEFAULT", "70s")
|
|
t.Setenv("SIGNOZ_APISERVER_TIMEOUT_MAX", "700s")
|
|
t.Setenv("SIGNOZ_APISERVER_TIMEOUT_EXCLUDED__ROUTES", "/excluded1,/excluded2")
|
|
|
|
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("apiserver", actual)
|
|
|
|
require.NoError(t, err)
|
|
|
|
expected := &Config{
|
|
Timeout: Timeout{
|
|
Default: 70 * time.Second,
|
|
Max: 700 * time.Second,
|
|
ExcludedRoutes: []string{
|
|
"/excluded1",
|
|
"/excluded2",
|
|
},
|
|
},
|
|
}
|
|
|
|
assert.Equal(t, expected, actual)
|
|
}
|