mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-29 22:06:38 +08:00
ryan
39a81f7723
Context 新增 Config() 访问器，注册表随 Fork 共享（配置声明是进程级事实）， 并在 types.go 导出配置别名与 ConfigGatedPlugin 可选接口，为插件门禁做准备。
93 lines
2.6 KiB
Go
93 lines
2.6 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package core_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"Wavelet/core"
|
|
"Wavelet/core/extpoints"
|
|
)
|
|
|
|
// mapSource implements extpoints.ConfigSource over static maps.
|
|
type mapSource struct {
|
|
values map[string]any
|
|
env map[string]string
|
|
}
|
|
|
|
func (m *mapSource) Lookup(path string) (any, bool) {
|
|
v, ok := m.values[path]
|
|
return v, ok
|
|
}
|
|
|
|
func (m *mapSource) LookupEnv(name string) (string, bool) {
|
|
v, ok := m.env[name]
|
|
return v, ok
|
|
}
|
|
|
|
func (m *mapSource) Describe() string { return "map" }
|
|
|
|
type otelConfig struct {
|
|
SamplingRate float64 `config:"sampling_rate" env:"OTEL_SAMPLING_RATE"`
|
|
}
|
|
|
|
// newOtelRegistry declares the otel section against a source carrying the given file values.
|
|
func newOtelRegistry(t *testing.T, values map[string]any) extpoints.ConfigExtension {
|
|
t.Helper()
|
|
|
|
r := extpoints.NewConfigRegistry(&mapSource{values: values, env: map[string]string{}})
|
|
require.NoError(t, r.Declare("host", extpoints.ConfigBinding{Prefix: "otel", Target: &otelConfig{}}))
|
|
require.NoError(t, r.Resolve())
|
|
return r
|
|
}
|
|
|
|
func TestConfigGetReturnsDeclaredType(t *testing.T) {
|
|
view := newOtelRegistry(t, map[string]any{"otel.sampling_rate": 0.25})
|
|
|
|
rate, err := core.ConfigGet[float64](view, "otel.sampling_rate")
|
|
require.NoError(t, err)
|
|
assert.Equal(t, 0.25, rate)
|
|
}
|
|
|
|
func TestConfigGetRejectsTypeMismatch(t *testing.T) {
|
|
view := newOtelRegistry(t, map[string]any{"otel.sampling_rate": 0.25})
|
|
|
|
text, err := core.ConfigGet[string](view, "otel.sampling_rate")
|
|
require.ErrorIs(t, err, extpoints.ErrConfigType)
|
|
assert.Empty(t, text)
|
|
}
|
|
|
|
func TestConfigGetRejectsUndeclaredKey(t *testing.T) {
|
|
view := newOtelRegistry(t, nil)
|
|
|
|
_, err := core.ConfigGet[float64](view, "otel.unregistered")
|
|
require.ErrorIs(t, err, extpoints.ErrConfigUnknownKey)
|
|
}
|
|
|
|
func TestConfigGetRejectsNilView(t *testing.T) {
|
|
_, err := core.ConfigGet[float64](nil, "otel.sampling_rate")
|
|
require.ErrorIs(t, err, extpoints.ErrConfigNotResolved)
|
|
}
|
|
|
|
func TestContextConfigIsSharedAcrossForks(t *testing.T) {
|
|
ctx := core.NewContext(nil)
|
|
child := ctx.Fork()
|
|
|
|
require.NotNil(t, ctx.Config())
|
|
assert.Same(t, ctx.Config(), child.Config(), "configuration declarations are process-wide facts")
|
|
|
|
require.NoError(t, child.Config().Declare("cache",
|
|
extpoints.ConfigBinding{Prefix: "otel", Target: &otelConfig{}}))
|
|
|
|
declared := false
|
|
for _, entry := range ctx.Config().Entries() {
|
|
declared = declared || entry.Key == "otel.sampling_rate"
|
|
}
|
|
assert.True(t, declared, "a declaration made in a plugin scope must be visible to the root")
|
|
assert.False(t, ctx.Config().Resolved())
|
|
}
|