yao/store/store_test.go
Max 5d3c03db1f Update OAuth store management and enhance cache configurations
- Added support for a new OAuth data store in the system store management, allowing for better organization and retrieval of OAuth-related data.
- Updated the OAuth cache store configuration to increase the cache size from 4096 to 8192, improving performance and storage capacity.
- Enhanced test utilities to include the new OAuth store, ensuring comprehensive testing coverage for the updated store management features.
2025-07-20 11:47:57 +08:00

46 lines
945 B
Go

package store
import (
"testing"
"github.com/stretchr/testify/assert"
"github.com/yaoapp/gou/store"
"github.com/yaoapp/yao/config"
"github.com/yaoapp/yao/connector"
"github.com/yaoapp/yao/test"
)
func TestLoad(t *testing.T) {
test.Prepare(t, config.Conf)
defer test.Clean()
loadConnectors(t)
Load(config.Conf)
check(t)
}
func check(t *testing.T) {
ids := map[string]bool{}
for id := range store.Pools {
ids[id] = true
}
assert.True(t, ids["cache"])
assert.True(t, ids["data"])
assert.True(t, ids["share"])
// System stores
assert.True(t, ids["__yao.store"])
assert.True(t, ids["__yao.cache"])
assert.True(t, ids["__yao.oauth.store"])
assert.True(t, ids["__yao.oauth.client"])
assert.True(t, ids["__yao.oauth.cache"])
assert.True(t, ids["__yao.agent.memory"])
assert.True(t, ids["__yao.agent.cache"])
}
func loadConnectors(t *testing.T) {
err := connector.Load(config.Conf)
if err != nil {
t.Fatal(err)
}
}