mirror of
https://github.com/sipeed/picoclaw.git
synced 2026-06-12 18:08:54 +00:00
30 lines
569 B
Go
30 lines
569 B
Go
package tools
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/sipeed/picoclaw/pkg/providers"
|
|
)
|
|
|
|
func TestSubagentManager_SetMediaResolver_StoresResolver(t *testing.T) {
|
|
manager := NewSubagentManager(nil, "gpt-test", "/tmp")
|
|
|
|
called := false
|
|
manager.SetMediaResolver(func(msgs []providers.Message) []providers.Message {
|
|
called = true
|
|
return msgs
|
|
})
|
|
|
|
manager.mu.RLock()
|
|
got := manager.mediaResolver
|
|
manager.mu.RUnlock()
|
|
|
|
if got == nil {
|
|
t.Fatal("expected mediaResolver to be set")
|
|
}
|
|
|
|
if called {
|
|
t.Fatal("resolver should not be called during SetMediaResolver")
|
|
}
|
|
}
|