632c0f5674
- Replaced all instances of `ctx.Space` with `ctx.Memory.Context` in the context management code, ensuring a more structured approach to handling temporary request-scoped data. - Updated related test cases to reflect the changes in context memory usage, enhancing the reliability and clarity of tests. - Removed the deprecated `Space` references and adjusted comments and documentation to align with the new memory management strategy.
58 lines
1.3 KiB
Go
58 lines
1.3 KiB
Go
package store
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"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)
|
|
|
|
// Remove the data store (For cleaning the stores whitch created by the test)
|
|
var path = filepath.Join(config.Conf.DataRoot, "stores")
|
|
os.RemoveAll(path)
|
|
|
|
err := Load(config.Conf)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
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.user"])
|
|
assert.True(t, ids["__yao.agent.memory.team"])
|
|
assert.True(t, ids["__yao.agent.memory.chat"])
|
|
assert.True(t, ids["__yao.agent.memory.context"])
|
|
assert.True(t, ids["__yao.agent.cache"])
|
|
}
|
|
|
|
func loadConnectors(t *testing.T) {
|
|
err := connector.Load(config.Conf)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|