feat - add mysql tests

Author: Nicola Strappazzon C

.gitignore

@@ -0,0 +1,5 @@
+# Test binary, build with `go test -c`
+*.test
+
+# MAC OS X
+.DS_Store

log/main.go

@@ -8,6 +8,10 @@ type Fields map[string]interface{}
 
 type Level uint32
 
+type Formatter struct {
+	logrus.TextFormatter
+}
+
 const (
 	PanicLevel Level = iota
 	FatalLevel
@@ -18,6 +22,13 @@ const (
 	TraceLevel
 )
 
+func UseTimestamp() {
+	logrus.SetFormatter(&Formatter{logrus.TextFormatter{
+		FullTimestamp:   true,
+		TimestampFormat: "2006-01-02 15:04:05",
+	}})
+}
+
 func SetLevel(level Level) {
 	logrus.SetLevel(logrus.Level(level))
 }

mysql/main.go

@@ -1,6 +1,7 @@
 package mysql
 
 import (
+	"bytes"
 	"database/sql"
 	"errors"
 
@@ -116,7 +117,15 @@ func (c *Connection) FetchBool(query string) bool {
 		return false
 	}
 
-	return true
+	if bytes.Equal(r.([]byte), []byte{48}) {
+		return false
+	}
+
+	if bytes.Equal(r.([]byte), []byte{49}) {
+		return true
+	}
+
+	return false
 }
 
 func (c *Connection) FetchOne(query string) any {

mysql/main_test.go

@@ -1,6 +1,7 @@
 package mysql_test
 
 import (
+	"fmt"
 	"os"
 	"testing"
 
@@ -29,6 +30,57 @@ func TestGet(t *testing.T) {
 	assert.NotEmpty(t, c.Instance, nil)
 }
 
+func TestFetchOne(t *testing.T) {
+	var r any
+	c := mysql.Get("test")
+
+	r = c.FetchOne("SELECT 'foo' AS one;")
+	assert.NotEmpty(t, r, nil)
+	assert.Equal(t, fmt.Sprintf("%s", r), "foo")
+
+	r = c.FetchOne("SELECT 'foo' AS one, 'bar' AS two;")
+	assert.Empty(t, r, nil)
+
+	r = c.FetchOne("SELECT 'foo' AS one UNION SELECT 'bar';")
+	assert.Equal(t, fmt.Sprintf("%s", r), "foo")
+}
+
+func TestFetchBool(t *testing.T) {
+	var r bool
+	c := mysql.Get("test")
+	r = c.FetchBool("SELECT true;")
+	assert.True(t, r)
+
+	r = c.FetchBool("SELECT false;")
+	assert.False(t, r)
+
+	r = c.FetchBool("SELECT 0;")
+	assert.False(t, r)
+
+	r = c.FetchBool("SELECT 1;")
+	assert.True(t, r)
+
+	r = c.FetchBool("SELECT null;")
+	assert.False(t, r)
+}
+
+func TestFetchAll(t *testing.T) {
+	count := 0
+	rows := []map[string]string{
+		{"c1": "a", "c2": "0"},
+		{"c1": "b", "c2": "1"},
+		{"c1": "c", "c2": "2"},
+	}
+
+	c := mysql.Get("test")
+	c.FetchAll("SELECT 'a' AS c1, 0 AS c2 UNION SELECT 'b', 1 UNION SELECT 'c', 2;", func(r map[string]string) {
+		assert.Equal(t, r, rows[count])
+		count++
+	})
+
+	assert.Equal(t, count, len(rows))
+}
+
 func TestClose(t *testing.T) {
 	c := mysql.Get("test")
 	assert.NotEmpty(t, c.Instance, nil)