import (
"errors"
- "fmt"
- "net/http"
"os"
"path/filepath"
"strings"
"testing"
+ "testing/synctest"
"time"
"github.com/stretchr/testify/require"
"go.mau.fi/whatsmeow"
- "go.mau.fi/whatsmeow/proto/waE2E"
- "go.mau.fi/whatsmeow/proto/waWeb"
"go.mau.fi/whatsmeow/types"
"go.mau.fi/whatsmeow/types/events"
)
func TestWhatsAppStatusFollowsTheConnection(t *testing.T) {
dir := t.TempDir()
whatsappTestSession(t, dir, true)
+
clock := newTestClock()
whatsapp, fake, _ := whatsappStartTest(t, dir, clock)
daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
require.Equal(t, "WhatsApp is linked and connected since 2026-01-01 12:00.", whatsapp.Status())
clock.now = clock.now.Add(20 * time.Minute)
+
require.True(t, fake.Emit(&events.Disconnected{}))
+
clock.now = clock.now.Add(time.Minute)
+
require.True(t, fake.Emit(&events.Disconnected{}))
require.Equal(t, "WhatsApp is linked and disconnected since 2026-01-01 12:20. The last connection started at 2026-01-01 12:00.", whatsapp.Status())
}
func TestWhatsAppStatusOfAFirstConnectionThatFailed(t *testing.T) {
+ synctest.Test(t, func(t *testing.T) {
+ test := whatsappRetryTest(t)
+ test.fake.connectErr = errors.New("network unavailable")
+ test.start(t)
+ synctest.Wait()
+
+ require.Equal(t, "WhatsApp is linked and disconnected since "+time.Now().Format(whatsappTimeLayout)+".", test.whatsapp.Status())
+ require.Equal(t, "<4>first connect failed, retrying: network unavailable\n", test.logs.String())
+ })
+}
+
+func TestWhatsAppRetriesTheFirstConnectionWithCappedBackoff(t *testing.T) {
+ synctest.Test(t, func(t *testing.T) {
+ test := whatsappRetryTest(t)
+ for range 10 {
+ test.fake.connectErrors = append(test.fake.connectErrors, errors.New("network unavailable"))
+ }
+
+ test.start(t)
+ synctest.Wait()
+ require.Len(t, test.fake.Connections(), 1)
+ require.True(t, test.fake.Emit(&events.Disconnected{}))
+ require.True(t, test.fake.Emit(&events.Disconnected{}))
+
+ delays := []time.Duration{2 * time.Second, 4 * time.Second, 8 * time.Second, 16 * time.Second, 32 * time.Second, 64 * time.Second, 128 * time.Second, 256 * time.Second, 5 * time.Minute, 5 * time.Minute}
+
+ var outage time.Duration
+
+ for index, delay := range delays {
+ time.Sleep(delay - time.Nanosecond)
+ synctest.Wait()
+ require.Len(t, test.fake.Connections(), index+1)
+ time.Sleep(time.Nanosecond)
+ synctest.Wait()
+ require.Len(t, test.fake.Connections(), index+2)
+
+ outage += delay
+ }
+
+ times := test.fake.ConnectionTimes()
+ for index, delay := range delays {
+ require.Equal(t, delay, times[index+1].Sub(times[index]))
+ }
+
+ require.True(t, test.fake.Emit(&events.Connected{}))
+ require.True(t, test.fake.Emit(&events.OfflineSyncCompleted{}))
+ synctest.Wait()
+ time.Sleep(10 * time.Minute)
+ synctest.Wait()
+ require.Len(t, test.fake.Connections(), 11)
+ require.Equal(t, "<4>first connect failed, retrying: network unavailable\n<6>connected again after="+durationText(outage)+"\n", test.logs.String())
+ })
+}
+
+func TestWhatsAppFirstConnectionRetriesStopWhenContextEnds(t *testing.T) {
+ synctest.Test(t, func(t *testing.T) {
+ test := whatsappRetryTest(t)
+ test.fake.connectErr = errors.New("network unavailable")
+ test.start(t)
+ synctest.Wait()
+ time.Sleep(2 * time.Second)
+ synctest.Wait()
+ require.Len(t, test.fake.Connections(), 2)
+
+ test.cancel()
+ require.NoError(t, <-test.done)
+ time.Sleep(10 * time.Minute)
+ synctest.Wait()
+ require.Len(t, test.fake.Connections(), 2)
+ })
+}
+
+func TestWhatsAppLinkedPairRequestKeepsFirstConnectionRetries(t *testing.T) {
+ synctest.Test(t, func(t *testing.T) {
+ test := whatsappRetryTest(t)
+ test.fake.connectErr = errors.New("network unavailable")
+ test.start(t)
+ synctest.Wait()
+
+ answers := make(chan string, 1)
+ test.whatsapp.Command("pair 15550001111", func(text string) {
+ answers <- text
+ })
+ require.Equal(t, "WhatsApp is linked already. Run /whatsapp unlink first.", <-answers)
+ synctest.Wait()
+ test.fake.mutex.Lock()
+
+ test.fake.connectErr = nil
+ test.fake.mutex.Unlock()
+ time.Sleep(whatsappRetryDelay)
+ synctest.Wait()
+ require.Len(t, test.fake.Connections(), 2)
+ require.True(t, test.fake.Emit(&events.Connected{}))
+ require.True(t, test.fake.Emit(&events.OfflineSyncCompleted{}))
+ require.Contains(t, test.whatsapp.Status(), "linked and connected since")
+ time.Sleep(whatsappRetryLimit)
+ synctest.Wait()
+ require.Len(t, test.fake.Connections(), 2)
+ require.Equal(t, "<4>first connect failed, retrying: network unavailable\n<6>connected again after=2s\n", test.logs.String())
+ })
+}
+
+func TestWhatsAppFirstConnectionRetriesStopForLinkChanges(t *testing.T) {
+ testCases := []struct {
+ name string
+ stop func(test *WhatsAppRetryTest)
+ }{
+ {
+ name: "unlink",
+ stop: func(test *WhatsAppRetryTest) {
+ test.whatsapp.Command("unlink", func(_ string) {
+ })
+ },
+ },
+ {
+ name: "logout",
+ stop: func(test *WhatsAppRetryTest) {
+ test.fake.Emit(&events.LoggedOut{})
+ },
+ },
+ {
+ name: "permanent disconnect",
+ stop: func(test *WhatsAppRetryTest) {
+ test.fake.Emit(&events.StreamReplaced{})
+ },
+ },
+ }
+
+ for _, tc := range testCases {
+ t.Run(tc.name, func(t *testing.T) {
+ synctest.Test(t, func(t *testing.T) {
+ test := whatsappRetryTest(t)
+ test.fake.connectErr = errors.New("network unavailable")
+ test.start(t)
+ synctest.Wait()
+ tc.stop(test)
+ synctest.Wait()
+ time.Sleep(10 * time.Minute)
+ synctest.Wait()
+ require.Len(t, test.fake.Connections(), 1)
+ })
+ })
+ }
+}
+
+func TestWhatsAppPairsAfterFirstConnectionRetriesStop(t *testing.T) {
+ synctest.Test(t, func(t *testing.T) {
+ test := whatsappRetryTest(t)
+ test.fake.connectErr = errors.New("network unavailable")
+ test.start(t)
+ synctest.Wait()
+ test.whatsapp.Command("unlink", func(_ string) {
+ })
+ synctest.Wait()
+ require.False(t, test.whatsapp.isLinked())
+
+ test.fake.mutex.Lock()
+
+ test.fake.connectErr = nil
+ test.fake.mutex.Unlock()
+ test.whatsapp.Command("pair 15550001111", func(_ string) {
+ })
+
+ items := <-test.fake.pairings
+ items <- whatsmeow.QRChannelItem{Event: whatsmeow.QRChannelEventCode}
+
+ synctest.Wait()
+ require.True(t, test.fake.Emit(&events.PairSuccess{
+ ID: types.NewADJID(whatsappTestNumber, 0, 1),
+ LID: types.NewADJID(whatsappTestLID, 0, 1),
+ }))
+
+ items <- whatsmeow.QRChannelSuccess
+
+ synctest.Wait()
+ require.True(t, test.whatsapp.isLinked())
+ time.Sleep(10 * time.Minute)
+ synctest.Wait()
+ require.Len(t, test.fake.Connections(), 2)
+ require.Equal(t, []string{whatsappTestNumber}, test.fake.Phones())
+ })
+}
+
+func TestWhatsAppInboxFailureStopsFirstConnectionRetries(t *testing.T) {
+ synctest.Test(t, func(t *testing.T) {
+ test := whatsappRetryTest(t)
+ test.fake.connectErr = errors.New("network unavailable")
+ test.start(t)
+ synctest.Wait()
+
+ failure := errors.New("state write failed")
+ test.whatsapp.inbox.failures <- failure
+
+ require.ErrorIs(t, <-test.done, failure)
+ time.Sleep(10 * time.Minute)
+ synctest.Wait()
+ require.Len(t, test.fake.Connections(), 1)
+ })
+}
+
+func TestWhatsAppLongFirstConnectionOutageGivesOneNotice(t *testing.T) {
+ synctest.Test(t, func(t *testing.T) {
+ test := whatsappRetryTest(t)
+ test.fake.connectErr = errors.New("network unavailable")
+ test.start(t)
+ synctest.Wait()
+
+ since := time.Now().Format(whatsappTimeLayout)
+
+ time.Sleep(whatsappOutageNoticeDelay)
+ synctest.Wait()
+ require.Equal(t, 1, test.events.Count("notice"))
+ require.Equal(t, "WhatsApp is disconnected since "+since+". OneChat tries again.", test.events.Last("notice"))
+
+ time.Sleep(whatsappOutageNoticeDelay)
+ synctest.Wait()
+ require.Equal(t, 1, test.events.Count("notice"))
+ test.fake.mutex.Lock()
+
+ test.fake.connectErr = nil
+ test.fake.mutex.Unlock()
+ time.Sleep(whatsappRetryLimit)
+ synctest.Wait()
+ require.True(t, test.fake.Emit(&events.Connected{}))
+ require.True(t, test.fake.Emit(&events.OfflineSyncCompleted{}))
+ synctest.Wait()
+ require.Equal(t, 2, test.events.Count("notice"))
+ require.Equal(t, "WhatsApp is connected again.", test.events.Last("notice"))
+ })
+}
+
+func TestWhatsAppStartingRetriesDoNotDuplicateLibraryRetries(t *testing.T) {
dir := t.TempDir()
whatsappTestSession(t, dir, true)
- whatsapp, fake, _ := whatsappStartTest(t, dir, newTestClock())
+
+ whatsapp, _, _ := whatsappStartTest(t, dir, newTestClock())
+ client, ok := whatsappNewClient(whatsapp.device, whatsapp.handle).(*whatsmeow.Client)
+ require.True(t, ok)
+ require.False(t, client.InitialAutoReconnect)
+ require.True(t, client.EnableAutoReconnect)
+
+ device := whatsapp.container.NewDevice()
+ client, ok = whatsappNewClient(device, whatsapp.handle).(*whatsmeow.Client)
+ require.True(t, ok)
+ require.True(t, client.InitialAutoReconnect)
+}
+
+func TestWhatsAppLogsEachChangeOfTheConnectionOnce(t *testing.T) {
+ dir := t.TempDir()
+ whatsappTestSession(t, dir, true)
+
+ clock := newTestClock()
+ whatsapp, fake, logs := whatsappStartTest(t, dir, clock)
daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
<-fake.connects
+ require.True(t, fake.Emit(&events.Connected{}))
+
+ clock.now = clock.now.Add(time.Minute)
+
+ require.True(t, fake.Emit(&events.Disconnected{}))
+
+ clock.now = clock.now.Add(time.Minute)
+
require.True(t, fake.Emit(&events.Disconnected{}))
- require.Equal(t, "WhatsApp is linked and disconnected since 2026-01-01 12:00.", whatsapp.Status())
+
+ clock.now = clock.now.Add(25 * time.Second)
+
+ require.True(t, fake.Emit(&events.Connected{}))
+
+ require.Equal(t, []string{
+ "<6>connected",
+ "<4>disconnected, retrying",
+ "<6>connected again after=1m25s",
+ }, strings.Split(strings.TrimSuffix(logs.String(), "\n"), "\n"))
+ require.NotContains(t, logs.String(), whatsappTestNumber)
+ require.NotContains(t, logs.String(), whatsappTestLID)
}
func TestWhatsAppIsNotLinkedWithoutALinkedDevice(t *testing.T) {
}{
{
name: "no session file",
- session: func(t *testing.T, dir string) {
+ session: func(_ *testing.T, _ string) {
},
},
{
t.Run(tc.name, func(t *testing.T) {
dir := t.TempDir()
tc.session(t, dir)
+
whatsapp, _, _ := whatsappStartTest(t, dir, newTestClock())
require.Equal(t, "WhatsApp is not linked. Run /whatsapp pair PHONE to link it.", whatsapp.Status())
})
func TestWhatsAppPairsWithACode(t *testing.T) {
dir := t.TempDir()
- whatsapp, fake, _ := whatsappStartTest(t, dir, newTestClock())
+ whatsapp, fake, logs := whatsappStartTest(t, dir, newTestClock())
server := newModelFake(t)
socket, memory := daemonStartTestWithWhatsApp(t, server, nil, whatsapp)
asking := terminalStartTestClient(t, socket)
other.WaitFor(t, "[connected to OneChat]")
asking.Type("/whatsapp pair +1 555-000-1111")
+
items := <-fake.pairings
items <- whatsmeow.QRChannelItem{
Event: whatsmeow.QRChannelEventCode,
ID: types.NewADJID(whatsappTestNumber, 0, 1),
LID: types.NewADJID(whatsappTestLID, 0, 1),
}))
+
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.Zero(t, server.Calls())
require.Equal(t, []string{"WhatsApp is linked."}, whatsappTestLogTexts(t, memory))
+ require.Equal(t, "<6>linked\n", logs.String())
info, err := os.Stat(filepath.Join(dir, whatsappDirName))
require.NoError(t, err)
client.Type("/whatsapp pair 15550001111")
<-fake.pairings
client.Type("/whatsapp pair 15550002222")
+
items := <-fake.pairings
items <- whatsmeow.QRChannelItem{
Event: whatsmeow.QRChannelEventCode,
func TestWhatsAppPairWhenLinked(t *testing.T) {
dir := t.TempDir()
whatsappTestSession(t, dir, true)
+
whatsapp, _, _ := whatsappStartTest(t, dir, newTestClock())
socket, _ := daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
client := terminalStartTestClient(t, socket)
t.Run(tc.name, func(t *testing.T) {
dir := t.TempDir()
whatsappTestSession(t, dir, true)
- whatsapp, fake, _ := whatsappStartTest(t, dir, newTestClock())
+
+ whatsapp, fake, logs := whatsappStartTest(t, dir, newTestClock())
fake.logoutErr = tc.logoutErr
socket, memory := daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
client := terminalStartTestClient(t, socket)
+
<-fake.connects
client.Type("/whatsapp unlink")
client.WaitFor(t, tc.answer)
require.Equal(t, 1, fake.Logouts())
+ require.Equal(t, "<6>unlinked\n", logs.String())
require.NoFileExists(t, filepath.Join(dir, whatsappDirName, whatsappSessionFileName))
require.Equal(t, "WhatsApp is not linked. Run /whatsapp pair PHONE to link it.", whatsapp.Status())
require.Equal(t, []string{"WhatsApp is unlinked."}, whatsappTestLogTexts(t, memory))
func TestWhatsAppLogoutFromThePhone(t *testing.T) {
dir := t.TempDir()
whatsappTestSession(t, dir, true)
- whatsapp, fake, _ := whatsappStartTest(t, dir, newTestClock())
+
+ whatsapp, fake, logs := whatsappStartTest(t, dir, newTestClock())
telegram := newTelegramFake(t)
bot, _ := telegramStartTestBot(t, telegram)
socket, memory := daemonStartTestWithWhatsApp(t, newModelFake(t), bot, whatsapp)
<-fake.connects
require.True(t, fake.Emit(&events.LoggedOut{}))
+
notice := "WhatsApp was unlinked from the phone. Run /whatsapp pair PHONE to link it again."
client.WaitFor(t, notice)
telegramWaitForSent(t, telegram, notice)
+ require.Equal(t, "<3>logged out by the phone, run /whatsapp pair PHONE\n", logs.String())
require.NoFileExists(t, filepath.Join(dir, whatsappDirName, whatsappSessionFileName))
require.Equal(t, "WhatsApp is not linked. Run /whatsapp pair PHONE to link it.", whatsapp.Status())
require.Equal(t, []string{"WhatsApp was unlinked from the phone."}, whatsappTestLogTexts(t, memory))
}
func TestWhatsAppLongDisconnectionGivesOneNotice(t *testing.T) {
- dir := t.TempDir()
- whatsappTestSession(t, dir, true)
- clock := newTestClock()
- whatsapp, fake, _ := whatsappStartTest(t, dir, clock)
- whatsapp.noticeDelay = 50 * time.Millisecond
- socket, _ := daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
- client := terminalStartTestClient(t, socket)
- client.WaitFor(t, "[connected to OneChat]")
- <-fake.connects
-
- 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)
- require.Equal(t, "[connected to OneChat]\n", client.Output())
-
- clock.now = clock.now.Add(time.Hour)
- 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)
-
- require.True(t, fake.Emit(&events.Connected{}))
- client.WaitFor(t, "WhatsApp is connected again.\n")
- require.Equal(t, 1, strings.Count(client.Output(), "WhatsApp is disconnected since"))
+ synctest.Test(t, func(t *testing.T) {
+ test := whatsappRetryTest(t)
+ test.start(t)
+ synctest.Wait()
+
+ require.True(t, test.fake.Emit(&events.Connected{}))
+ require.True(t, test.fake.Emit(&events.Disconnected{}))
+ require.True(t, test.fake.Emit(&events.Connected{}))
+ time.Sleep(whatsappOutageNoticeDelay + time.Second)
+ synctest.Wait()
+ require.Zero(t, test.events.Count("notice"))
+
+ require.True(t, test.fake.Emit(&events.Disconnected{}))
+ require.True(t, test.fake.Emit(&events.Disconnected{}))
+ time.Sleep(whatsappOutageNoticeDelay)
+ synctest.Wait()
+ require.Equal(t, 1, test.events.Count("notice"))
+ require.Contains(t, test.events.Last("notice"), "OneChat tries again.")
+ require.True(t, test.fake.Emit(&events.Disconnected{}))
+ time.Sleep(whatsappOutageNoticeDelay)
+ synctest.Wait()
+ require.Equal(t, 1, test.events.Count("notice"))
+
+ require.True(t, test.fake.Emit(&events.Connected{}))
+ require.Equal(t, 2, test.events.Count("notice"))
+ require.Equal(t, "WhatsApp is connected again.", test.events.Last("notice"))
+ })
}
func TestWhatsAppDisconnectionWithNoRetryGivesANoticeAtOnce(t *testing.T) {
dir := t.TempDir()
whatsappTestSession(t, dir, true)
- whatsapp, fake, _ := whatsappStartTest(t, dir, newTestClock())
+
+ whatsapp, fake, logs := whatsappStartTest(t, dir, newTestClock())
socket, _ := daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
client := terminalStartTestClient(t, socket)
client.WaitFor(t, "[connected to OneChat]")
require.True(t, fake.Emit(&events.StreamReplaced{}))
client.WaitFor(t, "WhatsApp is disconnected and does not connect again by itself: stream replaced. Restart the daemon to try again.")
+ require.Equal(t, "<3>disconnected for good, restart to retry: stream replaced\n", logs.String())
}
func TestWhatsAppLogsNoOwnNumber(t *testing.T) {
dir := t.TempDir()
whatsappTestSession(t, dir, true)
+
whatsapp, fake, logs := whatsappStartTest(t, dir, newTestClock())
fake.connectErr = errors.New("no route for " + whatsappTestNumber + "@s.whatsapp.net and " + whatsappTestLID + "@lid")
+
daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
<-fake.connects
require.Eventually(t, func() bool {
return logs.String() != ""
}, 10*time.Second, time.Millisecond)
- require.Equal(t, "connect: no route for OWN@s.whatsapp.net and OWN@lid\n", logs.String())
+ require.Equal(t, "<4>first connect failed, retrying: no route for OWN@s.whatsapp.net and OWN@lid\n", logs.String())
}
func TestWhatsAppPhone(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()
-
- require.Eventually(t, func() bool {
- return memory.Len() > 0
- }, 10*time.Second, time.Millisecond)
-
- var texts []string
- for id := range memory.Len() {
- message, err := memory.Message(id)
- require.NoError(t, err)
-
- texts = append(texts, message.Text)
- }
-
- return texts
-}
-
-func TestWhatsAppPrivateBatchStartsATurnWithAnEmptyReply(t *testing.T) {
- whatsapp, fake := whatsappStartTestLinked(t)
- require.NoError(t, whatsapp.device.Contacts.PutContactName(t.Context(), whatsappTestAna, "Ana", "Ana Souza"))
-
- server := newModelFake(t)
- server.reply = func(w http.ResponseWriter, call ModelFakeCall) {
- modelWriteEvents(w, modelCompletedEvent([]any{modelMessageItem("")}, 100, 0, 5))
- }
-
- telegram := newTelegramFake(t)
- bot, _ := telegramStartTestBot(t, telegram)
- socket, memory := daemonStartTestWithWhatsApp(t, server, bot, whatsapp)
- client := terminalStartTestClient(t, socket)
- client.WaitFor(t, "[connected to OneChat]")
- whatsappTestConnect(t, fake)
-
- at := time.Date(2026, 1, 1, 12, 0, 0, 0, time.UTC)
- require.True(t, fake.Emit(whatsappTestMessage(whatsappTestAna, whatsappTestAna, "m1", "Are you at home?", at)))
- require.True(t, fake.Emit(whatsappTestMessage(whatsappTestAna, whatsappTestOwn, "m2", "Yes", at)))
-
- text := "[WhatsApp] Ana Souza, private chat 15550002222@s.whatsapp.net, Thursday 2026-01-01\n12:00 Ana Souza: Are you at home?\n12:00 You: Yes\n" + whatsappBatchEmptyReply
- client.WaitFor(t, text+"\n[working]\n[done]\n")
- require.Equal(t, 1, server.Calls())
- require.Contains(t, string(server.Call(0).raw), "Are you at home?")
- require.Equal(t, []string{text}, whatsappTestLogTexts(t, memory))
- require.Empty(t, telegram.Sent())
- require.Eventually(t, func() bool {
- return len(whatsapp.batches.Closed()) == 0
- }, 10*time.Second, time.Millisecond)
-}
-
-func TestWhatsAppGroupTurns(t *testing.T) {
- testCases := []struct {
- name string
- message func(event *events.Message)
- line string
- turn bool
- }{
- {
- name: "plain",
- message: func(event *events.Message) {
- },
- line: "12:00 Bruno Lima: Who brings the cake?",
- },
- {
- name: "mention",
- message: func(event *events.Message) {
- event.Message = &waE2E.Message{
- ExtendedTextMessage: &waE2E.ExtendedTextMessage{
- Text: new("@" + whatsappTestLID + " can you bring the cake? Call " + whatsappTestNumber + "."),
- ContextInfo: &waE2E.ContextInfo{
- MentionedJID: []string{whatsappTestLID + "@lid"},
- },
- },
- }
- },
- line: "12:00 Bruno Lima (mentions you): @you can you bring the cake? Call OWN.",
- turn: true,
- },
- {
- name: "reply",
- message: func(event *events.Message) {
- event.Message = &waE2E.Message{
- ExtendedTextMessage: &waE2E.ExtendedTextMessage{
- Text: new("Who brings the cake?"),
- ContextInfo: &waE2E.ContextInfo{
- Participant: new(whatsappTestOwn.String()),
- },
- },
- }
- },
- line: "12:00 Bruno Lima (replies to you): Who brings the cake?",
- turn: true,
- },
- }
-
- for _, tc := range testCases {
- t.Run(tc.name, func(t *testing.T) {
- whatsapp, fake := whatsappStartTestLinked(t)
- fake.groups = []*types.GroupInfo{
- {
- JID: whatsappTestFamily,
- GroupName: types.GroupName{
- Name: "Family",
- },
- },
- }
-
- server := newModelFake(t)
- socket, memory := daemonStartTestWithWhatsApp(t, server, nil, whatsapp)
- client := terminalStartTestClient(t, socket)
- client.WaitFor(t, "[connected to OneChat]")
- whatsappTestConnect(t, fake)
-
- event := whatsappTestMessage(whatsappTestFamily, whatsappTestBruno, "m1", "Who brings the cake?", time.Date(2026, 1, 1, 12, 0, 0, 0, time.UTC))
- event.Info.PushName = "Bruno Lima"
- tc.message(event)
- require.True(t, fake.Emit(event))
-
- text := "[WhatsApp] Family, group 120363000000000001@g.us, Thursday 2026-01-01\n" + tc.line
- if tc.turn {
- text += "\n" + whatsappBatchEmptyReply
- client.WaitFor(t, "[done]")
- require.Equal(t, 1, server.Calls())
- } else {
- client.WaitFor(t, text)
- time.Sleep(100 * time.Millisecond)
- require.Zero(t, server.Calls())
- }
-
- require.Equal(t, text, whatsappTestLogTexts(t, memory)[0])
- })
- }
-}
-
-func TestWhatsAppTakesOnlyText(t *testing.T) {
- at := time.Date(2026, 1, 1, 12, 0, 0, 0, time.UTC)
- testCases := []struct {
- name string
- change func(event *events.Message)
- }{
- {
- name: "an edit",
- change: func(event *events.Message) {
- event.IsEdit = true
- },
- },
- {
- name: "old history",
- change: func(event *events.Message) {
- event.SourceWebMsg = &waWeb.WebMessageInfo{}
- },
- },
- {
- name: "a view-once message",
- change: func(event *events.Message) {
- event.IsViewOnce = true
- },
- },
- {
- name: "a disappearing message",
- change: func(event *events.Message) {
- event.IsEphemeral = true
- },
- },
- {
- name: "the chat of the user with itself",
- change: func(event *events.Message) {
- event.Info.Chat = whatsappTestOwn
- },
- },
- {
- name: "a status broadcast",
- change: func(event *events.Message) {
- event.Info.Chat = types.StatusBroadcastJID
- },
- },
- {
- name: "a photo with a caption",
- change: func(event *events.Message) {
- event.Message = &waE2E.Message{
- ImageMessage: &waE2E.ImageMessage{
- Caption: new("look"),
- },
- }
- },
- },
- {
- name: "a text of spaces",
- change: func(event *events.Message) {
- event.Message.Conversation = new(" ")
- },
- },
- }
-
- for _, tc := range testCases {
- t.Run(tc.name, func(t *testing.T) {
- whatsapp, _ := whatsappStartTestLinked(t)
- event := whatsappTestMessage(whatsappTestAna, whatsappTestAna, "m1", "hi", at)
- tc.change(event)
- require.True(t, whatsapp.handle(event))
- require.NoError(t, whatsapp.batches.CloseAll())
- require.Empty(t, whatsapp.batches.Closed())
- })
- }
-}
-
-// whatsappStartTestLinked opens a WhatsApp with a linked device in a temporary
-// directory, with a quiet time of 50 ms for its batches.
-func whatsappStartTestLinked(t *testing.T) (*WhatsApp, *WhatsAppFake) {
- t.Helper()
-
- dir := t.TempDir()
- whatsappTestSession(t, dir, true)
- whatsapp, fake, _ := whatsappStartTest(t, dir, newTestClock())
- whatsapp.batches.quiet = 50 * time.Millisecond
-
- return whatsapp, fake
-}
-
-func TestWhatsAppCatchupGivesOneTurn(t *testing.T) {
- whatsapp, fake := whatsappStartTestLinked(t)
- server := newModelFake(t)
- socket, memory := daemonStartTestWithWhatsApp(t, server, nil, whatsapp)
- client := terminalStartTestClient(t, socket)
- client.WaitFor(t, "[connected to OneChat]")
- <-fake.connects
-
- require.True(t, fake.Emit(&events.Connected{}))
- at := time.Date(2026, 1, 1, 8, 0, 0, 0, time.UTC)
- for index, event := range []*events.Message{
- whatsappTestMessage(whatsappTestAna, whatsappTestAna, "m1", "Good morning", at),
- whatsappTestMessage(whatsappTestBruno, whatsappTestBruno, "m2", "Call me", at.Add(time.Hour)),
- whatsappTestMessage(whatsappTestAna, whatsappTestAna, "m3", "Are you there?", at.Add(2*time.Hour)),
- } {
- event.Info.PushName = []string{"Ana", "Bruno", "Ana"}[index]
- require.True(t, fake.Emit(event))
- }
-
- time.Sleep(100 * time.Millisecond)
- require.Zero(t, memory.Len())
-
- require.True(t, fake.Emit(&events.OfflineSyncCompleted{}))
- client.WaitFor(t, "[done]")
- require.Equal(t, []string{
- "[WhatsApp] Ana, private chat 15550002222@s.whatsapp.net, Thursday 2026-01-01\n08:00 Ana: Good morning\n" + whatsappBatchEmptyReply,
- "[WhatsApp] Bruno, private chat 15550003333@s.whatsapp.net, Thursday 2026-01-01\n09:00 Bruno: Call me\n" + whatsappBatchEmptyReply,
- "[WhatsApp] Ana, private chat 15550002222@s.whatsapp.net, Thursday 2026-01-01\n10:00 Ana: Are you there?\n" + whatsappBatchEmptyReply,
- }, whatsappTestLogTexts(t, memory)[:3])
-
- time.Sleep(100 * time.Millisecond)
- require.Equal(t, 1, server.Calls())
- for _, text := range []string{"Good morning", "Call me", "Are you there?"} {
- require.Contains(t, string(server.Call(0).raw), text)
- }
-}
-
-func TestWhatsAppCatchupEndsAfterItsLimit(t *testing.T) {
- whatsapp, fake := whatsappStartTestLinked(t)
- whatsapp.catchupLimit = 100 * time.Millisecond
- socket, memory := daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
- client := terminalStartTestClient(t, socket)
- client.WaitFor(t, "[connected to OneChat]")
- <-fake.connects
-
- require.True(t, fake.Emit(&events.Connected{}))
- require.True(t, fake.Emit(whatsappTestMessage(whatsappTestAna, whatsappTestOwn, "m1", "On my way.", time.Date(2026, 1, 1, 12, 0, 0, 0, time.UTC))))
- client.WaitFor(t, "12:00 You: On my way.")
- require.Len(t, whatsappTestLogTexts(t, memory), 1)
-}
-
-func TestWhatsAppBatchesOfTheStateFileWaitForTheCatchup(t *testing.T) {
- dir := t.TempDir()
- whatsappTestSession(t, dir, true)
- before, _, _ := whatsappStartTest(t, dir, newTestClock())
- _, err := before.batches.Add(whatsappTestChat, WhatsAppMessage{
- ID: "m1",
- Sender: "Ana Souza",
- Time: time.Date(2026, 1, 1, 12, 0, 0, 0, time.UTC),
- Text: "Did you see my message?",
- }, time.Now().Add(-time.Hour))
- require.NoError(t, err)
-
- whatsapp, fake, _ := whatsappStartTest(t, dir, newTestClock())
- socket, memory := daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
- client := terminalStartTestClient(t, socket)
- client.WaitFor(t, "[connected to OneChat]")
- <-fake.connects
-
- time.Sleep(100 * time.Millisecond)
- require.Zero(t, memory.Len())
-
- require.True(t, fake.Emit(&events.Connected{}))
- require.True(t, fake.Emit(&events.OfflineSyncCompleted{}))
- client.WaitFor(t, "12:00 Ana Souza: Did you see my message?")
- client.WaitFor(t, "[done]")
-}
-
-// whatsappTestConnect waits for the connection of the daemon and gives it the
-// events of a connection with an empty catch-up.
-func whatsappTestConnect(t *testing.T, fake *WhatsAppFake) {
- t.Helper()
-
- <-fake.connects
- require.True(t, fake.Emit(&events.Connected{}))
- require.True(t, fake.Emit(&events.OfflineSyncCompleted{}))
-}
-
-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"}`))
-}
-
-// 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"))
+ require.Equal(t, `error: unknown tool "whatsapp_send"`, box.run(t.Context(), "whatsapp_send", `{"id":"a","chat":"15550002222@s.whatsapp.net","text":"hi"}`))
}
projects / onechat / blobdiff