mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-10-01 06:36:38 +08:00
ryan
c77b5358e2
配置读取框架的内核侧抽象：插件用带 config/env/default/autoEnable/secret tag 的结构体声明自己读哪些字段，注册表按 key 归集并对重复声明做一致性 校验，为后续按声明解析与门禁求值提供基础。
87 lines
2.8 KiB
Go
87 lines
2.8 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package extpoints_test
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"Wavelet/core/extpoints"
|
|
)
|
|
|
|
// fakeSource is an in-memory extpoints.ConfigSource used by configuration engine tests.
|
|
type fakeSource struct {
|
|
values map[string]any
|
|
env map[string]string
|
|
}
|
|
|
|
func newFakeSource() *fakeSource {
|
|
return &fakeSource{values: map[string]any{}, env: map[string]string{}}
|
|
}
|
|
|
|
func (f *fakeSource) Lookup(path string) (any, bool) {
|
|
v, ok := f.values[path]
|
|
return v, ok
|
|
}
|
|
|
|
func (f *fakeSource) LookupEnv(name string) (string, bool) {
|
|
v, ok := f.env[name]
|
|
return v, ok
|
|
}
|
|
|
|
func (f *fakeSource) Describe() string { return "fake" }
|
|
|
|
// redisConfig mirrors how a plugin declares the configuration it reads.
|
|
type redisConfig struct {
|
|
Enabled bool `config:"enabled" env:"REDIS_ENABLED" default:"false" autoEnable:"REDIS_ADDR"`
|
|
Addrs []string `config:"addrs" env:"REDIS_ADDR"`
|
|
DB int `config:"db" env:"REDIS_DB"`
|
|
KeyPrefix string `config:"key_prefix" env:"REDIS_KEY_PREFIX"`
|
|
Dial time.Duration `config:"dial_timeout" env:"REDIS_DIAL_TIMEOUT"`
|
|
Ignored string `config:"-"`
|
|
private string
|
|
}
|
|
|
|
func TestDeclareRegistersTaggedLeafKeys(t *testing.T) {
|
|
r := extpoints.NewConfigRegistry(newFakeSource())
|
|
|
|
require.NoError(t, r.Declare("cache", extpoints.ConfigBinding{Prefix: "redis", Target: &redisConfig{}}))
|
|
|
|
keys := make([]string, 0)
|
|
for _, e := range r.Entries() {
|
|
keys = append(keys, e.Key)
|
|
}
|
|
assert.Equal(t, []string{
|
|
"redis.addrs", "redis.db", "redis.dial_timeout", "redis.enabled", "redis.key_prefix",
|
|
}, keys)
|
|
}
|
|
|
|
func TestDeclareRejectsNonStructPointerTarget(t *testing.T) {
|
|
r := extpoints.NewConfigRegistry(newFakeSource())
|
|
|
|
assert.ErrorIs(t, r.Declare("cache", extpoints.ConfigBinding{Prefix: "redis", Target: redisConfig{}}),
|
|
extpoints.ErrConfigTarget)
|
|
assert.ErrorIs(t, r.Declare("cache", extpoints.ConfigBinding{Prefix: "redis", Target: (*redisConfig)(nil)}),
|
|
extpoints.ErrConfigTarget)
|
|
}
|
|
|
|
func TestDeclareAllowsIdenticalDuplicateAndRejectsConflictingMetadata(t *testing.T) {
|
|
r := extpoints.NewConfigRegistry(newFakeSource())
|
|
binding := extpoints.ConfigBinding{Prefix: "redis", Target: &redisConfig{}}
|
|
require.NoError(t, r.Declare("cache", binding))
|
|
require.NoError(t, r.Declare("cache_memory", binding), "identical shared declarations must be allowed")
|
|
|
|
type conflictingConfig struct {
|
|
Enabled bool `config:"enabled" env:"REDIS_ON" default:"true"`
|
|
}
|
|
err := r.Declare("driver_http", extpoints.ConfigBinding{Prefix: "redis", Target: &conflictingConfig{}})
|
|
require.ErrorIs(t, err, extpoints.ErrConfigConflict)
|
|
assert.Contains(t, err.Error(), "redis.enabled")
|
|
assert.Contains(t, err.Error(), "cache")
|
|
assert.Contains(t, err.Error(), "driver_http")
|
|
}
|