update all

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

wrapper/sql/wrapper_test.go

@@ -1,25 +1,25 @@
-// +build ignore
-
 package wrapper_test
 
 import (
+	"bytes"
 	"context"
-	"fmt"
+	"database/sql"
 	"testing"
+	"time"
 
 	"github.com/jmoiron/sqlx"
-	//_ "github.com/mattn/go-sqlite3"
-	wrapper "github.com/unistack-org/micro-wrapper-sql"
+	vmeter "go.unistack.org/micro-meter-victoriametrics/v3"
+	wrapper "go.unistack.org/micro-wrapper-sql/v3"
+	"go.unistack.org/micro/v3/meter"
+	"modernc.org/sqlite"
 )
 
-var (
-	schema = `
+var schema = `
   CREATE TABLE IF NOT EXISTS person (
 	  first_name text,
 	  last_name text,
 	  email text
 	);`
-)
 
 type Person struct {
 	FirstName string `db:"first_name"`
@@ -28,25 +28,51 @@ type Person struct {
 }
 
 func TestWrapper(t *testing.T) {
-	db, err := sqlx.Connect("sqlite3", "test.db")
+	wrapper.DefaultMeterStatsInterval = 100 * time.Millisecond
+	meter.DefaultMeter = vmeter.NewMeter()
+
+	sql.Register("micro-wrapper-sql", wrapper.NewWrapper(&sqlite.Driver{}))
+	wdb, err := sql.Open("micro-wrapper-sql", ":memory:")
 	if err != nil {
 		t.Fatal(err)
 	}
 
-	w := wrapper.NewWrapper(db)
-	defer w.Close()
-	ctx := context.Background()
+	db := sqlx.NewDb(wdb, "sqlite")
+	ctx, cancel := context.WithCancel(context.Background())
+	defer cancel()
 
-	db.MustExec(schema)
-
-	tx := db.MustBegin()
-	//tx.NamedExec("INSERT OR REPLACE INTO person (first_name, last_name, email) VALUES (:first_name, :last_name, :email)", &Person{"Jane", "Citizen", "jane.citzen@example.com"})
-	tx.Commit()
-
-	people := &Person{}
-	if err := sqlx.GetContext(ctx, w, people, "SELECT * FROM person limit 1"); err != nil {
+	wrapper.NewStatsMeter(ctx, db, wrapper.DatabaseHost("localhost"), wrapper.DatabaseName("memory"))
+	if _, err := db.Exec(schema); err != nil {
 		t.Fatal(err)
 	}
 
-	fmt.Printf("%#+v\n", people)
+	tx, err := db.BeginTxx(ctx, nil)
+	if 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{"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 {
+		t.Fatal(err)
+	}
+
+	if err := tx.Commit(); err != nil {
+		t.Fatal(err)
+	}
+
+	var peoples []*Person
+	if err := sqlx.SelectContext(ctx, 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))
+
+	if !bytes.Contains(buf.Bytes(), []byte(`micro_sql_idle_connections`)) {
+		t.Fatalf("micro-wrapper-sql output contains invalid output: %s", buf.Bytes())
+	}
 }