From 12b412a989e2635b7038eea089b920ffe0dffeca Mon Sep 17 00:00:00 2001
From: nitroflap <fe.lap.prog@gmail.com>
Date: Wed, 16 Feb 2022 16:39:50 +0300
Subject: [PATCH] feat(discord_test): rewrote init as TestMain

---
 discord_test.go | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/discord_test.go b/discord_test.go
index f353185..f2dd28c 100644
--- a/discord_test.go
+++ b/discord_test.go
@@ -22,7 +22,7 @@ var (
 	envAdmin       = os.Getenv("DG_ADMIN")        // User ID of admin user to use for tests
 )
 
-func init() {
+func TestMain(m *testing.M) {
 	fmt.Println("Init is being called.")
 	if envBotToken != "" {
 		if d, err := New(envBotToken); err == nil {
@@ -35,6 +35,8 @@ func init() {
 			dg = d
 		}
 	}
+
+	os.Exit(m.Run())
 }
 
 //////////////////////////////////////////////////////////////////////////////