Move whatsapp_find to its own file

[onechat] / whatsapp_test.go

diff --git a/whatsapp_test.go b/whatsapp_test.go
index 3d79def16b0bdea77d4be979bb2c9b9038d3b92d..2bdb4cc349e77f8f1fa901f98c11c3641fc4b611 100644 (file)
--- a/whatsapp_test.go
+++ b/whatsapp_test.go
@@ -4,7 +4,6 @@ import (
        "crypto/sha256"
        "encoding/hex"
        "errors"
        "crypto/sha256"
        "encoding/hex"
        "errors"

-       "fmt"

        "net/http"
        "os"
        "path/filepath"
        "net/http"
        "os"
        "path/filepath"


@@ -460,132 +459,12 @@ func TestWhatsAppCommandArgs(t *testing.T) {
        }
 }
 
        }
 }
 

-func TestWhatsAppFindMatchesNamesWithNoRegardToCaseAndAccents(t *testing.T) {
-       whatsapp, fake := whatsappStartTestLinked(t)
-       fake.groups = []*types.GroupInfo{
-               {
-                       JID: whatsappTestFamily,
-                       GroupName: types.GroupName{
-                               Name: "Família Souza",
-                       },
-               },
-       }
-
-       whatsappTestContacts(t, whatsapp)
-       daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
-       whatsappTestConnect(t, fake)
-
-       testCases := []struct {
-               name   string
-               query  string
-               answer string
-       }{
-               {
-                       name:   "no accent finds an accent",
-                       query:  "joao",
-                       answer: "15550004444@s.whatsapp.net | João Silva | contact",
-               },
-               {
-                       name:   "a part of a name finds contacts and groups",
-                       query:  "SOUZA",
-                       answer: "15550002222@s.whatsapp.net | Ana Souza | contact\n120363000000000001@g.us | Família Souza | group",
-               },
-               {
-                       name:   "a business name",
-                       query:  "bakery",
-                       answer: "15550005555@s.whatsapp.net | Corner Bakery | contact",
-               },
-               {
-                       name:   "no match",
-                       query:  "Zed",
-                       answer: "No contact or group has a name with \"Zed\".",
-               },
-       }
-
-       for _, tc := range testCases {
-               t.Run(tc.name, func(t *testing.T) {
-                       answer, err := whatsapp.Find(t.Context(), tc.query)
-                       require.NoError(t, err)
-                       require.Equal(t, tc.answer, answer)
-               })
-       }
-}
-
-func TestWhatsAppFindGivesAtMostTwentyMatches(t *testing.T) {
-       whatsapp, _ := whatsappStartTestLinked(t)
-       for index := range 25 {
-               jid := types.NewJID(fmt.Sprintf("1555001%04d", index), types.DefaultUserServer)
-               require.NoError(t, whatsapp.device.Contacts.PutContactName(t.Context(), jid, "", fmt.Sprintf("Member %02d", index)))
-       }
-
-       answer, err := whatsapp.Find(t.Context(), "member")
-       require.NoError(t, err)
-
-       lines := strings.Split(answer, "\n")
-       require.Len(t, lines, 22)
-       require.Equal(t, "15550010000@s.whatsapp.net | Member 00 | contact", lines[0])
-       require.Equal(t, "5 more: give a longer part of the name.", lines[20])
-       require.Equal(t, "The groups are not in this answer, because WhatsApp is disconnected.", lines[21])
-}
-
-func TestWhatsAppFindBeforeTheNamesArrive(t *testing.T) {
-       whatsapp, fake := whatsappStartTestLinked(t)
-       daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
-       whatsappTestConnect(t, fake)
-
-       answer, err := whatsapp.Find(t.Context(), "ana")
-       require.NoError(t, err)
-       require.Equal(t, whatsappNamesNotReady, answer)
-}
-
-func TestWhatsAppFindWithoutALink(t *testing.T) {
-       whatsapp, _, _ := whatsappStartTest(t, t.TempDir(), newTestClock())
-
-       answer, err := whatsapp.Find(t.Context(), "ana")
-       require.NoError(t, err)
-       require.Equal(t, whatsappNotLinked, answer)
-}
-
-func TestAgentFindsAWhatsAppContact(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_find", `{"name":"ana"}`)}, 100, 0, 5))
-                       return
-               }
-
-               modelWriteEvents(w, modelCompletedEvent([]any{modelMessageItem("Found Ana.")}, 100, 0, 5))
-       }
-
-       socket, _ := daemonStartTestWithWhatsApp(t, server, nil, whatsapp)
-       client := terminalStartTestClient(t, socket)
-       whatsappTestConnect(t, fake)
-
-       client.Type("Find Ana on WhatsApp.")
-       client.WaitFor(t, "[result] 15550002222@s.whatsapp.net | Ana Souza | contact\nFound Ana.\n[done]")
-}
-

 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"}`))
 }
 
 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"}`))
 }
 

-// whatsappTestContacts puts the contacts of the tests into the session of
-// whatsapp: Ana Souza, Joao Silva with an accent, a business, and the user.
-func whatsappTestContacts(t *testing.T, whatsapp *WhatsApp) {
-       t.Helper()
-
-       contacts := whatsapp.device.Contacts
-       require.NoError(t, contacts.PutContactName(t.Context(), whatsappTestAna, "Ana", "Ana Souza"))
-       require.NoError(t, contacts.PutContactName(t.Context(), types.NewJID("15550004444", types.DefaultUserServer), "João", "João Silva"))
-       _, _, err := contacts.PutBusinessName(t.Context(), types.NewJID("15550005555", types.DefaultUserServer), "Corner Bakery")
-       require.NoError(t, err)
-       require.NoError(t, contacts.PutContactName(t.Context(), whatsappTestOwn, "Me", "Me Souza"))
-}
-

 func TestWhatsAppSendSendsOnceForEachID(t *testing.T) {
        whatsapp, fake := whatsappStartTestLinked(t)
        whatsappTestContacts(t, whatsapp)
 func TestWhatsAppSendSendsOnceForEachID(t *testing.T) {
        whatsapp, fake := whatsappStartTestLinked(t)
        whatsappTestContacts(t, whatsapp)