replace Wrap with NewDB/NewConn/NewTx

1 files changed, 8 insertions, 8 deletions

diff --git a/select_test.go b/select_test.go
index 7b4191a..c55fe42 100644
--- a/select_test.go
+++ b/select_test.go
@@ -20,7 +20,7 @@ import (
 func TestSelectReturningSomeRecords(t *testing.T) {
 	ctx := t.Context()
 	md := mock.NewDriver()
-	db := oblast.Wrap(sql.OpenDB(md))
+	db := oblast.NewDB(sql.OpenDB(md))
 
 	type basicRecord struct {
 		ID   int64  `db:"id"`
@@ -138,7 +138,7 @@ func TestSelectReturningSomeRecords(t *testing.T) {
 func TestSelectReturningNoRecords(t *testing.T) {
 	ctx := t.Context()
 	md := mock.NewDriver()
-	db := oblast.Wrap(sql.OpenDB(md))
+	db := oblast.NewDB(sql.OpenDB(md))
 
 	type basicRecord struct {
 		ID   int64  `db:"id"`
@@ -229,7 +229,7 @@ func TestSelectReturningNoRecords(t *testing.T) {
 func TestSelectIntoUnexpectedField(t *testing.T) {
 	ctx := t.Context()
 	md := mock.NewDriver()
-	db := oblast.Wrap(sql.OpenDB(md))
+	db := oblast.NewDB(sql.OpenDB(md))
 
 	type basicRecord struct {
 		ID          int64  `db:"id"`
@@ -268,7 +268,7 @@ func TestSelectIntoUnexpectedField(t *testing.T) {
 func TestSelectWithScanError(t *testing.T) {
 	ctx := t.Context()
 	md := mock.NewDriver()
-	db := oblast.Wrap(sql.OpenDB(md))
+	db := oblast.NewDB(sql.OpenDB(md))
 
 	type basicRecord struct {
 		ID        int64     `db:"id"`
@@ -331,7 +331,7 @@ func TestSelectWithScanError(t *testing.T) {
 func TestSelectIntoEmbeddedTypes(t *testing.T) {
 	ctx := t.Context()
 	md := mock.NewDriver()
-	db := oblast.Wrap(sql.OpenDB(md))
+	db := oblast.NewDB(sql.OpenDB(md))
 
 	type HasCreatedAt struct {
 		CreatedAt time.Time `db:"created_at"`
@@ -442,7 +442,7 @@ func TestSelectIntoEmbeddedTypes(t *testing.T) {
 func TestSelectCapturingQueryError(t *testing.T) {
 	ctx := t.Context()
 	md := mock.NewDriver()
-	db := oblast.Wrap(sql.OpenDB(md))
+	db := oblast.NewDB(sql.OpenDB(md))
 
 	type basicRecord struct {
 		ID   int64  `db:"id"`
@@ -490,7 +490,7 @@ func TestSelectCapturingQueryError(t *testing.T) {
 func TestSelectCapturingCloseError(t *testing.T) {
 	ctx := t.Context()
 	md := mock.NewDriver()
-	db := oblast.Wrap(sql.OpenDB(md))
+	db := oblast.NewDB(sql.OpenDB(md))
 
 	type basicRecord struct {
 		ID   int64  `db:"id"`
@@ -553,7 +553,7 @@ func TestSelectCapturingCloseError(t *testing.T) {
 func TestSelectNotPossibleWithoutTableName(t *testing.T) {
 	ctx := t.Context()
 	md := mock.NewDriver()
-	db := oblast.Wrap(sql.OpenDB(md))
+	db := oblast.NewDB(sql.OpenDB(md))
 
 	type basicRecord struct {
 		ID   int64  `db:"id"`