43 lines
775 B
Go
43 lines
775 B
Go
package command
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/yaoapp/yao/config"
|
|
"github.com/yaoapp/yao/neo/command/driver"
|
|
"github.com/yaoapp/yao/test"
|
|
)
|
|
|
|
func TestLoad(t *testing.T) {
|
|
test.Prepare(t, config.Conf)
|
|
defer test.Clean()
|
|
|
|
Commands = map[string]*Command{}
|
|
Load(config.Conf)
|
|
check(t)
|
|
}
|
|
|
|
func TestLoadWithStore(t *testing.T) {
|
|
test.Prepare(t, config.Conf)
|
|
defer test.Clean()
|
|
|
|
Commands = map[string]*Command{}
|
|
mem, err := driver.NewMemory("gpt-3_5-turbo", nil)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
SetStore(mem)
|
|
Load(config.Conf)
|
|
check(t)
|
|
}
|
|
|
|
func check(t *testing.T) {
|
|
ids := map[string]bool{}
|
|
for id := range Commands {
|
|
ids[id] = true
|
|
}
|
|
assert.True(t, ids["table.data"])
|
|
assert.GreaterOrEqual(t, len(Autopilots), 1)
|
|
}
|