From a9071d629a02d697e18c45a227e292573f515dac Mon Sep 17 00:00:00 2001
From: Steven <stevenlgtm@gmail.com>
Date: Sun, 17 Dec 2023 13:59:15 +0800
Subject: [PATCH] chore: fix testing profile

---
 test/test.go | 17 ++++++++++++++++-
 1 file changed, 16 insertions(+), 1 deletion(-)

diff --git a/test/test.go b/test/test.go
index 6bd507c..8266a81 100644
--- a/test/test.go
+++ b/test/test.go
@@ -3,6 +3,7 @@ package test
 import (
 	"fmt"
 	"net"
+	"os"
 	"testing"
 
 	"github.com/yourselfhosted/slash/server/profile"
@@ -27,11 +28,25 @@ func GetTestingProfile(t *testing.T) *profile.Profile {
 	dir := t.TempDir()
 	mode := "dev"
 	port := getUnusedPort()
+	driver := getDriverFromEnv()
+	dsn := os.Getenv("DSN")
+	if driver == "sqlite" {
+		dsn = fmt.Sprintf("%s/slash_%s.db", dir, mode)
+	}
 	return &profile.Profile{
 		Mode:    mode,
 		Port:    port,
 		Data:    dir,
-		DSN:     fmt.Sprintf("%s/slash_%s.db", dir, mode),
+		DSN:     dsn,
+		Driver:  driver,
 		Version: version.GetCurrentVersion(mode),
 	}
 }
+
+func getDriverFromEnv() string {
+	driver := os.Getenv("DRIVER")
+	if driver == "" {
+		driver = "sqlite"
+	}
+	return driver
+}