GinkgorRecover

fakelag · Feb 25, 2024 · adf87f4 · adf87f4
1 parent ac72704
commit adf87f4
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/discordplayer/voiceworker.go b/discordplayer/voiceworker.go
@@ -11,6 +11,7 @@ import (
 	"github.com/fakelag/streaming-music-bot/entities"
 
 	"github.com/fakelag/dca"
+	. "github.com/onsi/ginkgo/v2"
 )
 
 type DcaMediaSession struct {
@@ -366,6 +367,8 @@ func (dms *DiscordMusicSession) hasNonBotMembersInVoiceChannel(voiceChannelID st
 }
 
 func (dms *DiscordMusicSession) disconnectAndExitWorker() {
+	defer GinkgoRecover()
+
 	dms.mutex.Lock()
 	defer dms.mutex.Unlock()
 
@@ -408,6 +411,7 @@ func (dms *DiscordMusicSession) cleanupEncodingAndVoiceSession(
 	encodingSession *dca.EncodeSession,
 	voiceConnection discordinterface.DiscordVoiceConnection,
 ) {
+	defer GinkgoRecover()
 
 	if encodingSession != nil {
 		encodingSession.Cleanup()