X-Git-Url: https://git.giacomolli.uk/onechat/blobdiff_plain/a606686934b4028c29dca8e9dae6b2e188cee9c5..723f6e18adc6482ed913e1618c8a76e981720c2b:/rebuild_test.go?ds=sidebyside

diff --git a/rebuild_test.go b/rebuild_test.go
index e3757ea..fc4ff7e 100644
--- a/rebuild_test.go
+++ b/rebuild_test.go
@@ -47,7 +47,9 @@ func TestRebuildMergesDatesAndKeepsAgentFiles(t *testing.T) {
 	before := rebuildTestSnapshot(t, agent.dir)
 	server := newModelFake(t)
 	client := newModelClient(server.URL, ModelAPIKey(modelTestKey), agent.CacheKey(), filepath.Join(t.TempDir(), "usage.jsonl"))
+
 	var report strings.Builder
+
 	added := []DatedMessage{
 		{
 			Kind: "work",
@@ -61,7 +63,7 @@ func TestRebuildMergesDatesAndKeepsAgentFiles(t *testing.T) {
 		},
 	}
 
-	backup, err := rebuild(t.Context(), agent, added, client, newCompactor(client, agent.Prompt, defaultConfig().Compactor), &report)
+	backup, err := rebuild(t.Context(), agent, added, client, newCompactor(client, agent.Prompt, defaultConfig().Compactor, agent.Tools()), &report)
 	require.NoError(t, err)
 	require.Equal(t, before, rebuildTestSnapshot(t, backup))
 
@@ -141,7 +143,7 @@ func TestRebuildFailureAndCancellationKeepOriginal(t *testing.T) {
 				Kind: "work",
 				Text: strings.Repeat("long ", 200),
 				Date: "2025-01-01T00:00:00Z",
-			}}, client, newCompactor(client, agent.Prompt, defaultConfig().Compactor), &strings.Builder{})
+			}}, client, newCompactor(client, agent.Prompt, defaultConfig().Compactor, agent.Tools()), &strings.Builder{})
 			require.Error(t, err)
 			require.Empty(t, backup)
 			require.Equal(t, before, rebuildTestSnapshot(t, agent.dir))
@@ -169,6 +171,7 @@ func TestRebuildKeepsOriginalWhenAnInvisibleMergeFails(t *testing.T) {
 	}
 
 	rebuildTestLog(t, agent.dir, old)
+
 	before := rebuildTestSnapshot(t, agent.dir)
 	server := newModelFake(t)
 	server.reply = func(w http.ResponseWriter, _ ModelFakeCall) {
@@ -176,7 +179,7 @@ func TestRebuildKeepsOriginalWhenAnInvisibleMergeFails(t *testing.T) {
 	}
 
 	client := newModelClient(server.URL, ModelAPIKey(modelTestKey), agent.CacheKey(), filepath.Join(t.TempDir(), "usage.jsonl"))
-	_, err := rebuild(t.Context(), agent, nil, client, newCompactor(client, agent.Prompt, defaultConfig().Compactor), &strings.Builder{})
+	_, err := rebuild(t.Context(), agent, nil, client, newCompactor(client, agent.Prompt, defaultConfig().Compactor, agent.Tools()), &strings.Builder{})
 	require.ErrorContains(t, err, "summary tree has unfinished nodes")
 	require.Equal(t, before, rebuildTestSnapshot(t, agent.dir))
 }