Builds: design 47 G §576, §518, §622, A-49 Claude-Session: https://claude.ai/code/session_01NcbgfGKaxw984XyAPfLm4h
69 lines
2.4 KiB
Go
69 lines
2.4 KiB
Go
package launch
|
|
|
|
import (
|
|
"encoding/json"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"sync"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
// puttingBundle asks the capability words, and once they are answered with put-sessions sends the one chunk of a put
|
|
// session, as mesh-sdk's bundle does; it then ends, which closes its channel.
|
|
const puttingBundle = `#!/bin/sh
|
|
while IFS= read -r line; do
|
|
id=$(printf '%s' "$line" | sed -n 's/.*"id":\([0-9]*\)[,}].*/\1/p')
|
|
case "$line" in
|
|
*'"method":"initialize"'*) printf '{"jsonrpc":"2.0","id":%s,"result":{}}\n' "$id" ;;
|
|
*'"method":"tools/list"'*) printf '{"jsonrpc":"2.0","id":%s,"result":{"tools":[]}}\n' "$id"
|
|
printf '{"jsonrpc":"2.0","id":"w1","method":"mesh/capabilities","params":{}}\n' ;;
|
|
*'"id":"w1"'**put-sessions*) printf '{"jsonrpc":"2.0","id":"p1","method":"mesh/objects.put-chunk","params":{"objects":"o","name":"n","call":"c","seq":1,"last":true,"data":"aGk="}}\n' ;;
|
|
*'"id":"p1"'*) exit 0 ;;
|
|
esac
|
|
done
|
|
`
|
|
|
|
type sessionBus struct {
|
|
subscribing
|
|
mu sync.Mutex
|
|
chunks []string
|
|
dropped chan struct{}
|
|
}
|
|
|
|
func (b *sessionBus) CapabilityWords() []string { return []string{"put-sessions"} }
|
|
func (b *sessionBus) BusState(method string, params json.RawMessage) (json.RawMessage, error) {
|
|
b.mu.Lock()
|
|
b.chunks = append(b.chunks, method+" "+string(params))
|
|
b.mu.Unlock()
|
|
return json.RawMessage(`{"name":"n"}`), nil
|
|
}
|
|
func (b *sessionBus) DropSessions() { close(b.dropped) }
|
|
|
|
// Design 47 G §576, §518: a bundle reads the runner's capability words with mesh/capabilities, then runs its put session
|
|
// through the runner's put-chunk, and the sessions of its channel are dropped when the channel closes.
|
|
func TestABundleReadsTheCapabilityWordsRunsAPutSessionAndItsChannelsClosingDropsThem(t *testing.T) {
|
|
dir := t.TempDir()
|
|
entry := filepath.Join(dir, "bundle")
|
|
if err := os.WriteFile(entry, []byte(puttingBundle), 0o755); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
b := &sessionBus{dropped: make(chan struct{})}
|
|
l, err := Start("putter", entry, os.Environ(), b, func(string, ...any) {})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
t.Cleanup(l.Stop)
|
|
select {
|
|
case <-b.dropped:
|
|
case <-time.After(5 * time.Second):
|
|
t.Fatal("the bundle's sessions were not dropped when its channel closed")
|
|
}
|
|
b.mu.Lock()
|
|
defer b.mu.Unlock()
|
|
if len(b.chunks) != 1 || !strings.HasPrefix(b.chunks[0], "mesh/objects.put-chunk ") || !strings.Contains(b.chunks[0], `"seq":1`) {
|
|
t.Fatalf("the runner took %v", b.chunks)
|
|
}
|
|
}
|