update test

Sora233 · Nov 5, 2021 · 5fe724b · 5fe724b
1 parent af5f51d
commit 5fe724b
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/lsp/weibo/freshCookie_test.go b/lsp/weibo/freshCookie_test.go
@@ -1,12 +1,20 @@
 package weibo
 
 import (
+	localutils "github.com/Sora233/DDBOT/utils"
 	"github.com/stretchr/testify/assert"
+	"net/http"
 	"testing"
+	"time"
 )
 
 func TestFreshCookie(t *testing.T) {
-	cookies, err := FreshCookie()
+	var cookies []*http.Cookie
+	var err error
+	localutils.Retry(5, time.Second, func() bool {
+		cookies, err = FreshCookie()
+		return err == nil
+	})
 	assert.Nil(t, err)
 	assert.NotEmpty(t, cookies)
 }