Move backend tests to table driven tests instead of suite

Author: cschleiden

backend/mysql/mysql_test.go

@@ -25,37 +25,33 @@ func Test_MysqlBackend(t *testing.T) {
 
 	dbName := "test_" + strings.Replace(uuid.NewString(), "-", "", -1)
 
-	test.TestBackend(t, test.Tester{
-		New: func() backend.Backend {
-			db, err := sql.Open("mysql", fmt.Sprintf("%s:%s@/?parseTime=true&interpolateParams=true", testUser, testPassword))
-			if err != nil {
-				panic(err)
-			}
-
-			if _, err := db.Exec("CREATE DATABASE " + dbName); err != nil {
-				panic(errors.Wrap(err, "could not create database"))
-			}
-
-			if err := db.Close(); err != nil {
-				panic(err)
-			}
-
-			return NewMysqlBackend("localhost", 3306, testUser, testPassword, dbName, backend.WithStickyTimeout(0))
-		},
-
-		Teardown: func() {
-			db, err := sql.Open("mysql", fmt.Sprintf("%s:%s@/?parseTime=true&interpolateParams=true", testUser, testPassword))
-			if err != nil {
-				panic(err)
-			}
-
-			if _, err := db.Exec("DROP DATABASE IF EXISTS " + dbName); err != nil {
-				panic(errors.Wrap(err, "could not drop database"))
-			}
-
-			if err := db.Close(); err != nil {
-				panic(err)
-			}
-		},
+	test.BackendTest(t, func() backend.Backend {
+		db, err := sql.Open("mysql", fmt.Sprintf("%s:%s@/?parseTime=true&interpolateParams=true", testUser, testPassword))
+		if err != nil {
+			panic(err)
+		}
+
+		if _, err := db.Exec("CREATE DATABASE " + dbName); err != nil {
+			panic(errors.Wrap(err, "could not create database"))
+		}
+
+		if err := db.Close(); err != nil {
+			panic(err)
+		}
+
+		return NewMysqlBackend("localhost", 3306, testUser, testPassword, dbName, backend.WithStickyTimeout(0))
+	}, func(b backend.Backend) {
+		db, err := sql.Open("mysql", fmt.Sprintf("%s:%s@/?parseTime=true&interpolateParams=true", testUser, testPassword))
+		if err != nil {
+			panic(err)
+		}
+
+		if _, err := db.Exec("DROP DATABASE IF EXISTS " + dbName); err != nil {
+			panic(errors.Wrap(err, "could not drop database"))
+		}
+
+		if err := db.Close(); err != nil {
+			panic(err)
+		}
 	})
 }

backend/redis/redis_test.go

@@ -1,22 +1,38 @@
 package redis
 
 import (
+	"context"
 	"testing"
 	"time"
 
 	"github.com/cschleiden/go-workflows/backend"
 	"github.com/cschleiden/go-workflows/backend/test"
+	"github.com/go-redis/redis/v8"
 	"github.com/stretchr/testify/require"
 )
 
 func Test_RedisBackend(t *testing.T) {
-	test.TestBackend(t, test.Tester{
-		New: func() backend.Backend {
-			// Disable sticky workflow behavior for the test execution
-			b, err := NewRedisBackend("localhost:6379", "", "RedisPassw0rd", 0,
-				WithBlockTimeout(time.Millisecond*2))
-			require.NoError(t, err)
-			return b
-		},
-	})
+	test.BackendTest(t, func() backend.Backend {
+		address := "localhost:6379"
+		user := ""
+		password := "RedisPassw0rd"
+
+		// Flush database
+		client := redis.NewUniversalClient(&redis.UniversalOptions{
+			Addrs:    []string{address},
+			Username: user,
+			Password: password,
+			DB:       0,
+		})
+
+		if err := client.FlushDB(context.Background()).Err(); err != nil {
+			panic(err)
+		}
+
+		// Disable sticky workflow behavior for the test execution
+		b, err := NewRedisBackend(address, user, password, 0, WithBlockTimeout(time.Millisecond*2))
+		require.NoError(t, err)
+
+		return b
+	}, nil)
 }

backend/sqlite/sqlite_test.go

@@ -8,10 +8,8 @@ import (
 )
 
 func Test_SqliteBackend(t *testing.T) {
-	test.TestBackend(t, test.Tester{
-		New: func() backend.Backend {
-			// Disable sticky workflow behavior for the test execution
-			return NewInMemoryBackend(backend.WithStickyTimeout(0))
-		},
-	})
+	test.BackendTest(t, func() backend.Backend {
+		// Disable sticky workflow behavior for the test execution
+		return NewInMemoryBackend(backend.WithStickyTimeout(0))
+	}, nil)
 }