Retry the first WhatsApp connection

[onechat] / probe_test.go

diff --git a/probe_test.go b/probe_test.go
index 265d49d907fb0dd7abc89fe581c25e797688ac9d..1e07fd5fb8962acdb354edb3448953553aba9c0e 100644 (file)
--- a/probe_test.go
+++ b/probe_test.go
@@ -31,7 +31,7 @@ func TestProbeLogHasTheSyntheticMix(t *testing.T) {
        long := 0
        for _, message := range messages {
                counts[message.Kind]++
        long := 0
        for _, message := range messages {
                counts[message.Kind]++

-               require.True(t, kinds[message.Kind], message.Kind)
+               require.True(t, logKinds[message.Kind], message.Kind)

                require.NotEmpty(t, message.Text)
                for _, char := range message.Text {
                        require.True(t, char == '\n' || (char >= ' ' && char <= '~'), "character %q", char)
                require.NotEmpty(t, message.Text)
                for _, char := range message.Text {
                        require.True(t, char == '\n' || (char >= ' ' && char <= '~'), "character %q", char)


@@ -394,7 +394,7 @@ func (g *ProbeWriter) outputLine() string {
 }
 
 // probeLog returns count messages of made-up work on a small bakery order
 }
 
 // probeLog returns count messages of made-up work on a small bakery order

-// system, with the mix of kinds and sizes of syntheticMessage. The same count
+// system, with the mix of kinds and sizes of replayMessage. The same count

 // always gives the same log.
 func probeLog(count int) []ProbeEntry {
        g := ProbeWriter{
 // always gives the same log.
 func probeLog(count int) []ProbeEntry {
        g := ProbeWriter{


@@ -419,7 +419,7 @@ func newTestProbeRun(t *testing.T, server *ModelFake) *ProbeRun {
        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))
+       memory, err := openMemory(t.Context(), root, newTestClock().Now, newCompactor(client, probePrompt, defaultConfig().Compactor), discardLog())

        require.NoError(t, err)
 
        return &ProbeRun{
        require.NoError(t, err)
 
        return &ProbeRun{