15 "github.com/stretchr/testify/require"
17 "go.mau.fi/whatsmeow/proto/waE2E"
18 "go.mau.fi/whatsmeow/proto/waWeb"
19 "go.mau.fi/whatsmeow/types"
20 "go.mau.fi/whatsmeow/types/events"
23 func TestWhatsAppStatusFollowsTheConnection(t *testing.T) {
25 whatsappTestSession(t, dir, true)
26 clock := newTestClock()
27 whatsapp, fake, _ := whatsappStartTest(t, dir, clock)
28 daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
31 require.Equal(t, "WhatsApp is linked and not connected yet.", whatsapp.Status())
33 require.True(t, fake.Emit(&events.Connected{}))
34 require.Equal(t, "WhatsApp is linked and connected since 2026-01-01 12:00.", whatsapp.Status())
36 clock.now = clock.now.Add(20 * time.Minute)
37 require.True(t, fake.Emit(&events.Disconnected{}))
38 clock.now = clock.now.Add(time.Minute)
39 require.True(t, fake.Emit(&events.Disconnected{}))
40 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())
43 func TestWhatsAppStatusOfAFirstConnectionThatFailed(t *testing.T) {
45 whatsappTestSession(t, dir, true)
46 whatsapp, fake, _ := whatsappStartTest(t, dir, newTestClock())
47 daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
50 require.True(t, fake.Emit(&events.Disconnected{}))
51 require.Equal(t, "WhatsApp is linked and disconnected since 2026-01-01 12:00.", whatsapp.Status())
54 func TestWhatsAppIsNotLinkedWithoutALinkedDevice(t *testing.T) {
55 testCases := []struct {
57 session func(t *testing.T, dir string)
60 name: "no session file",
61 session: func(t *testing.T, dir string) {
65 name: "a session of a pairing that did not end",
66 session: func(t *testing.T, dir string) {
67 whatsappTestSession(t, dir, false)
72 for _, tc := range testCases {
73 t.Run(tc.name, func(t *testing.T) {
76 whatsapp, _, _ := whatsappStartTest(t, dir, newTestClock())
77 require.Equal(t, "WhatsApp is not linked. Run /whatsapp pair PHONE to link it.", whatsapp.Status())
82 func TestWhatsAppRefusesASessionThatOthersCanRead(t *testing.T) {
84 whatsappTestSession(t, dir, true)
85 require.NoError(t, os.Chmod(filepath.Join(dir, whatsappDirName, whatsappSessionFileName), 0o644))
87 _, err := openWhatsApp(t.Context(), dir, time.Now, whatsappDiscardLog(), whatsappNewClient)
88 require.EqualError(t, err, "whatsapp/session.db holds the WhatsApp session, so its mode must be 0600")
91 func TestWhatsAppCommandsAnswerOnlyTheClientThatAsked(t *testing.T) {
92 server := newModelFake(t)
93 telegram := newTelegramFake(t)
94 bot, _ := telegramStartTestBot(t, telegram)
95 socket := daemonStartTestWithBot(t, server, bot)
96 asking := terminalStartTestClient(t, socket)
97 other := terminalStartTestClient(t, socket)
98 asking.WaitFor(t, "[connected to OneChat]")
99 other.WaitFor(t, "[connected to OneChat]")
101 asking.Type("/whatsapp status")
102 asking.WaitFor(t, "WhatsApp is not linked.")
103 asking.Type(" /whatsapp unknown ")
104 asking.WaitFor(t, whatsappUsage)
105 asking.Type("/whatsapp pair 0123")
106 asking.WaitFor(t, "PHONE must be the number of your WhatsApp account")
108 telegram.Add(TelegramUpdate{
110 Message: telegramOwnerMessage("/whatsapp status"),
113 telegramWaitForSent(t, telegram, "WhatsApp is not linked. Run /whatsapp pair PHONE to link it.")
114 require.Len(t, telegram.Sent(), 1)
115 require.Equal(t, "[connected to OneChat]\n", other.Output())
116 require.Zero(t, server.Calls())
119 func TestWhatsAppPairsWithACode(t *testing.T) {
121 whatsapp, fake, _ := whatsappStartTest(t, dir, newTestClock())
122 server := newModelFake(t)
123 socket, memory := daemonStartTestWithWhatsApp(t, server, nil, whatsapp)
124 asking := terminalStartTestClient(t, socket)
125 other := terminalStartTestClient(t, socket)
126 asking.WaitFor(t, "[connected to OneChat]")
127 other.WaitFor(t, "[connected to OneChat]")
129 asking.Type("/whatsapp pair +1 555-000-1111")
130 items := <-fake.pairings
131 items <- whatsmeow.QRChannelItem{
132 Event: whatsmeow.QRChannelEventCode,
135 asking.WaitFor(t, whatsappPairSteps+"\n"+whatsappTestCode+"\n")
136 require.True(t, fake.Emit(&events.PairSuccess{
137 ID: types.NewADJID(whatsappTestNumber, 0, 1),
138 LID: types.NewADJID(whatsappTestLID, 0, 1),
140 items <- whatsmeow.QRChannelSuccess
141 asking.WaitFor(t, "WhatsApp is linked to +15550001111.")
143 // The library connects again after the link with the context of the
144 // connection, so the end of the pairing must not end it.
145 connections := fake.Connections()
146 require.Len(t, connections, 1)
147 require.Never(t, func() bool {
148 return connections[0].Err() != nil
149 }, 200*time.Millisecond, time.Millisecond)
151 require.Equal(t, []string{whatsappTestNumber}, fake.Phones())
152 require.Equal(t, "WhatsApp is linked and not connected yet.", whatsapp.Status())
153 require.Equal(t, "[connected to OneChat]\n", other.Output())
154 require.Zero(t, server.Calls())
155 require.Equal(t, []string{"WhatsApp is linked."}, whatsappTestLogTexts(t, memory))
157 info, err := os.Stat(filepath.Join(dir, whatsappDirName))
158 require.NoError(t, err)
159 require.Equal(t, os.FileMode(0o700), info.Mode().Perm())
161 info, err = os.Stat(filepath.Join(dir, whatsappDirName, whatsappSessionFileName))
162 require.NoError(t, err)
163 require.Equal(t, os.FileMode(0o600), info.Mode().Perm())
166 func TestWhatsAppPairingCodeIsATelegramMessageAlone(t *testing.T) {
167 whatsapp, fake, _ := whatsappStartTest(t, t.TempDir(), newTestClock())
168 telegram := newTelegramFake(t)
169 bot, _ := telegramStartTestBot(t, telegram)
170 daemonStartTestWithWhatsApp(t, newModelFake(t), bot, whatsapp)
172 telegram.Add(TelegramUpdate{
174 Message: telegramOwnerMessage("/whatsapp pair 15550001111"),
177 items := <-fake.pairings
178 items <- whatsmeow.QRChannelItem{
179 Event: whatsmeow.QRChannelEventCode,
182 telegramWaitForSent(t, telegram, whatsappPairSteps, whatsappTestCode)
185 func TestWhatsAppPairingEnds(t *testing.T) {
186 testCases := []struct {
188 end func(items chan whatsmeow.QRChannelItem)
192 name: "the code expires",
193 end: func(items chan whatsmeow.QRChannelItem) {
194 items <- whatsmeow.QRChannelTimeout
196 answer: whatsappPairExpired,
199 name: "the library closes the pairing",
200 end: func(items chan whatsmeow.QRChannelItem) {
203 answer: whatsappPairExpired,
206 name: "the pairing fails",
207 end: func(items chan whatsmeow.QRChannelItem) {
208 items <- whatsmeow.QRChannelItem{
209 Event: whatsmeow.QRChannelEventError,
210 Error: errors.New("bad handshake"),
213 answer: "The pairing failed: bad handshake. Run /whatsapp pair PHONE again.",
217 for _, tc := range testCases {
218 t.Run(tc.name, func(t *testing.T) {
219 whatsapp, fake, _ := whatsappStartTest(t, t.TempDir(), newTestClock())
220 socket, _ := daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
221 client := terminalStartTestClient(t, socket)
222 client.Type("/whatsapp pair 15550001111")
224 items := <-fake.pairings
225 items <- whatsmeow.QRChannelItem{
226 Event: whatsmeow.QRChannelEventCode,
229 client.WaitFor(t, whatsappTestCode)
231 client.WaitFor(t, tc.answer)
232 require.Equal(t, "WhatsApp is not linked. Run /whatsapp pair PHONE to link it.", whatsapp.Status())
237 func TestWhatsAppNewPairingEndsTheOneBefore(t *testing.T) {
238 whatsapp, fake, _ := whatsappStartTest(t, t.TempDir(), newTestClock())
239 socket, _ := daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
240 client := terminalStartTestClient(t, socket)
242 client.Type("/whatsapp pair 15550001111")
244 client.Type("/whatsapp pair 15550002222")
245 items := <-fake.pairings
246 items <- whatsmeow.QRChannelItem{
247 Event: whatsmeow.QRChannelEventCode,
250 client.WaitFor(t, whatsappTestCode)
251 require.Equal(t, []string{"15550002222"}, fake.Phones())
252 require.NotContains(t, client.Output(), "did not start the pairing")
255 func TestWhatsAppPairWhenLinked(t *testing.T) {
257 whatsappTestSession(t, dir, true)
258 whatsapp, _, _ := whatsappStartTest(t, dir, newTestClock())
259 socket, _ := daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
260 client := terminalStartTestClient(t, socket)
262 client.Type("/whatsapp pair 15550001111")
263 client.WaitFor(t, "WhatsApp is linked already. Run /whatsapp unlink first.")
266 func TestWhatsAppUnlink(t *testing.T) {
267 testCases := []struct {
273 name: "with a logout",
274 answer: "WhatsApp is unlinked.",
277 name: "with no logout",
278 logoutErr: errors.New("not connected"),
279 answer: "WhatsApp is unlinked here, but the phone did not get the logout. In WhatsApp on your phone, open Linked devices and remove the device.",
283 for _, tc := range testCases {
284 t.Run(tc.name, func(t *testing.T) {
286 whatsappTestSession(t, dir, true)
287 whatsapp, fake, _ := whatsappStartTest(t, dir, newTestClock())
288 fake.logoutErr = tc.logoutErr
289 socket, memory := daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
290 client := terminalStartTestClient(t, socket)
293 client.Type("/whatsapp unlink")
294 client.WaitFor(t, tc.answer)
295 require.Equal(t, 1, fake.Logouts())
296 require.NoFileExists(t, filepath.Join(dir, whatsappDirName, whatsappSessionFileName))
297 require.Equal(t, "WhatsApp is not linked. Run /whatsapp pair PHONE to link it.", whatsapp.Status())
298 require.Equal(t, []string{"WhatsApp is unlinked."}, whatsappTestLogTexts(t, memory))
300 client.Type("/whatsapp unlink")
301 client.WaitFor(t, "WhatsApp is not linked.\n")
306 func TestWhatsAppLogoutFromThePhone(t *testing.T) {
308 whatsappTestSession(t, dir, true)
309 whatsapp, fake, _ := whatsappStartTest(t, dir, newTestClock())
310 telegram := newTelegramFake(t)
311 bot, _ := telegramStartTestBot(t, telegram)
312 socket, memory := daemonStartTestWithWhatsApp(t, newModelFake(t), bot, whatsapp)
313 client := terminalStartTestClient(t, socket)
314 client.WaitFor(t, "[connected to OneChat]")
317 require.True(t, fake.Emit(&events.LoggedOut{}))
318 notice := "WhatsApp was unlinked from the phone. Run /whatsapp pair PHONE to link it again."
319 client.WaitFor(t, notice)
320 telegramWaitForSent(t, telegram, notice)
321 require.NoFileExists(t, filepath.Join(dir, whatsappDirName, whatsappSessionFileName))
322 require.Equal(t, "WhatsApp is not linked. Run /whatsapp pair PHONE to link it.", whatsapp.Status())
323 require.Equal(t, []string{"WhatsApp was unlinked from the phone."}, whatsappTestLogTexts(t, memory))
326 func TestWhatsAppLongDisconnectionGivesOneNotice(t *testing.T) {
328 whatsappTestSession(t, dir, true)
329 clock := newTestClock()
330 whatsapp, fake, _ := whatsappStartTest(t, dir, clock)
331 whatsapp.noticeDelay = 300 * time.Millisecond
332 socket, _ := daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
333 client := terminalStartTestClient(t, socket)
334 client.WaitFor(t, "[connected to OneChat]")
337 require.True(t, fake.Emit(&events.Connected{}))
338 require.True(t, fake.Emit(&events.Disconnected{}))
339 require.True(t, fake.Emit(&events.Connected{}))
340 time.Sleep(400 * time.Millisecond)
341 require.Equal(t, "[connected to OneChat]\n", client.Output())
343 clock.now = clock.now.Add(time.Hour)
344 require.True(t, fake.Emit(&events.Disconnected{}))
345 require.True(t, fake.Emit(&events.Disconnected{}))
346 client.WaitFor(t, "WhatsApp is disconnected since 2026-01-01 13:00. OneChat tries again.\n")
347 require.True(t, fake.Emit(&events.Disconnected{}))
348 time.Sleep(400 * time.Millisecond)
350 require.True(t, fake.Emit(&events.Connected{}))
351 client.WaitFor(t, "WhatsApp is connected again.\n")
352 require.Equal(t, 1, strings.Count(client.Output(), "WhatsApp is disconnected since"))
355 func TestWhatsAppDisconnectionWithNoRetryGivesANoticeAtOnce(t *testing.T) {
357 whatsappTestSession(t, dir, true)
358 whatsapp, fake, _ := whatsappStartTest(t, dir, newTestClock())
359 socket, _ := daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
360 client := terminalStartTestClient(t, socket)
361 client.WaitFor(t, "[connected to OneChat]")
364 require.True(t, fake.Emit(&events.StreamReplaced{}))
365 client.WaitFor(t, "WhatsApp is disconnected and does not connect again by itself: stream replaced. Restart the daemon to try again.")
368 func TestWhatsAppLogsNoOwnNumber(t *testing.T) {
370 whatsappTestSession(t, dir, true)
371 whatsapp, fake, logs := whatsappStartTest(t, dir, newTestClock())
372 fake.connectErr = errors.New("no route for " + whatsappTestNumber + "@s.whatsapp.net and " + whatsappTestLID + "@lid")
373 daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
376 require.Eventually(t, func() bool {
377 return logs.String() != ""
378 }, 10*time.Second, time.Millisecond)
379 require.Equal(t, "connect: no route for OWN@s.whatsapp.net and OWN@lid\n", logs.String())
382 func TestWhatsAppPhone(t *testing.T) {
383 testCases := []struct {
389 name: "a number with a plus and dashes",
390 phone: "+1-555-000-1111",
394 name: "a number with no country code",
395 phone: "0115550001111",
399 name: "a short number",
404 name: "a long number",
405 phone: "1234567890123456",
409 name: "a number with letters",
410 phone: "1555ABC1111",
415 for _, tc := range testCases {
416 t.Run(tc.name, func(t *testing.T) {
417 _, valid := whatsappPhone(tc.phone)
418 require.Equal(t, tc.valid, valid)
423 func TestWhatsAppCommandArgs(t *testing.T) {
424 testCases := []struct {
431 name: "a command with its arguments",
432 line: " /whatsapp pair +1 555 ",
437 name: "the command alone",
443 name: "a word that starts with the command",
444 line: "/whatsappstatus",
449 name: "a message that names the command",
450 line: "what does /whatsapp status do?",
456 for _, tc := range testCases {
457 t.Run(tc.name, func(t *testing.T) {
458 args, found := whatsappCommandArgs(tc.line)
459 require.Equal(t, tc.args, args)
460 require.Equal(t, tc.found, found)
465 // whatsappTestLogTexts returns the texts of the messages in the log of memory,
466 // after the log holds at least one message.
467 func whatsappTestLogTexts(t *testing.T, memory *Memory) []string {
470 require.Eventually(t, func() bool {
471 return memory.Len() > 0
472 }, 10*time.Second, time.Millisecond)
475 for id := range memory.Len() {
476 message, err := memory.Message(id)
477 require.NoError(t, err)
479 texts = append(texts, message.Text)
485 func TestWhatsAppPrivateBatchStartsATurnWithAnEmptyReply(t *testing.T) {
486 whatsapp, fake := whatsappStartTestLinked(t)
487 require.NoError(t, whatsapp.device.Contacts.PutContactName(t.Context(), whatsappTestAna, "Ana", "Ana Souza"))
489 server := newModelFake(t)
490 server.reply = func(w http.ResponseWriter, call ModelFakeCall) {
491 modelWriteEvents(w, modelCompletedEvent([]any{modelMessageItem("")}, 100, 0, 5))
494 telegram := newTelegramFake(t)
495 bot, _ := telegramStartTestBot(t, telegram)
496 socket, memory := daemonStartTestWithWhatsApp(t, server, bot, whatsapp)
497 client := terminalStartTestClient(t, socket)
498 client.WaitFor(t, "[connected to OneChat]")
501 // The catch-up holds the batch open until its end, so the two messages
502 // form one batch also on a slow machine.
503 require.True(t, fake.Emit(&events.Connected{}))
504 at := time.Date(2026, 1, 1, 12, 0, 0, 0, time.UTC)
505 require.True(t, fake.Emit(whatsappTestMessage(whatsappTestAna, whatsappTestAna, "m1", "Are you at home?", at)))
506 require.True(t, fake.Emit(whatsappTestMessage(whatsappTestAna, whatsappTestOwn, "m2", "Yes", at)))
507 require.True(t, fake.Emit(&events.OfflineSyncCompleted{}))
509 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
510 client.WaitFor(t, text+"\n[working]\n[done]\n")
511 require.Equal(t, 1, server.Calls())
512 require.Contains(t, string(server.Call(0).raw), "Are you at home?")
513 require.Equal(t, []string{text}, whatsappTestLogTexts(t, memory))
514 require.Empty(t, telegram.Sent())
515 require.Eventually(t, func() bool {
516 return len(whatsapp.state.Closed()) == 0
517 }, 10*time.Second, time.Millisecond)
520 func TestWhatsAppGroupTurns(t *testing.T) {
521 testCases := []struct {
523 message func(event *events.Message)
529 message: func(event *events.Message) {
531 line: "12:00 Bruno Lima: Who brings the cake?",
535 message: func(event *events.Message) {
536 event.Message = &waE2E.Message{
537 ExtendedTextMessage: &waE2E.ExtendedTextMessage{
538 Text: new("@" + whatsappTestLID + " can you bring the cake? Call " + whatsappTestNumber + "."),
539 ContextInfo: &waE2E.ContextInfo{
540 MentionedJID: []string{whatsappTestLID + "@lid"},
545 line: "12:00 Bruno Lima (mentions you): @you can you bring the cake? Call OWN.",
550 message: func(event *events.Message) {
551 event.Message = &waE2E.Message{
552 ExtendedTextMessage: &waE2E.ExtendedTextMessage{
553 Text: new("Who brings the cake?"),
554 ContextInfo: &waE2E.ContextInfo{
555 Participant: new(whatsappTestOwn.String()),
560 line: "12:00 Bruno Lima (replies to you): Who brings the cake?",
565 for _, tc := range testCases {
566 t.Run(tc.name, func(t *testing.T) {
567 whatsapp, fake := whatsappStartTestLinked(t)
568 fake.groups = []*types.GroupInfo{
570 JID: whatsappTestFamily,
571 GroupName: types.GroupName{
577 server := newModelFake(t)
578 socket, memory := daemonStartTestWithWhatsApp(t, server, nil, whatsapp)
579 client := terminalStartTestClient(t, socket)
580 client.WaitFor(t, "[connected to OneChat]")
581 whatsappTestConnect(t, fake)
583 event := whatsappTestMessage(whatsappTestFamily, whatsappTestBruno, "m1", "Who brings the cake?", time.Date(2026, 1, 1, 12, 0, 0, 0, time.UTC))
584 event.Info.PushName = "Bruno Lima"
586 require.True(t, fake.Emit(event))
588 text := "[WhatsApp] Family, group 120363000000000001@g.us, Thursday 2026-01-01\n" + tc.line
590 text += "\n" + whatsappBatchEmptyReply
591 client.WaitFor(t, "[done]")
592 require.Equal(t, 1, server.Calls())
594 client.WaitFor(t, text)
595 time.Sleep(100 * time.Millisecond)
596 require.Zero(t, server.Calls())
599 require.Equal(t, text, whatsappTestLogTexts(t, memory)[0])
604 func TestWhatsAppTakesOnlyText(t *testing.T) {
605 at := time.Date(2026, 1, 1, 12, 0, 0, 0, time.UTC)
606 testCases := []struct {
608 change func(event *events.Message)
612 change: func(event *events.Message) {
618 change: func(event *events.Message) {
619 event.SourceWebMsg = &waWeb.WebMessageInfo{}
623 name: "a view-once message",
624 change: func(event *events.Message) {
625 event.IsViewOnce = true
629 name: "a disappearing message",
630 change: func(event *events.Message) {
631 event.IsEphemeral = true
635 name: "the chat of the user with itself",
636 change: func(event *events.Message) {
637 event.Info.Chat = whatsappTestOwn
641 name: "a status broadcast",
642 change: func(event *events.Message) {
643 event.Info.Chat = types.StatusBroadcastJID
647 name: "a photo with a caption",
648 change: func(event *events.Message) {
649 event.Message = &waE2E.Message{
650 ImageMessage: &waE2E.ImageMessage{
651 Caption: new("look"),
657 name: "a text of spaces",
658 change: func(event *events.Message) {
659 event.Message.Conversation = new(" ")
664 for _, tc := range testCases {
665 t.Run(tc.name, func(t *testing.T) {
666 whatsapp, _ := whatsappStartTestLinked(t)
667 event := whatsappTestMessage(whatsappTestAna, whatsappTestAna, "m1", "hi", at)
669 require.True(t, whatsapp.handle(event))
670 require.NoError(t, whatsapp.state.CloseAll())
671 require.Empty(t, whatsapp.state.Closed())
676 // whatsappStartTestLinked opens a WhatsApp with a linked device in a temporary
677 // directory, with a quiet time of 50 ms for its batches.
678 func whatsappStartTestLinked(t *testing.T) (*WhatsApp, *WhatsAppFake) {
682 whatsappTestSession(t, dir, true)
683 whatsapp, fake, _ := whatsappStartTest(t, dir, newTestClock())
684 whatsapp.state.quiet = 50 * time.Millisecond
686 return whatsapp, fake
689 func TestWhatsAppCatchupGivesOneTurn(t *testing.T) {
690 whatsapp, fake := whatsappStartTestLinked(t)
691 server := newModelFake(t)
692 socket, memory := daemonStartTestWithWhatsApp(t, server, nil, whatsapp)
693 client := terminalStartTestClient(t, socket)
694 client.WaitFor(t, "[connected to OneChat]")
697 require.True(t, fake.Emit(&events.Connected{}))
698 at := time.Date(2026, 1, 1, 8, 0, 0, 0, time.UTC)
699 for index, event := range []*events.Message{
700 whatsappTestMessage(whatsappTestAna, whatsappTestAna, "m1", "Good morning", at),
701 whatsappTestMessage(whatsappTestBruno, whatsappTestBruno, "m2", "Call me", at.Add(time.Hour)),
702 whatsappTestMessage(whatsappTestAna, whatsappTestAna, "m3", "Are you there?", at.Add(2*time.Hour)),
704 event.Info.PushName = []string{"Ana", "Bruno", "Ana"}[index]
705 require.True(t, fake.Emit(event))
708 time.Sleep(100 * time.Millisecond)
709 require.Zero(t, memory.Len())
711 require.True(t, fake.Emit(&events.OfflineSyncCompleted{}))
712 client.WaitFor(t, "[done]")
713 require.Equal(t, []string{
714 "[WhatsApp] Ana, private chat 15550002222@s.whatsapp.net, Thursday 2026-01-01\n08:00 Ana: Good morning\n" + whatsappBatchEmptyReply,
715 "[WhatsApp] Bruno, private chat 15550003333@s.whatsapp.net, Thursday 2026-01-01\n09:00 Bruno: Call me\n" + whatsappBatchEmptyReply,
716 "[WhatsApp] Ana, private chat 15550002222@s.whatsapp.net, Thursday 2026-01-01\n10:00 Ana: Are you there?\n" + whatsappBatchEmptyReply,
717 }, whatsappTestLogTexts(t, memory)[:3])
719 time.Sleep(100 * time.Millisecond)
720 require.Equal(t, 1, server.Calls())
721 for _, text := range []string{"Good morning", "Call me", "Are you there?"} {
722 require.Contains(t, string(server.Call(0).raw), text)
726 func TestWhatsAppCatchupEndsAfterItsLimit(t *testing.T) {
727 whatsapp, fake := whatsappStartTestLinked(t)
728 whatsapp.catchupLimit = 100 * time.Millisecond
729 socket, memory := daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
730 client := terminalStartTestClient(t, socket)
731 client.WaitFor(t, "[connected to OneChat]")
734 require.True(t, fake.Emit(&events.Connected{}))
735 require.True(t, fake.Emit(whatsappTestMessage(whatsappTestAna, whatsappTestOwn, "m1", "On my way.", time.Date(2026, 1, 1, 12, 0, 0, 0, time.UTC))))
736 client.WaitFor(t, "12:00 You: On my way.")
737 require.Len(t, whatsappTestLogTexts(t, memory), 1)
740 func TestWhatsAppStateOfTheStateFileWaitForTheCatchup(t *testing.T) {
742 whatsappTestSession(t, dir, true)
743 before, _, _ := whatsappStartTest(t, dir, newTestClock())
744 _, err := before.state.Add(whatsappTestChat, WhatsAppMessage{
747 Time: time.Date(2026, 1, 1, 12, 0, 0, 0, time.UTC),
748 Text: "Did you see my message?",
749 }, time.Now().Add(-time.Hour))
750 require.NoError(t, err)
752 whatsapp, fake, _ := whatsappStartTest(t, dir, newTestClock())
753 socket, memory := daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
754 client := terminalStartTestClient(t, socket)
755 client.WaitFor(t, "[connected to OneChat]")
758 time.Sleep(100 * time.Millisecond)
759 require.Zero(t, memory.Len())
761 require.True(t, fake.Emit(&events.Connected{}))
762 require.True(t, fake.Emit(&events.OfflineSyncCompleted{}))
763 client.WaitFor(t, "12:00 Ana Souza: Did you see my message?")
764 client.WaitFor(t, "[done]")
767 // whatsappTestConnect waits for the connection of the daemon and gives it the
768 // events of a connection with an empty catch-up.
769 func whatsappTestConnect(t *testing.T, fake *WhatsAppFake) {
773 require.True(t, fake.Emit(&events.Connected{}))
774 require.True(t, fake.Emit(&events.OfflineSyncCompleted{}))
777 func TestWhatsAppFindMatchesNamesWithNoRegardToCaseAndAccents(t *testing.T) {
778 whatsapp, fake := whatsappStartTestLinked(t)
779 fake.groups = []*types.GroupInfo{
781 JID: whatsappTestFamily,
782 GroupName: types.GroupName{
783 Name: "Família Souza",
788 whatsappTestContacts(t, whatsapp)
789 daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
790 whatsappTestConnect(t, fake)
792 testCases := []struct {
798 name: "no accent finds an accent",
800 answer: "15550004444@s.whatsapp.net | João Silva | contact",
803 name: "a part of a name finds contacts and groups",
805 answer: "15550002222@s.whatsapp.net | Ana Souza | contact\n120363000000000001@g.us | Família Souza | group",
808 name: "a business name",
810 answer: "15550005555@s.whatsapp.net | Corner Bakery | contact",
815 answer: "No contact or group has a name with \"Zed\".",
819 for _, tc := range testCases {
820 t.Run(tc.name, func(t *testing.T) {
821 answer, err := whatsapp.Find(t.Context(), tc.query)
822 require.NoError(t, err)
823 require.Equal(t, tc.answer, answer)
828 func TestWhatsAppFindGivesAtMostTwentyMatches(t *testing.T) {
829 whatsapp, _ := whatsappStartTestLinked(t)
830 for index := range 25 {
831 jid := types.NewJID(fmt.Sprintf("1555001%04d", index), types.DefaultUserServer)
832 require.NoError(t, whatsapp.device.Contacts.PutContactName(t.Context(), jid, "", fmt.Sprintf("Member %02d", index)))
835 answer, err := whatsapp.Find(t.Context(), "member")
836 require.NoError(t, err)
838 lines := strings.Split(answer, "\n")
839 require.Len(t, lines, 22)
840 require.Equal(t, "15550010000@s.whatsapp.net | Member 00 | contact", lines[0])
841 require.Equal(t, "5 more: give a longer part of the name.", lines[20])
842 require.Equal(t, "The groups are not in this answer, because WhatsApp is disconnected.", lines[21])
845 func TestWhatsAppFindBeforeTheNamesArrive(t *testing.T) {
846 whatsapp, fake := whatsappStartTestLinked(t)
847 daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
848 whatsappTestConnect(t, fake)
850 answer, err := whatsapp.Find(t.Context(), "ana")
851 require.NoError(t, err)
852 require.Equal(t, whatsappNamesNotReady, answer)
855 func TestWhatsAppFindWithoutALink(t *testing.T) {
856 whatsapp, _, _ := whatsappStartTest(t, t.TempDir(), newTestClock())
858 answer, err := whatsapp.Find(t.Context(), "ana")
859 require.NoError(t, err)
860 require.Equal(t, whatsappNotLinked, answer)
863 func TestAgentFindsAWhatsAppContact(t *testing.T) {
864 whatsapp, fake := whatsappStartTestLinked(t)
865 whatsappTestContacts(t, whatsapp)
866 server := newModelFake(t)
867 server.reply = func(w http.ResponseWriter, call ModelFakeCall) {
868 if call.number == 0 {
869 modelWriteEvents(w, modelCompletedEvent([]any{modelFunctionCallItem("fc_1", "call_1", "whatsapp_find", `{"name":"ana"}`)}, 100, 0, 5))
873 modelWriteEvents(w, modelCompletedEvent([]any{modelMessageItem("Found Ana.")}, 100, 0, 5))
876 socket, _ := daemonStartTestWithWhatsApp(t, server, nil, whatsapp)
877 client := terminalStartTestClient(t, socket)
878 whatsappTestConnect(t, fake)
880 client.Type("Find Ana on WhatsApp.")
881 client.WaitFor(t, "[result] 15550002222@s.whatsapp.net | Ana Souza | contact\nFound Ana.\n[done]")
884 func TestSubagentHasNoWhatsAppTools(t *testing.T) {
886 require.Equal(t, `error: unknown tool "whatsapp_find"`, box.run(t.Context(), "whatsapp_find", `{"name":"ana"}`))
887 require.Equal(t, `error: unknown tool "whatsapp_send"`, box.run(t.Context(), "whatsapp_send", `{"id":"a","chat":"15550002222@s.whatsapp.net","text":"hi"}`))
890 // whatsappTestContacts puts the contacts of the tests into the session of
891 // whatsapp: Ana Souza, Joao Silva with an accent, a business, and the user.
892 func whatsappTestContacts(t *testing.T, whatsapp *WhatsApp) {
895 contacts := whatsapp.device.Contacts
896 require.NoError(t, contacts.PutContactName(t.Context(), whatsappTestAna, "Ana", "Ana Souza"))
897 require.NoError(t, contacts.PutContactName(t.Context(), types.NewJID("15550004444", types.DefaultUserServer), "João", "João Silva"))
898 _, _, err := contacts.PutBusinessName(t.Context(), types.NewJID("15550005555", types.DefaultUserServer), "Corner Bakery")
899 require.NoError(t, err)
900 require.NoError(t, contacts.PutContactName(t.Context(), whatsappTestOwn, "Me", "Me Souza"))
903 func TestWhatsAppSendSendsOnceForEachID(t *testing.T) {
904 whatsapp, fake := whatsappStartTestLinked(t)
905 whatsappTestContacts(t, whatsapp)
906 daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
907 whatsappTestConnect(t, fake)
909 answer, err := whatsapp.Send(t.Context(), "keys", whatsappTestAna.String(), "The keys are under the mat.")
910 require.NoError(t, err)
911 require.Equal(t, "Sent to Ana Souza at 12:05.", answer)
913 answer, err = whatsapp.Send(t.Context(), "keys", whatsappTestAna.String(), "The keys are under the mat.")
914 require.NoError(t, err)
915 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)
917 _, err = whatsapp.Send(t.Context(), "keys", whatsappTestAna.String(), "Another text.")
918 require.EqualError(t, err, "the id keys belongs to another message: give a new id for each message")
920 require.Equal(t, []WhatsAppFakeSend{
922 to: whatsappTestAna.String(),
923 text: "The keys are under the mat.",
928 again, err := openWhatsAppState(filepath.Join(whatsapp.dir, whatsappStateFileName), whatsappBatchQuiet, whatsappBatchWindow)
929 require.NoError(t, err)
931 record, found := again.Send("keys")
932 require.True(t, found)
933 require.Equal(t, whatsappSent, record.State)
934 require.Equal(t, "MSG1", record.MessageID)
937 func TestWhatsAppSendToAGroup(t *testing.T) {
938 whatsapp, fake := whatsappStartTestLinked(t)
939 fake.groups = []*types.GroupInfo{
941 JID: whatsappTestFamily,
942 GroupName: types.GroupName{
948 daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
949 whatsappTestConnect(t, fake)
951 answer, err := whatsapp.Send(t.Context(), "cake", whatsappTestFamily.String(), "I bring the cake.")
952 require.NoError(t, err)
953 require.Equal(t, "Sent to Family at 12:05.", answer)
956 func TestWhatsAppSendThatFails(t *testing.T) {
957 whatsapp, fake := whatsappStartTestLinked(t)
958 fake.sendErr = errors.New("server error for " + whatsappTestNumber)
959 daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
960 whatsappTestConnect(t, fake)
963 answer, err := whatsapp.Send(t.Context(), "keys", whatsappTestAna.String(), "hi")
964 require.NoError(t, err)
965 require.True(t, strings.HasPrefix(answer, "The send failed: server error for OWN."), answer)
968 require.Len(t, fake.Sent(), 1)
971 func TestWhatsAppSendCutByAStop(t *testing.T) {
972 whatsapp, fake := whatsappStartTestLinked(t)
973 daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
974 whatsappTestConnect(t, fake)
976 digest := sha256.Sum256([]byte("hi"))
977 require.NoError(t, whatsapp.state.PutSend(WhatsAppSend{
979 Chat: whatsappTestAna.String(),
980 Digest: hex.EncodeToString(digest[:]),
981 State: whatsappSending,
984 answer, err := whatsapp.Send(t.Context(), "keys", whatsappTestAna.String(), "hi")
985 require.NoError(t, err)
986 require.Contains(t, answer, "its result is unknown")
987 require.Empty(t, fake.Sent())
990 func TestWhatsAppSendRefuses(t *testing.T) {
991 testCases := []struct {
1001 chat: whatsappTestAna.String(),
1003 err: "the id must have 1 to 100 bytes",
1008 chat: whatsappTestAna.String(),
1010 err: "the text must have 1 to 65536 bytes",
1013 name: "a name for a chat",
1017 err: `"Ana Souza" is not the ID of a WhatsApp chat: find the ID with whatsapp_find`,
1020 name: "a status broadcast",
1022 chat: types.StatusBroadcastJID.String(),
1024 err: `"status@broadcast" is not a private chat or a group`,
1027 name: "the chat of the user",
1029 chat: whatsappTestOwn.String(),
1031 err: "this is the chat of the user with itself: send to another chat",
1035 whatsapp, fake := whatsappStartTestLinked(t)
1036 daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
1037 whatsappTestConnect(t, fake)
1039 for _, tc := range testCases {
1040 t.Run(tc.name, func(t *testing.T) {
1041 _, err := whatsapp.Send(t.Context(), tc.id, tc.chat, tc.text)
1042 require.EqualError(t, err, tc.err)
1046 require.Empty(t, fake.Sent())
1049 func TestWhatsAppSendWhileDisconnected(t *testing.T) {
1050 whatsapp, fake := whatsappStartTestLinked(t)
1051 daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
1052 whatsappTestConnect(t, fake)
1053 require.True(t, fake.Emit(&events.Disconnected{}))
1055 _, err := whatsapp.Send(t.Context(), "keys", whatsappTestAna.String(), "hi")
1056 require.EqualError(t, err, "WhatsApp is disconnected, so nothing was sent: try again later")
1058 require.True(t, fake.Emit(&events.Connected{}))
1059 answer, err := whatsapp.Send(t.Context(), "keys", whatsappTestAna.String(), "hi")
1060 require.NoError(t, err)
1061 require.Equal(t, "Sent to +15550002222 at 12:05.", answer)
1064 func TestWhatsAppSendWithoutALink(t *testing.T) {
1065 whatsapp, _, _ := whatsappStartTest(t, t.TempDir(), newTestClock())
1067 answer, err := whatsapp.Send(t.Context(), "keys", whatsappTestAna.String(), "hi")
1068 require.NoError(t, err)
1069 require.Equal(t, whatsappNotLinked, answer)
1072 func TestAgentSendsAWhatsAppMessage(t *testing.T) {
1073 whatsapp, fake := whatsappStartTestLinked(t)
1074 whatsappTestContacts(t, whatsapp)
1075 server := newModelFake(t)
1076 server.reply = func(w http.ResponseWriter, call ModelFakeCall) {
1077 if call.number == 0 {
1078 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))
1082 modelWriteEvents(w, modelCompletedEvent([]any{modelMessageItem("Sent.")}, 100, 0, 5))
1085 socket, _ := daemonStartTestWithWhatsApp(t, server, nil, whatsapp)
1086 client := terminalStartTestClient(t, socket)
1087 whatsappTestConnect(t, fake)
1089 client.Type("Tell Ana that the keys are under the mat.")
1090 client.WaitFor(t, "[result] Sent to Ana Souza at 12:05.\nSent.\n[done]")
1091 require.Len(t, fake.Sent(), 1)