Add testutils for database

Author: jrauh01

database/example_upsert_streamed_test.go

@@ -3,50 +3,25 @@ package database
 import (
 	"context"
 	"fmt"
-	"github.com/creasty/defaults"
-	"github.com/icinga/icinga-go-library/logging"
-	"github.com/icinga/icinga-go-library/utils"
-	"go.uber.org/zap/zapcore"
 	"golang.org/x/sync/errgroup"
 	"time"
 )
 
 func ExampleUpsertStreamed() {
 	var (
-		testEntites = []MockEntity{
+		testEntites = []User{
 			{Id: 1, Name: "test1", Age: 10, Email: "[email protected]"},
 			{Id: 2, Name: "test2", Age: 20, Email: "[email protected]"},
 		}
-		g, ctx         = errgroup.WithContext(context.Background())
-		entities       = make(chan MockEntity, len(testEntites))
-		defaultOptions Options
+		g, ctx   = errgroup.WithContext(context.Background())
+		entities = make(chan User, len(testEntites))
+		logs     = getTestLogging()
+		log      = logs.GetLogger()
+		db       = getTestDb(logs)
+		err      error
 	)
 
-	logs, err := logging.NewLoggingFromConfig(
-		"Icinga Go Library",
-		logging.Config{Level: zapcore.DebugLevel, Output: "console", Interval: time.Second * 10},
-	)
-	if err != nil {
-		utils.PrintErrorThenExit(err, 1)
-	}
-
-	log := logs.GetLogger()
-
-	err = defaults.Set(&defaultOptions)
-	if err != nil {
-		log.Fatalf("cannot set default options: %v", err)
-	}
-
-	db, err := NewDbFromConfig(
-		&Config{Type: "sqlite", Database: ":memory:example-upsert", Options: defaultOptions},
-		logs.GetChildLogger("database"),
-		RetryConnectorCallbacks{},
-	)
-	if err != nil {
-		log.Fatalf("cannot create database: %v", err)
-	}
-
-	_, err = db.Query("CREATE TABLE IF NOT EXISTS mock_entity (id INTEGER PRIMARY KEY, name VARCHAR(255), age INTEGER, email VARCHAR(255))")
+	_, err = db.Query("CREATE TABLE IF NOT EXISTS user (id INTEGER PRIMARY KEY, name VARCHAR(255), age INTEGER, email VARCHAR(255))")
 	if err != nil {
 		log.Fatalf("cannot create table in db: %v", err)
 	}
@@ -63,9 +38,9 @@ func ExampleUpsertStreamed() {
 
 	time.Sleep(time.Second)
 
-	testSelect := &[]MockEntity{}
+	testSelect := &[]User{}
 
-	err = db.Select(testSelect, "SELECT * FROM mock_entity")
+	err = db.Select(testSelect, "SELECT * FROM user")
 	if err != nil {
 		log.Fatalf("cannot select from db: %v", err)
 	}