X-Git-Url: https://git.giacomolli.uk/onechat/blobdiff_plain/b2873a2a0bf8b0ba648aebac51759e5fed89a2e2..4b063ca25798c78cc03d1d07f67b02c94c95248d:/daemon_test.go?ds=sidebyside

diff --git a/daemon_test.go b/daemon_test.go
index 00680e6..47fb9a5 100644
--- a/daemon_test.go
+++ b/daemon_test.go
@@ -242,6 +242,13 @@ func (b *syncBuffer) String() string {
 func startTestDaemon(t *testing.T, server *fakeResponses) string {
 	t.Helper()
 
+	return startTestDaemonWithBot(t, server, nil)
+}
+
+// startTestDaemonWithBot is startTestDaemon with bot, which can be nil.
+func startTestDaemonWithBot(t *testing.T, server *fakeResponses, bot *telegramBot) string {
+	t.Helper()
+
 	dir := t.TempDir()
 	agent, err := openAgent(filepath.Join(dir, agentDirName))
 	require.NoError(t, err)
@@ -259,7 +266,7 @@ func startTestDaemon(t *testing.T, server *fakeResponses) string {
 	ctx, cancel := context.WithCancel(t.Context())
 	done := make(chan error)
 	go func() {
-		done <- serveAgent(ctx, listener, func(notify func(Event)) (*loop, error) {
+		done <- serveAgent(ctx, listener, bot, func(notify func(Event)) (*loop, error) {
 			return newLoop(agent, memory, client, clock.Now, notify)
 		})
 	}()