Name and order the Telegram bot code

[onechat] / daemon_test.go

diff --git a/daemon_test.go b/daemon_test.go
index 00680e6fec697e028bdeeadf25688f698257b3fc..4bcc05a50c2343690a1263f37b89ce1153c22625 100644 (file)
--- 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)
                })
        }()