X-Git-Url: https://git.giacomolli.uk/onechat/blobdiff_plain/38a9b0502a1ccefc6ccb3472c4a21805b222caa6..e73e445bd78ba98271da6b5165f132299fca04b1:/subagent_run_test.go

diff --git a/subagent_run_test.go b/subagent_run_test.go
index 9b8b8c5..a39d81c 100644
--- a/subagent_run_test.go
+++ b/subagent_run_test.go
@@ -66,6 +66,7 @@ func TestSpawnRunsASubagentInTheBackground(t *testing.T) {
 		Source: name,
 		Status: subagentFinished,
 	}, test.events.LastEvent("work"))
+	require.Equal(t, "subagent "+name+" started\nsubagent "+name+" done after 0s\n", test.logs.String())
 
 	var first, second *ModelFakeCall
 	for n := range server.Calls() {
@@ -188,6 +189,7 @@ func TestFullContextGivesAFailedReport(t *testing.T) {
 		"reply: It failed.",
 	}, test.Log(t))
 	require.Equal(t, subagentFailed, test.events.LastEvent("work").Status)
+	require.Equal(t, "subagent "+name+" started\nsubagent "+name+" failed after 0s: model call: HTTP 400: "+subagentTestFullContext+"\n", test.logs.String())
 }
 
 func TestSubagentHasItsOwnEffortAndNoSpawn(t *testing.T) {
@@ -372,6 +374,7 @@ func TestStopEndsASubagentAndItsCommand(t *testing.T) {
 		"work: " + subagentReport(name, []string{"stopped"}),
 	}, test.Log(t))
 	require.Equal(t, subagentStopped, test.events.LastEvent("work").Status)
+	require.Equal(t, "subagent "+name+" started\nsubagent "+name+" stopped after 0s\n", test.logs.String())
 
 	log, err := subagent.Show()
 	require.NoError(t, err)
@@ -511,13 +514,13 @@ func subagentTestStart(t *testing.T, agent *Agent, server *ModelFake, events *Tu
 
 	agent.config.Workdir = t.TempDir()
 	clock := newTestClock()
-	memory, err := openMemory(t.Context(), agent.dir, clock.Now, newFakeSummarizer())
+	memory, err := openMemory(t.Context(), agent.dir, clock.Now, newFakeSummarizer(), discardLog())
 	require.NoError(t, err)
 
 	subagents := newSubagents(filepath.Join(agent.dir, subagentDirName), clock.Now, subagentNameRandom)
 	client := newModelClient(server.URL, ModelAPIKey(modelTestKey), agent.CacheKey(), filepath.Join(agent.dir, "usage.jsonl"))
 	whatsapp, _, _ := whatsappStartTest(t, t.TempDir(), clock)
-	loop, err := newTurnLoop(agent, memory, subagents, whatsapp, client, clock.Now, events.Add)
+	loop, err := newTurnLoop(agent, memory, subagents, whatsapp, newTestSchedules(t, clock.Now), client, clock.Now, discardLog(), events.Add)
 	require.NoError(t, err)
 
 	ctx, cancel := context.WithCancel(t.Context())