Add tests (#1)

Author: vearutop

README.md

@@ -1,7 +1,6 @@
 # dbwrap
 
 [![Build Status](https://github.com/bool64/dbwrap/workflows/test-unit/badge.svg)](https://github.com/bool64/dbwrap/actions?query=branch%3Amaster+workflow%3Atest-unit)
-[![Coverage Status](https://codecov.io/gh/bool64/dbwrap/branch/master/graph/badge.svg)](https://codecov.io/gh/bool64/dbwrap)
 [![GoDevDoc](https://img.shields.io/badge/dev-doc-00ADD8?logo=go)](https://pkg.go.dev/github.com/bool64/dbwrap)
 [![Time Tracker](https://wakatime.com/badge/github/bool64/dbwrap.svg)](https://wakatime.com/badge/github/bool64/dbwrap)
 ![Code lines](https://sloc.xyz/github/bool64/dbwrap/?category=code)

driver_ext_test.go

@@ -24,8 +24,12 @@ func TestRegister(t *testing.T) {
 	var l []string
 
 	driverName, err := dbwrap.Register("sqlmock",
+		dbwrap.WithOptions(dbwrap.Options{}),
+		dbwrap.WithAllOperations(),
+
 		dbwrap.WithMiddleware(
 			func(ctx context.Context, operation dbwrap.Operation, statement string, args []driver.NamedValue) (nCtx context.Context, onFinish func(error)) {
+				assert.Equal(t, "bool64/dbwrap_test.TestRegister", dbwrap.Caller())
 				l = append(l, "mw1 triggered: "+string(operation)+": "+statement)
 
 				return ctx, func(err error) {
@@ -48,6 +52,7 @@ func TestRegister(t *testing.T) {
 				}
 			},
 		),
+
 		dbwrap.WithInterceptor(func(ctx context.Context, operation dbwrap.Operation, statement string, args []driver.NamedValue) (context.Context, string, []driver.NamedValue) {
 			l = append(l, "intercepted: "+string(operation)+": "+statement)
 
@@ -171,6 +176,18 @@ mw1 triggered: query: SELECT a FROM b WHERE c = ? #intercepted
 mw2 triggered: query: SELECT a FROM b WHERE c = ? #intercepted
 mw2 done
 mw1 done
+mw1 triggered: rows_next: 
+mw2 triggered: rows_next: 
+mw2 done
+mw1 done
+mw1 triggered: rows_next: 
+mw2 triggered: rows_next: 
+mw2 done
+mw1 done
+mw1 triggered: rows_next: 
+mw2 triggered: rows_next: 
+mw2 failed: EOF
+mw1 failed: EOF
 mw1 triggered: rows_close: 
 mw2 triggered: rows_close: 
 mw2 done

options.go

@@ -78,6 +78,27 @@ func WithOperations(op ...Operation) Option {
 	}
 }
 
+// WithAllOperations enables all operations to be wrapped with middleware.
+// This also includes RowsNext.
+func WithAllOperations() Option {
+	return WithOperations(
+		Ping,
+		Exec,
+		Query,
+		Prepare,
+		Begin,
+		LastInsertID,
+		RowsAffected,
+		StmtExec,
+		StmtQuery,
+		StmtClose,
+		RowsClose,
+		RowsNext,
+		Commit,
+		Rollback,
+	)
+}
+
 // prepareOptions returns prepared Options and flag if they are operational.
 func prepareOptions(options []Option) (Options, bool) {
 	o := Options{}

runtime.go

@@ -24,11 +24,17 @@ func Caller(skipPackages ...string) string {
 
 	for i := 0; i < n; i++ {
 		f := runtime.FuncForPC(pc[i])
+
+		// Skip unnamed literals.
+		if strings.Contains(f.Name(), "{") {
+			continue
+		}
+
 		parts := strings.Split(f.Name(), "/")
 		parts[len(parts)-1] = strings.Split(parts[len(parts)-1], ".")[0]
 		p = strings.Join(parts, "/")
 
-		if p == "database/sql" {
+		if p == "database/sql" || p == "github.com/bool64/dbwrap" {
 			continue
 		}