Move whatsapp_find to its own file

[onechat] / whatsapp_test.go

diff --git a/whatsapp_test.go b/whatsapp_test.go
index e0018e5ba2a744d99f6130ed03b269dd366ebfc1..2bdb4cc349e77f8f1fa901f98c11c3641fc4b611 100644 (file)
--- a/whatsapp_test.go
+++ b/whatsapp_test.go
@@ -1,7 +1,10 @@
 package main
 
 import (
 package main
 
 import (

+       "crypto/sha256"
+       "encoding/hex"

        "errors"
        "errors"

+       "net/http"

        "os"
        "path/filepath"
        "strings"
        "os"
        "path/filepath"
        "strings"


@@ -134,6 +137,14 @@ func TestWhatsAppPairsWithACode(t *testing.T) {
        items <- whatsmeow.QRChannelSuccess
        asking.WaitFor(t, "WhatsApp is linked to +15550001111.")
 
        items <- whatsmeow.QRChannelSuccess
        asking.WaitFor(t, "WhatsApp is linked to +15550001111.")
 

+       // The library connects again after the link with the context of the
+       // connection, so the end of the pairing must not end it.
+       connections := fake.Connections()
+       require.Len(t, connections, 1)
+       require.Never(t, func() bool {
+               return connections[0].Err() != nil
+       }, 200*time.Millisecond, time.Millisecond)
+

        require.Equal(t, []string{whatsappTestNumber}, fake.Phones())
        require.Equal(t, "WhatsApp is linked and not connected yet.", whatsapp.Status())
        require.Equal(t, "[connected to OneChat]\n", other.Output())
        require.Equal(t, []string{whatsappTestNumber}, fake.Phones())
        require.Equal(t, "WhatsApp is linked and not connected yet.", whatsapp.Status())
        require.Equal(t, "[connected to OneChat]\n", other.Output())


@@ -314,7 +325,7 @@ func TestWhatsAppLongDisconnectionGivesOneNotice(t *testing.T) {
        whatsappTestSession(t, dir, true)
        clock := newTestClock()
        whatsapp, fake, _ := whatsappStartTest(t, dir, clock)
        whatsappTestSession(t, dir, true)
        clock := newTestClock()
        whatsapp, fake, _ := whatsappStartTest(t, dir, clock)

-       whatsapp.noticeDelay = 50 * time.Millisecond
+       whatsapp.noticeDelay = 300 * time.Millisecond

        socket, _ := daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
        client := terminalStartTestClient(t, socket)
        client.WaitFor(t, "[connected to OneChat]")
        socket, _ := daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
        client := terminalStartTestClient(t, socket)
        client.WaitFor(t, "[connected to OneChat]")


@@ -323,7 +334,7 @@ func TestWhatsAppLongDisconnectionGivesOneNotice(t *testing.T) {
        require.True(t, fake.Emit(&events.Connected{}))
        require.True(t, fake.Emit(&events.Disconnected{}))
        require.True(t, fake.Emit(&events.Connected{}))
        require.True(t, fake.Emit(&events.Connected{}))
        require.True(t, fake.Emit(&events.Disconnected{}))
        require.True(t, fake.Emit(&events.Connected{}))

-       time.Sleep(100 * time.Millisecond)
+       time.Sleep(400 * time.Millisecond)

        require.Equal(t, "[connected to OneChat]\n", client.Output())
 
        clock.now = clock.now.Add(time.Hour)
        require.Equal(t, "[connected to OneChat]\n", client.Output())
 
        clock.now = clock.now.Add(time.Hour)


@@ -331,7 +342,7 @@ func TestWhatsAppLongDisconnectionGivesOneNotice(t *testing.T) {
        require.True(t, fake.Emit(&events.Disconnected{}))
        client.WaitFor(t, "WhatsApp is disconnected since 2026-01-01 13:00. OneChat tries again.\n")
        require.True(t, fake.Emit(&events.Disconnected{}))
        require.True(t, fake.Emit(&events.Disconnected{}))
        client.WaitFor(t, "WhatsApp is disconnected since 2026-01-01 13:00. OneChat tries again.\n")
        require.True(t, fake.Emit(&events.Disconnected{}))

-       time.Sleep(100 * time.Millisecond)
+       time.Sleep(400 * time.Millisecond)

 
        require.True(t, fake.Emit(&events.Connected{}))
        client.WaitFor(t, "WhatsApp is connected again.\n")
 
        require.True(t, fake.Emit(&events.Connected{}))
        client.WaitFor(t, "WhatsApp is connected again.\n")


@@ -448,22 +459,199 @@ func TestWhatsAppCommandArgs(t *testing.T) {
        }
 }
 
        }
 }
 

-// whatsappTestLogTexts returns the texts of the messages in the log of memory,
-// after the log holds at least one message.
-func whatsappTestLogTexts(t *testing.T, memory *Memory) []string {
-       t.Helper()
+func TestSubagentHasNoWhatsAppTools(t *testing.T) {
+       box := &Toolbox{}
+       require.Equal(t, `error: unknown tool "whatsapp_find"`, box.run(t.Context(), "whatsapp_find", `{"name":"ana"}`))
+       require.Equal(t, `error: unknown tool "whatsapp_send"`, box.run(t.Context(), "whatsapp_send", `{"id":"a","chat":"15550002222@s.whatsapp.net","text":"hi"}`))
+}

 
 

-       require.Eventually(t, func() bool {
-               return memory.Len() > 0
-       }, 10*time.Second, time.Millisecond)
+func TestWhatsAppSendSendsOnceForEachID(t *testing.T) {
+       whatsapp, fake := whatsappStartTestLinked(t)
+       whatsappTestContacts(t, whatsapp)
+       daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
+       whatsappTestConnect(t, fake)
+
+       answer, err := whatsapp.Send(t.Context(), "keys", whatsappTestAna.String(), "The keys are under the mat.")
+       require.NoError(t, err)
+       require.Equal(t, "Sent to Ana Souza at 12:05.", answer)
+
+       answer, err = whatsapp.Send(t.Context(), "keys", whatsappTestAna.String(), "The keys are under the mat.")
+       require.NoError(t, err)
+       require.Equal(t, "Sent to Ana Souza at 12:05. This is the answer of the first call with this id: nothing was sent again.", answer)
+
+       _, err = whatsapp.Send(t.Context(), "keys", whatsappTestAna.String(), "Another text.")
+       require.EqualError(t, err, "the id keys belongs to another message: give a new id for each message")

 
 

-       var texts []string
-       for id := range memory.Len() {
-               message, err := memory.Message(id)
+       require.Equal(t, []WhatsAppFakeSend{
+               {
+                       to:   whatsappTestAna.String(),
+                       text: "The keys are under the mat.",
+                       id:   "MSG1",
+               },
+       }, fake.Sent())
+
+       again, err := openWhatsAppState(filepath.Join(whatsapp.dir, whatsappStateFileName), whatsappBatchQuiet, whatsappBatchWindow)
+       require.NoError(t, err)
+
+       record, found := again.Send("keys")
+       require.True(t, found)
+       require.Equal(t, whatsappSent, record.State)
+       require.Equal(t, "MSG1", record.MessageID)
+}
+
+func TestWhatsAppSendToAGroup(t *testing.T) {
+       whatsapp, fake := whatsappStartTestLinked(t)
+       fake.groups = []*types.GroupInfo{
+               {
+                       JID: whatsappTestFamily,
+                       GroupName: types.GroupName{
+                               Name: "Family",
+                       },
+               },
+       }
+
+       daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
+       whatsappTestConnect(t, fake)
+
+       answer, err := whatsapp.Send(t.Context(), "cake", whatsappTestFamily.String(), "I bring the cake.")
+       require.NoError(t, err)
+       require.Equal(t, "Sent to Family at 12:05.", answer)
+}
+
+func TestWhatsAppSendThatFails(t *testing.T) {
+       whatsapp, fake := whatsappStartTestLinked(t)
+       fake.sendErr = errors.New("server error for " + whatsappTestNumber)
+       daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
+       whatsappTestConnect(t, fake)
+
+       for range 2 {
+               answer, err := whatsapp.Send(t.Context(), "keys", whatsappTestAna.String(), "hi")

                require.NoError(t, err)
                require.NoError(t, err)

+               require.True(t, strings.HasPrefix(answer, "The send failed: server error for OWN."), answer)
+       }

 
 

-               texts = append(texts, message.Text)
+       require.Len(t, fake.Sent(), 1)
+}
+
+func TestWhatsAppSendCutByAStop(t *testing.T) {
+       whatsapp, fake := whatsappStartTestLinked(t)
+       daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
+       whatsappTestConnect(t, fake)
+
+       digest := sha256.Sum256([]byte("hi"))
+       require.NoError(t, whatsapp.state.PutSend(WhatsAppSend{
+               ID:     "keys",
+               Chat:   whatsappTestAna.String(),
+               Digest: hex.EncodeToString(digest[:]),
+               State:  whatsappSending,
+       }))
+
+       answer, err := whatsapp.Send(t.Context(), "keys", whatsappTestAna.String(), "hi")
+       require.NoError(t, err)
+       require.Contains(t, answer, "its result is unknown")
+       require.Empty(t, fake.Sent())
+}
+
+func TestWhatsAppSendRefuses(t *testing.T) {
+       testCases := []struct {
+               name string
+               id   string
+               chat string
+               text string
+               err  string
+       }{
+               {
+                       name: "no id",
+                       id:   " ",
+                       chat: whatsappTestAna.String(),
+                       text: "hi",
+                       err:  "the id must have 1 to 100 bytes",
+               },
+               {
+                       name: "no text",
+                       id:   "a",
+                       chat: whatsappTestAna.String(),
+                       text: "",
+                       err:  "the text must have 1 to 65536 bytes",
+               },
+               {
+                       name: "a name for a chat",
+                       id:   "a",
+                       chat: "Ana Souza",
+                       text: "hi",
+                       err:  `"Ana Souza" is not the ID of a WhatsApp chat: find the ID with whatsapp_find`,
+               },
+               {
+                       name: "a status broadcast",
+                       id:   "a",
+                       chat: types.StatusBroadcastJID.String(),
+                       text: "hi",
+                       err:  `"status@broadcast" is not a private chat or a group`,
+               },
+               {
+                       name: "the chat of the user",
+                       id:   "a",
+                       chat: whatsappTestOwn.String(),
+                       text: "hi",
+                       err:  "this is the chat of the user with itself: send to another chat",
+               },
+       }
+
+       whatsapp, fake := whatsappStartTestLinked(t)
+       daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
+       whatsappTestConnect(t, fake)
+
+       for _, tc := range testCases {
+               t.Run(tc.name, func(t *testing.T) {
+                       _, err := whatsapp.Send(t.Context(), tc.id, tc.chat, tc.text)
+                       require.EqualError(t, err, tc.err)
+               })
+       }
+
+       require.Empty(t, fake.Sent())
+}
+
+func TestWhatsAppSendWhileDisconnected(t *testing.T) {
+       whatsapp, fake := whatsappStartTestLinked(t)
+       daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
+       whatsappTestConnect(t, fake)
+       require.True(t, fake.Emit(&events.Disconnected{}))
+
+       _, err := whatsapp.Send(t.Context(), "keys", whatsappTestAna.String(), "hi")
+       require.EqualError(t, err, "WhatsApp is disconnected, so nothing was sent: try again later")
+
+       require.True(t, fake.Emit(&events.Connected{}))
+       answer, err := whatsapp.Send(t.Context(), "keys", whatsappTestAna.String(), "hi")
+       require.NoError(t, err)
+       require.Equal(t, "Sent to +15550002222 at 12:05.", answer)
+}
+
+func TestWhatsAppSendWithoutALink(t *testing.T) {
+       whatsapp, _, _ := whatsappStartTest(t, t.TempDir(), newTestClock())
+
+       answer, err := whatsapp.Send(t.Context(), "keys", whatsappTestAna.String(), "hi")
+       require.NoError(t, err)
+       require.Equal(t, whatsappNotLinked, answer)
+}
+
+func TestAgentSendsAWhatsAppMessage(t *testing.T) {
+       whatsapp, fake := whatsappStartTestLinked(t)
+       whatsappTestContacts(t, whatsapp)
+       server := newModelFake(t)
+       server.reply = func(w http.ResponseWriter, call ModelFakeCall) {
+               if call.number == 0 {
+                       modelWriteEvents(w, modelCompletedEvent([]any{modelFunctionCallItem("fc_1", "call_1", "whatsapp_send", `{"id":"keys","chat":"15550002222@s.whatsapp.net","text":"The keys are under the mat."}`)}, 100, 0, 5))
+                       return
+               }
+
+               modelWriteEvents(w, modelCompletedEvent([]any{modelMessageItem("Sent.")}, 100, 0, 5))

        }
 
        }
 

-       return texts
+       socket, _ := daemonStartTestWithWhatsApp(t, server, nil, whatsapp)
+       client := terminalStartTestClient(t, socket)
+       whatsappTestConnect(t, fake)
+
+       client.Type("Tell Ana that the keys are under the mat.")
+       client.WaitFor(t, "[result] Sent to Ana Souza at 12:05.\nSent.\n[done]")
+       require.Len(t, fake.Sent(), 1)

 }
 }