bot: use timeout for the default http client

opmega · Oct 4, 2022 · 2e4664a · 2e4664a
1 parent d4c80e6
commit 2e4664a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/bot.go b/bot.go
@@ -9,6 +9,7 @@ import (
 	"regexp"
 	"strconv"
 	"strings"
+	"time"
 )
 
 // NewBot does try to build a Bot with token `token`, which
@@ -20,7 +21,7 @@ func NewBot(pref Settings) (*Bot, error) {
 
 	client := pref.Client
 	if client == nil {
-		client = http.DefaultClient
+		client = &http.Client{Timeout: time.Minute}
 	}
 
 	if pref.URL == "" {

diff --git a/bot_test.go b/bot_test.go
@@ -54,8 +54,8 @@ func TestNewBot(t *testing.T) {
 
 	assert.NotNil(t, b.Me)
 	assert.Equal(t, DefaultApiURL, b.URL)
-	assert.Equal(t, http.DefaultClient, b.client)
 	assert.Equal(t, 100, cap(b.Updates))
+	assert.NotZero(t, b.client.Timeout)
 
 	pref = defaultSettings()
 	client := &http.Client{Timeout: time.Minute}