Cover by tests log func

gosql/common/common.go

@@ -35,7 +35,7 @@ var rLogSpacesAll = regexp.MustCompile(`[\s\t]+`)
 var rLogSpacesEnd = regexp.MustCompile(`[\s\t]+;$`)
 var rSqlParam = regexp.MustCompile(`\$\d+`)
 
-func log(m string, s time.Time, e error, tx bool, query string, args ...any) {
+func log(m string, s time.Time, e error, tx bool, query string, args ...any) string {
 	var tmsg string
 
 	if tx {
@@ -67,7 +67,9 @@ func log(m string, s time.Time, e error, tx bool, query string, args ...any) {
 		color = "1;33"
 	}
 
-	fmt.Fprintln(os.Stdout, "\033["+color+"m[SQL]"+tmsg+qmsg+astr+estr+fmt.Sprintf(" %.3f ms", time.Since(s).Seconds())+"\033[0m")
+	res := fmt.Sprintln("\033[" + color + "m[SQL]" + tmsg + qmsg + astr + estr + fmt.Sprintf(" %.3f ms", time.Since(s).Seconds()) + "\033[0m")
+	fmt.Fprintln(os.Stdout, res)
+	return res
 }
 
 func fixQuery(query string) string {

gosql/common/common_export_test.go

@@ -0,0 +1,4 @@
+package common
+
+var Log = log
+var FixQuery = fixQuery

gosql/common/common_test.go

@@ -1,10 +1,12 @@
 package common_test
 
 import (
+	"fmt"
 	"io/ioutil"
 	"net/url"
 	"path/filepath"
 	"testing"
+	"time"
 
 	. "github.com/onsi/ginkgo"
 	. "github.com/onsi/gomega"
@@ -12,6 +14,42 @@ import (
 )
 
 var _ = Describe("common", func() {
+	Context("log", func() {
+		Context("time", func() {
+			It("calculate one second", func() {
+				str := common.Log("[func Exec]", time.Now().Add(time.Second*-1), nil, false, "")
+				Expect(str).To(Equal("\x1b[0;33m[SQL] [func Exec] (empty) (nil) 1.000 ms\x1b[0m\n"))
+			})
+		})
+
+		Context("format", func() {
+			It("with func name", func() {
+				str := common.Log("[func Exec]", time.Now(), nil, false, "")
+				Expect(str).To(Equal("\x1b[0;33m[SQL] [func Exec] (empty) (nil) 0.000 ms\x1b[0m\n"))
+			})
+
+			It("with sql query", func() {
+				str := common.Log("[func Exec]", time.Now(), nil, false, "select * from users")
+				Expect(str).To(Equal("\x1b[0;33m[SQL] [func Exec] select * from users (empty) (nil) 0.000 ms\x1b[0m\n"))
+			})
+
+			It("with error message", func() {
+				str := common.Log("[func Exec]", time.Now(), fmt.Errorf("Exec error"), false, "select * from users")
+				Expect(str).To(Equal("\x1b[0;33m[SQL] [func Exec] select * from users (empty) \x1b[0m\x1b[0;31m(Exec error) 0.000 ms\x1b[0m\n"))
+			})
+
+			It("with transaction flag", func() {
+				str := common.Log("[func Exec]", time.Now(), fmt.Errorf("Exec error"), true, "select * from users")
+				Expect(str).To(Equal("\x1b[1;33m[SQL] [TX] [func Exec] select * from users (empty) \x1b[0m\x1b[0;31m(Exec error) 0.000 ms\x1b[0m\n"))
+			})
+
+			It("with sql query arguments", func() {
+				str := common.Log("[func Exec]", time.Now(), fmt.Errorf("Exec error"), true, "select * from users where id=$1", 100)
+				Expect(str).To(Equal("\x1b[1;33m[SQL] [TX] [func Exec] select * from users where id=$1 ([100]) \x1b[0m\x1b[0;31m(Exec error) 0.000 ms\x1b[0m\n"))
+			})
+		})
+	})
+
 	Context("ParseUrl", func() {
 		Context("Success", func() {
 			It("for MySQL", func() {