Bind external events to permanent agent IDs

[onechat] / build_run_test.go

diff --git a/build_run_test.go b/build_run_test.go
index aef89225211dca1b4b908c0bfd4e93b786ffa202..7e1bb864682dd1e7e379c1b4f9cdb12373c0bb9b 100644 (file)
--- a/build_run_test.go
+++ b/build_run_test.go
@@ -29,22 +29,22 @@ func TestBuildRunStopsOnFailedCallsOrQuotaGrowth(t *testing.T) {
                want    string
        }{
                {
-                       name:    "no calls",
+                       name:    "continues without recorded calls",
                        records: nil,
                        want:    "",
                },
                {
-                       name:    "growth within the limit",
+                       name:    "continues within the quota limit",
                        records: []ModelUsageRecord{quota("10", "40"), quota("25", "41")},
                        want:    "",
                },
                {
-                       name:    "growth over the limit",
+                       name:    "stops above the quota limit",
                        records: []ModelUsageRecord{quota("10", "40"), quota("12", "55.5")},
                        want:    "x-codex-secondary-used-percent grew from 40 to 55.5",
                },
                {
-                       name: "failed call",
+                       name: "stops after a failed call",
                        records: []ModelUsageRecord{quota("10", "40"), {
                                Status: 400,
                                Error:  "model call: HTTP 400",
@@ -82,6 +82,7 @@ func TestBuildRunStopsWhenTheQuotaGrows(t *testing.T) {
        require.Greater(t, len(records), int(buildQuotaLimit))
 
        var report strings.Builder
+
        require.NoError(t, buildWriteReport(&report, appended, stop, records, sizes))
        require.Contains(t, report.String(), "stopped early: ")
        require.Contains(t, report.String(), "cached (80.0%)")
@@ -111,11 +112,14 @@ func TestBuildRunStopsAtTheCallLimit(t *testing.T) {
 
 func TestBuildRunWaitsOnlyBeforeUserMessages(t *testing.T) {
        gate := make(chan struct{})
+
        var mutex sync.Mutex
+
        running, most := 0, 0
        server := newModelFake(t)
        server.reply = func(w http.ResponseWriter, _ ModelFakeCall) {
                mutex.Lock()
+
                running++
                most = max(most, running)
                mutex.Unlock()
@@ -123,6 +127,7 @@ func TestBuildRunWaitsOnlyBeforeUserMessages(t *testing.T) {
                <-gate
 
                mutex.Lock()
+
                running--
                mutex.Unlock()
 
@@ -210,7 +215,7 @@ func newTestBuildRun(t *testing.T, server *ModelFake) *BuildRun {
        root := t.TempDir()
        usagePath := filepath.Join(root, "usage.jsonl")
        client := newModelClient(server.URL, ModelAPIKey(modelTestKey), "onechat-"+probeName, usagePath)
-       memory, err := openMemory(t.Context(), root, newTestClock().Now, newCompactor(client, probePrompt, defaultConfig().Compactor), discardLog())
+       memory, err := openMemory(t.Context(), root, newTestClock().Now, newCompactor(client, probePrompt, defaultConfig().Compactor, promptTools(), &ImageStore{dir: t.TempDir()}), discardLog())
        require.NoError(t, err)
 
        return &BuildRun{