fix test error

eric2788 · Apr 28, 2022 · 1c5a3b5 · 1c5a3b5
1 parent 71f80c3
commit 1c5a3b5
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/services/blive/live_server.go b/services/blive/live_server.go
@@ -38,6 +38,8 @@ func coolDownLiveFetch(room int64) {
 
 func LaunchLiveServer(wg *sync.WaitGroup, room int64, handle func(data *LiveInfo, msg biligo.Msg), cancelGet func(context.CancelFunc)) {
 
+	defer wg.Done()
+
 	log.Debugf("[%v] 正在獲取直播資訊...", room)
 
 	liveInfo, err := GetLiveInfo(room) // 獲取直播資訊
@@ -151,7 +153,6 @@ func LaunchLiveServer(wg *sync.WaitGroup, room int64, handle func(data *LiveInfo
 	}
 
 	cancelGet(stop)
-	wg.Done()
 }
 
 func shortDur(d time.Duration) string {

diff --git a/services/blive/live_server_test.go b/services/blive/live_server_test.go
@@ -5,6 +5,7 @@ import (
 	live "github.com/eric2788/biligo-live"
 	"github.com/eric2788/biligo-live-ws/services/database"
 	"github.com/go-playground/assert/v2"
+	"github.com/sirupsen/logrus"
 	"sync"
 	"testing"
 	"time"
@@ -26,6 +27,7 @@ func TestLaunchLiveServer(t *testing.T) {
 
 	var cancel context.CancelFunc
 	wg := &sync.WaitGroup{}
+	wg.Add(1)
 
 	go LaunchLiveServer(wg, 24643640, func(data *LiveInfo, msg live.Msg) {
 		t.Log(data, msg)
@@ -44,5 +46,6 @@ func TestLaunchLiveServer(t *testing.T) {
 }
 
 func init() {
+	logrus.SetLevel(logrus.DebugLevel)
 	_ = database.StartDB()
 }