Return pointer when Open

main.go

@@ -36,7 +36,7 @@ type DB struct {
 	joinTableHandlers map[string]JoinTableHandler
 }
 
-func Open(dialect string, args ...interface{}) (DB, error) {
+func Open(dialect string, args ...interface{}) (*DB, error) {
 	var db DB
 	var err error
 
@@ -79,7 +79,7 @@ func Open(dialect string, args ...interface{}) (DB, error) {
 		}
 	}
 
-	return db, err
+	return &db, err
 }
 
 func (s *DB) Close() error {

main_test.go

@@ -20,7 +20,7 @@ import (
 )
 
 var (
-	DB                 gorm.DB
+	DB                 *gorm.DB
 	t1, t2, t3, t4, t5 time.Time
 )
 
@@ -41,7 +41,7 @@ func init() {
 	runMigration()
 }
 
-func OpenTestConnection() (db gorm.DB, err error) {
+func OpenTestConnection() (db *gorm.DB, err error) {
 	switch os.Getenv("GORM_DIALECT") {
 	case "mysql":
 		// CREATE USER 'gorm'@'localhost' IDENTIFIED BY 'gorm';