]> git.giacomolli.uk - onechat/blobdiff - build_run_test.go
Share general review and documentation rules
[onechat] / build_run_test.go
index fc8b4b7e916da95d69767c62c2ba3e2e03b03428..7e1bb864682dd1e7e379c1b4f9cdb12373c0bb9b 100644 (file)
@@ -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",
@@ -215,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{