add sql wrapper tests, regen

Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org>

wrapper/sql/wrapper_test.go

@@ -4,12 +4,14 @@ import (
 	"bytes"
 	"context"
 	"database/sql"
+	//"fmt"
 	"testing"
 	"time"
 
 	"github.com/jmoiron/sqlx"
 	vmeter "go.unistack.org/micro-meter-victoriametrics/v3"
 	wrapper "go.unistack.org/micro-wrapper-sql/v3"
+	"go.unistack.org/micro/v3/logger"
 	"go.unistack.org/micro/v3/meter"
 	"modernc.org/sqlite"
 )
@@ -28,21 +30,34 @@ type Person struct {
 }
 
 func TestWrapper(t *testing.T) {
+	ctx := context.Background()
 	wrapper.DefaultMeterStatsInterval = 100 * time.Millisecond
 	meter.DefaultMeter = vmeter.NewMeter()
+	buf := bytes.NewBuffer(nil)
+	logger.DefaultLogger = logger.NewLogger(logger.WithLevel(logger.DebugLevel), logger.WithOutput(buf))
 
-	sql.Register("micro-wrapper-sql", wrapper.NewWrapper(&sqlite.Driver{}))
+	if err := logger.DefaultLogger.Init(); err != nil {
+		t.Fatal(err)
+	}
+
+	sql.Register("micro-wrapper-sql", wrapper.NewWrapper(&sqlite.Driver{},
+		wrapper.DatabaseHost("localhost"),
+		wrapper.DatabaseName("memory"),
+		wrapper.LoggerLevel(logger.DebugLevel),
+		wrapper.LoggerEnabled(true),
+	))
 	wdb, err := sql.Open("micro-wrapper-sql", ":memory:")
 	if err != nil {
 		t.Fatal(err)
 	}
 
 	db := sqlx.NewDb(wdb, "sqlite")
-	ctx, cancel := context.WithCancel(context.Background())
+	var cancel func()
+	ctx, cancel = context.WithCancel(ctx)
 	defer cancel()
 
 	wrapper.NewStatsMeter(ctx, db, wrapper.DatabaseHost("localhost"), wrapper.DatabaseName("memory"))
-	if _, err := db.Exec(schema); err != nil {
+	if _, err := db.ExecContext(ctx, schema); err != nil {
 		t.Fatal(err)
 	}
 
@@ -51,10 +66,10 @@ func TestWrapper(t *testing.T) {
 		t.Fatal(err)
 	}
 
-	if _, err := tx.NamedExec("INSERT OR REPLACE INTO person (first_name, last_name, email) VALUES (:first_name, :last_name, :email)", &Person{"Fist1", "Last1", "Email1"}); err != nil {
+	if _, err := tx.NamedExecContext(ctx, "INSERT OR REPLACE INTO person (first_name, last_name, email) VALUES (:first_name, :last_name, :email)", &Person{"Fist1", "Last1", "Email1"}); err != nil {
 		t.Fatal(err)
 	}
-	if _, err := tx.NamedExec("INSERT OR REPLACE INTO person (first_name, last_name, email) VALUES (:first_name, :last_name, :email)", &Person{"Fist2", "Last2", "Email2"}); err != nil {
+	if _, err := tx.NamedExecContext(ctx, "INSERT OR REPLACE INTO person (first_name, last_name, email) VALUES (:first_name, :last_name, :email)", &Person{"Fist2", "Last2", "Email2"}); err != nil {
 		t.Fatal(err)
 	}
 
@@ -63,16 +78,30 @@ func TestWrapper(t *testing.T) {
 	}
 
 	var peoples []*Person
-	if err := sqlx.SelectContext(ctx, db, &peoples, "SELECT * FROM person limit 2"); err != nil {
+	if err := sqlx.SelectContext(wrapper.QueryName(ctx, "get_all_person"), db, &peoples, "SELECT * FROM person limit 2"); err != nil {
 		t.Fatal(err)
 	}
 
 	_ = peoples
 	time.Sleep(1 * time.Second)
-	buf := bytes.NewBuffer(nil)
-	_ = meter.DefaultMeter.Write(buf, meter.WriteProcessMetrics(true))
+	mbuf := bytes.NewBuffer(nil)
+	_ = meter.DefaultMeter.Write(mbuf, meter.WriteProcessMetrics(true))
 
-	if !bytes.Contains(buf.Bytes(), []byte(`micro_sql_idle_connections`)) {
-		t.Fatalf("micro-wrapper-sql output contains invalid output: %s", buf.Bytes())
+	if !bytes.Contains(mbuf.Bytes(), []byte(`micro_sql_idle_connections`)) {
+		t.Fatalf("micro-wrapper-sql meter output contains invalid output: %s", buf.Bytes())
+	}
+
+	for _, tcase := range [][]byte{
+		[]byte(`"method":"ExecContext"`),
+		[]byte(`"method":"Open"`),
+		[]byte(`"method":"BeginTx"`),
+		[]byte(`"method":"Commit"`),
+		[]byte(`"method":"QueryContext"`),
+		[]byte(`"query":"get_all_person"`),
+		[]byte(`"took":`),
+	} {
+		if !bytes.Contains(buf.Bytes(), tcase) {
+			t.Fatalf("micro-wrapper-sql logger output contains invalid output: %s", buf.Bytes())
+		}
 	}
 }