X-Git-Url: https://git.giacomolli.uk/onechat/blobdiff_plain/33aa709aa4e59e05ada6940a1c055f5e5646a418..d7a967f43fae27bcabc6072da3b5a2f923d47764:/whatsapp_test.go

diff --git a/whatsapp_test.go b/whatsapp_test.go
index 1ca0a52..7001ea1 100644
--- a/whatsapp_test.go
+++ b/whatsapp_test.go
@@ -37,12 +37,38 @@ func TestWhatsAppStatusFollowsTheConnection(t *testing.T) {
 func TestWhatsAppStatusOfAFirstConnectionThatFailed(t *testing.T) {
 	dir := t.TempDir()
 	whatsappTestSession(t, dir, true)
-	whatsapp, fake, _ := whatsappStartTest(t, dir, newTestClock())
+	whatsapp, fake, logs := whatsappStartTest(t, dir, newTestClock())
 	daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
 
 	<-fake.connects
 	require.True(t, fake.Emit(&events.Disconnected{}))
 	require.Equal(t, "WhatsApp is linked and disconnected since 2026-01-01 12:00.", whatsapp.Status())
+	require.Equal(t, "first connect failed, retrying\n", logs.String())
+}
+
+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{}))
+	clock.now = clock.now.Add(25 * time.Second)
+	require.True(t, fake.Emit(&events.Connected{}))
+
+	require.Equal(t, []string{
+		"connected",
+		"disconnected, retrying",
+		"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) {
@@ -370,7 +396,7 @@ func TestWhatsAppLogsNoOwnNumber(t *testing.T) {
 	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, "connect failed, restart to retry: no route for OWN@s.whatsapp.net and OWN@lid\n", logs.String())
 }
 
 func TestWhatsAppPhone(t *testing.T) {