X-Git-Url: http://git.giacomolli.uk/onechat/blobdiff_plain/a6a82d96b4906e9e1f09f45f69f48124f11ae1bb..2e2524881dfe8f6c5cabfd9d342d5d3492953bd4:/whatsapp_test.go?ds=inline diff --git a/whatsapp_test.go b/whatsapp_test.go index 3d79def..2bdb4cc 100644 --- a/whatsapp_test.go +++ b/whatsapp_test.go @@ -4,7 +4,6 @@ import ( "crypto/sha256" "encoding/hex" "errors" - "fmt" "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"}`)) } -// 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)