mirror of
https://github.com/YspCoder/clawgo.git
synced 2026-04-13 21:57:29 +08:00
33 lines
896 B
Go
33 lines
896 B
Go
package agent
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/YspCoder/clawgo/pkg/bus"
|
|
"github.com/YspCoder/clawgo/pkg/config"
|
|
)
|
|
|
|
func TestNewAgentLoopDisablesNodeP2PByDefault(t *testing.T) {
|
|
cfg := config.DefaultConfig()
|
|
cfg.Agents.Defaults.Workspace = t.TempDir()
|
|
|
|
loop := NewAgentLoop(cfg, bus.NewMessageBus(), stubLLMProvider{}, nil)
|
|
if loop.nodeRouter == nil {
|
|
t.Fatalf("expected node transport to be configured")
|
|
}
|
|
if loop.nodeRouter.P2P != nil {
|
|
t.Fatalf("expected node p2p transport to be disabled by default")
|
|
}
|
|
}
|
|
|
|
func TestNewAgentLoopEnablesNodeP2PWhenConfigured(t *testing.T) {
|
|
cfg := config.DefaultConfig()
|
|
cfg.Agents.Defaults.Workspace = t.TempDir()
|
|
cfg.Gateway.Nodes.P2P.Enabled = true
|
|
|
|
loop := NewAgentLoop(cfg, bus.NewMessageBus(), stubLLMProvider{}, nil)
|
|
if loop.nodeRouter == nil || loop.nodeRouter.P2P == nil {
|
|
t.Fatalf("expected node p2p transport to be enabled")
|
|
}
|
|
}
|