Send stored images to WhatsApp chats

[onechat] / build_run_test.go

diff --git a/build_run_test.go b/build_run_test.go
index aef89225211dca1b4b908c0bfd4e93b786ffa202..ef9c3c0fb48c98582888f440854a32d5577e6463 100644 (file)
--- a/build_run_test.go
+++ b/build_run_test.go
@@ -82,6 +82,7 @@ func TestBuildRunStopsWhenTheQuotaGrows(t *testing.T) {
        require.Greater(t, len(records), int(buildQuotaLimit))
 
        var report strings.Builder
        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%)")
        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{})
 
 func TestBuildRunWaitsOnlyBeforeUserMessages(t *testing.T) {
        gate := make(chan struct{})

+

        var mutex sync.Mutex
        var mutex sync.Mutex

+

        running, most := 0, 0
        server := newModelFake(t)
        server.reply = func(w http.ResponseWriter, _ ModelFakeCall) {
                mutex.Lock()
        running, most := 0, 0
        server := newModelFake(t)
        server.reply = func(w http.ResponseWriter, _ ModelFakeCall) {
                mutex.Lock()

+

                running++
                most = max(most, running)
                mutex.Unlock()
                running++
                most = max(most, running)
                mutex.Unlock()


@@ -123,6 +127,7 @@ func TestBuildRunWaitsOnlyBeforeUserMessages(t *testing.T) {
                <-gate
 
                mutex.Lock()
                <-gate
 
                mutex.Lock()

+

                running--
                mutex.Unlock()
 
                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)
        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{
        require.NoError(t, err)
 
        return &BuildRun{