package main
import (
- "context"
"errors"
"os"
"path/filepath"
synctest.Wait()
require.Equal(t, "WhatsApp is linked and disconnected since "+time.Now().Format(whatsappTimeLayout)+".", test.whatsapp.Status())
- require.Equal(t, "first connect failed, retrying: network unavailable\n", test.logs.String())
+ require.Equal(t, "<4>first connect failed, retrying: network unavailable\n", test.logs.String())
})
}
time.Sleep(10 * time.Minute)
synctest.Wait()
require.Len(t, test.fake.Connections(), 11)
- require.Equal(t, "first connect failed, retrying: network unavailable\nconnected again after "+durationText(outage)+"\n", test.logs.String())
+ require.Equal(t, "<4>first connect failed, retrying: network unavailable\n<6>connected again after="+durationText(outage)+"\n", test.logs.String())
})
}
time.Sleep(whatsappRetryLimit)
synctest.Wait()
require.Len(t, test.fake.Connections(), 2)
- require.Equal(t, "first connect failed, retrying: network unavailable\nconnected again after 2s\n", test.logs.String())
+ require.Equal(t, "<4>first connect failed, retrying: network unavailable\n<6>connected again after=2s\n", test.logs.String())
})
}
since := time.Now().Format(whatsappTimeLayout)
- time.Sleep(whatsappNoticeDelay)
+ 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(whatsappNoticeDelay)
+ time.Sleep(whatsappOutageNoticeDelay)
synctest.Wait()
require.Equal(t, 1, test.events.Count("notice"))
test.fake.mutex.Lock()
require.True(t, fake.Emit(&events.Connected{}))
require.Equal(t, []string{
- "connected",
- "disconnected, retrying",
- "connected again after 1m25s",
+ "<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 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)
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)
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)
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))
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)
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 = 300 * 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(400 * time.Millisecond)
- require.Equal(t, "[connected to OneChat]\n", client.Output())
+ synctest.Test(t, func(t *testing.T) {
+ test := whatsappRetryTest(t)
+ test.start(t)
+ synctest.Wait()
- clock.now = clock.now.Add(time.Hour)
+ 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, 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(400 * time.Millisecond)
+ 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, fake.Emit(&events.Connected{}))
- client.WaitFor(t, "WhatsApp is connected again.\n")
- require.Equal(t, 1, strings.Count(client.Output(), "WhatsApp is disconnected since"))
+ 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) {
require.Eventually(t, func() bool {
return logs.String() != ""
}, 10*time.Second, time.Millisecond)
- require.Equal(t, "first connect failed, retrying: 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) {
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"}`))
}
-
-// WhatsAppRetryTest runs WhatsApp without a model or network client call. All
-// timers and channels belong to its synctest bubble, so backoff uses fake time.
-type WhatsAppRetryTest struct {
- whatsapp *WhatsApp
- fake *WhatsAppFake
- logs *SyncBuffer
- loop *TurnLoop
- events *TurnEventRecorder
- cancel context.CancelFunc
- done chan error
-}
-
-func (r *WhatsAppRetryTest) start(t *testing.T) {
- t.Helper()
-
- ctx, cancel := context.WithCancel(t.Context())
- r.cancel = cancel
- r.done = make(chan error, 1)
- go func() {
- r.done <- r.whatsapp.Run(ctx, r.loop, r.events.Add)
- }()
-
- t.Cleanup(func() {
- cancel()
- synctest.Wait()
- })
-}
-
-func whatsappRetryTest(t *testing.T) *WhatsAppRetryTest {
- t.Helper()
-
- dir := t.TempDir()
- whatsappTestSession(t, dir, true)
-
- whatsapp, fake, logs := whatsappStartTest(t, dir, newTestClock())
- whatsapp.clock = time.Now
- agent, err := openAgent(t.TempDir())
- require.NoError(t, err)
-
- memory, err := openMemory(t.Context(), agent.dir, time.Now, newFakeSummarizer(), discardLog())
- require.NoError(t, err)
- t.Cleanup(func() {
- require.NoError(t, memory.Close())
- })
-
- events := &TurnEventRecorder{}
- client := newModelClient("https://example.com", ModelAPIKey(modelTestKey), agent.CacheKey(), filepath.Join(agent.dir, "usage.jsonl"))
- loop, err := newTurnLoop(agent, memory, newTestSubagents(t), whatsapp, newTestSchedules(t, time.Now), client, time.Now, discardLog(), events.Add)
- require.NoError(t, err)
-
- return &WhatsAppRetryTest{
- whatsapp: whatsapp,
- fake: fake,
- logs: logs,
- loop: loop,
- events: events,
- }
-}
projects / onechat / blobdiff