From d27bcb730b614ad9fa2190c86f46570c828acf75 Mon Sep 17 00:00:00 2001
From: Tom <tom@tdpain.net>
Date: Sun, 1 Nov 2020 15:25:11 +0000
Subject: [PATCH] =?UTF-8?q?=F0=9F=93=A6=20MySQL:=20Add=20tests=20and=20ben?=
 =?UTF-8?q?chmarks?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 mysql/mysql_test.go | 184 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 184 insertions(+)

diff --git a/mysql/mysql_test.go b/mysql/mysql_test.go
index b0843023..393a91c7 100644
--- a/mysql/mysql_test.go
+++ b/mysql/mysql_test.go
@@ -1 +1,185 @@
 package mysql
+
+import (
+	"os"
+	"testing"
+	"time"
+
+	"github.com/gofiber/utils"
+)
+
+var storeConfig = Config{
+	DropTable:    true,
+}
+
+func init() {
+	if v := os.Getenv("MYSQL_ADDRESS"); v != "" {
+		storeConfig.Address = v
+	}
+	if v := os.Getenv("MYSQL_USERNAME"); v != "" {
+		storeConfig.Username = v
+	}
+	if v := os.Getenv("MYSQL_PASSWORD"); v != "" {
+		storeConfig.Password = v
+	}
+	if v := os.Getenv("MYSQL_DATABASE"); v != "" {
+		storeConfig.DatabaseName = v
+	}
+}
+
+func Test_Set(t *testing.T) {
+
+	store := New(storeConfig)
+
+	id := "hello"
+	value := []byte("Hi there!")
+
+	store.Set(id, value, 0)
+
+	var (
+		returnedValue []byte
+		exp           int64
+	)
+
+	store.db.QueryRow(store.sqlSelect, id).Scan(&returnedValue, &exp)
+
+	utils.AssertEqual(t, returnedValue, value)
+	utils.AssertEqual(t, exp, int64(0))
+
+}
+
+func Test_SetExpiry(t *testing.T) {
+
+	store := New(storeConfig)
+
+	id := "hello"
+	value := []byte("Hi there!")
+	expiry := time.Second * 10
+
+	store.Set(id, value, expiry)
+
+	now := time.Now().Unix()
+	var (
+		returnedValue []byte
+		exp           int64
+	)
+	store.db.QueryRow(store.sqlSelect, id).Scan(&returnedValue, &exp)
+
+	delta := exp - now
+	upperBound := int64(expiry.Seconds())
+	lowerBound := upperBound - 2
+
+	if !(delta <= upperBound && delta > lowerBound) {
+		t.Fatalf("Test_SetExpiry: expiry delta out of bounds (is %d, must be %d<x<=%d)", delta, lowerBound, upperBound)
+	}
+
+}
+
+func Test_Get(t *testing.T) {
+
+	store := New(storeConfig)
+
+	id := "hello"
+	value := []byte("Hi there!")
+
+	store.db.Exec(store.sqlInsert, id, utils.UnsafeString(value), 0)
+
+	returnedValue, err := store.Get(id)
+	utils.AssertEqual(t, nil, err)
+	utils.AssertEqual(t, value, returnedValue)
+
+}
+
+func Test_Get_NoRows(t *testing.T) {
+
+	store := New(storeConfig)
+
+	id := "hello"
+
+	returnedValue, err := store.Get(id)
+	utils.AssertEqual(t, nil, err)
+	utils.AssertEqual(t, 0, len(returnedValue))
+
+}
+
+func Test_Get_Expired(t *testing.T) {
+
+	store := New(storeConfig)
+
+	id := "hello"
+	value := []byte("Hi there!")
+
+	store.db.Exec(store.sqlInsert, id, utils.UnsafeString(value), time.Now().Add(time.Minute*-1).Unix())
+
+	returnedValue, err := store.Get(id)
+	utils.AssertEqual(t, nil, err)
+	utils.AssertEqual(t, 0, len(returnedValue))
+}
+
+func Test_Delete(t *testing.T) {
+
+	store := New(storeConfig)
+
+	id := "hello"
+	value := []byte("Hi there!")
+
+	store.db.Exec(store.sqlInsert, id, utils.UnsafeString(value), 0)
+
+	err := store.Delete(id)
+	utils.AssertEqual(t, nil, err)
+
+	row := store.db.QueryRow(store.sqlSelect, id)
+	err = row.Scan()
+	utils.AssertEqual(t, noRows, err.Error())
+
+}
+
+func Test_Clear(t *testing.T) {
+
+	store := New(storeConfig)
+
+	id := "hello"
+	value := []byte("Hi there!")
+
+	store.db.Exec(store.sqlInsert, id, utils.UnsafeString(value), 0)
+
+	err := store.Clear()
+	utils.AssertEqual(t, nil, err)
+
+	row := store.db.QueryRow(store.sqlSelect, id)
+	err = row.Scan()
+	utils.AssertEqual(t, noRows, err.Error())
+
+}
+
+func Benchmark_Set(b *testing.B) {
+	store := New(storeConfig)
+
+	key := "aaaa"
+	val := []byte("This is a value")
+
+	expiry := time.Second * 60
+
+	b.ResetTimer()
+
+	for n := 0; n < b.N; n++ {
+		store.Set(key, val, expiry)
+	}
+
+}
+
+func Benchmark_Get(b *testing.B) {
+	store := New(storeConfig)
+
+	key := "aaaa"
+	val := []byte("This is a value")
+
+	store.Set(key, val, 0)
+
+	b.ResetTimer()
+
+	for n := 0; n < b.N; n++ {
+		store.Get(key)
+	}
+
+}