fix ci

axetroy · May 13, 2020 · 8afc282 · 8afc282
1 parent bcc1f69
commit 8afc282
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/internal/service/database/db.go b/internal/service/database/db.go
@@ -4,6 +4,7 @@ package database
 import (
 	"fmt"
 	"log"
+	"os"
 
 	"github.com/axetroy/go-server/internal/library/config"
 	"github.com/axetroy/go-server/internal/library/util"
@@ -24,6 +25,12 @@ func Dispose() {
 	}
 }
 
+func init() {
+	if os.Getenv("GO_TESTING") != "" {
+		Connect()
+	}
+}
+
 func Connect() {
 	DataSourceName := fmt.Sprintf("%s://%s:%s@%s:%s/%s?sslmode=disable", Config.Driver, Config.Username, Config.Password, Config.Host, Config.Port, Config.DatabaseName)
 

diff --git a/internal/service/redis/redis.go b/internal/service/redis/redis.go
@@ -4,6 +4,7 @@ package redis
 import (
 	"github.com/axetroy/go-server/internal/library/config"
 	"github.com/go-redis/redis"
+	"os"
 )
 
 var (
@@ -16,6 +17,12 @@ var (
 	Config               = config.Redis
 )
 
+func init() {
+	if os.Getenv("GO_TESTING") != "" {
+		Connect()
+	}
+}
+
 func Dispose() {
 	if Client != nil {
 		_ = Client.Close()