Link WhatsApp with a pairing code

[onechat] / daemon_start_helper_test.go

diff --git a/daemon_start_helper_test.go b/daemon_start_helper_test.go
index cf1fbf6a507e8bde0923646e679150f699daa209..b2c3787d4189a8a285b754feb1ffbe207e5100b7 100644 (file)
--- a/daemon_start_helper_test.go
+++ b/daemon_start_helper_test.go
@@ -13,6 +13,17 @@ import (
 func daemonStartTestWithBot(t *testing.T, server *ModelFake, bot *TelegramBot) string {
        t.Helper()
 
+       whatsapp, _, _ := whatsappStartTest(t, t.TempDir(), newTestClock())
+       socket, _ := daemonStartTestWithWhatsApp(t, server, bot, whatsapp)
+
+       return socket
+}
+
+// daemonStartTestWithWhatsApp is daemonStartTestWithBot with whatsapp. It also
+// returns the memory of the agent.
+func daemonStartTestWithWhatsApp(t *testing.T, server *ModelFake, bot *TelegramBot, whatsapp *WhatsApp) (string, *Memory) {
+       t.Helper()
+
        dir := t.TempDir()
        agent, err := openAgent(filepath.Join(dir, daemonAgentDirName))
        require.NoError(t, err)
@@ -22,7 +33,6 @@ func daemonStartTestWithBot(t *testing.T, server *ModelFake, bot *TelegramBot) s
        memory, err := openMemory(t.Context(), agent.dir, clock.Now, newFakeSummarizer())
        require.NoError(t, err)
 
-       whatsapp, _ := whatsappStartTest(t, dir)
        socket := filepath.Join(dir, socketFileName)
        listener, err := daemonListenSocket(socket)
        require.NoError(t, err)
@@ -42,5 +52,5 @@ func daemonStartTestWithBot(t *testing.T, server *ModelFake, bot *TelegramBot) s
                require.NoError(t, memory.Close())
        })
 
-       return socket
+       return socket, memory
 }