add test example

Author: denghejun

go.sum

@@ -173,6 +173,7 @@ github.com/jackc/puddle v0.0.0-20190413234325-e4ced69a3a2b/go.mod h1:m4B5Dj62Y0f
 github.com/jackc/puddle v0.0.0-20190608224051-11cab39313c9/go.mod h1:m4B5Dj62Y0fbyuIc15OsIqK0+JU8nkqQjsgx7dvjSWk=
 github.com/jackc/puddle v1.1.0/go.mod h1:m4B5Dj62Y0fbyuIc15OsIqK0+JU8nkqQjsgx7dvjSWk=
 github.com/jackc/puddle v1.1.1/go.mod h1:m4B5Dj62Y0fbyuIc15OsIqK0+JU8nkqQjsgx7dvjSWk=
+github.com/jackc/puddle v1.1.3 h1:JnPg/5Q9xVJGfjsO5CPUOjnJps1JaRUm8I9FXVCFK94=
 github.com/jackc/puddle v1.1.3/go.mod h1:m4B5Dj62Y0fbyuIc15OsIqK0+JU8nkqQjsgx7dvjSWk=
 github.com/jmespath/go-jmespath v0.0.0-20180206201540-c2b33e8439af/go.mod h1:Nht3zPeWKUH0NzdCt2Blrr5ys8VGpn0CEB0cQHVjt7k=
 github.com/jonboulle/clockwork v0.1.0/go.mod h1:Ii8DK3G1RaLaWxj9trq07+26W01tbo22gdxWY5EU2bo=

pgx_pool_mock.go order_dao_example.gopgx_pool_mock.go renamed to order_dao_example.go

@@ -0,0 +1,33 @@
+package pgxpoolmock_test
+
+import (
+	"testing"
+
+	"github.com/driftprogramming/pgxpoolmock"
+	"github.com/driftprogramming/pgxpoolmock/testdata"
+	"github.com/golang/mock/gomock"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestName(t *testing.T) {
+	t.Parallel()
+	ctrl := gomock.NewController(t)
+	defer ctrl.Finish()
+
+	// given
+	mockPool := pgxpoolmock.NewMockPgxPool(ctrl)
+	columns := []string{"id", "price"}
+	pgxRows := pgxpoolmock.NewRows(columns).AddRow(100, 100000.9).ToPgxRows()
+	mockPool.EXPECT().Query(gomock.Any(), gomock.Any(), gomock.Any()).Return(pgxRows, nil)
+	orderDao := testdata.OrderDAO{
+		Pool: mockPool,
+	}
+
+	// when
+	actualOrder := orderDao.GetOrderByID(1)
+
+	// then
+	assert.NotNil(t, actualOrder)
+	assert.Equal(t, 100, actualOrder.ID)
+	assert.Equal(t, 100000.9, actualOrder.Price)
+}

order_dao_example_test.go

@@ -0,0 +1,40 @@
+package testdata
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/jackc/pgx/v4/pgxpool"
+)
+
+func main() {
+	connectionString := getConnectionString()
+	realPool, _ := pgxpool.Connect(context.Background(), connectionString)
+	orderDao := OrderDAO{
+		Pool: realPool,
+	}
+	order := orderDao.GetOrderByID(100)
+	fmt.Print(order)
+}
+
+func getConnectionString() string {
+	databaseHost := "10.0.20.101"
+	databaseUser := "admin"
+	databasePassword := "admin_password"
+	databaseName := "orderdatabase"
+	databaseConnectionPoolMaxSize := 100
+	databaseConnectionPoolMinSize := 10
+	defaultPoolHealthCheckPeriod := "10s"
+	connectionString := fmt.Sprintf(
+		"host=%s user=%s password=%s dbname=%s pool_max_conns=%s pool_min_conns=%s pool_health_check_period=%s",
+		databaseHost,
+		databaseUser,
+		databasePassword,
+		databaseName,
+		databaseConnectionPoolMaxSize,
+		databaseConnectionPoolMinSize,
+		defaultPoolHealthCheckPeriod,
+	)
+
+	return connectionString
+}

pgx_pool_mock_test.go

@@ -0,0 +1,27 @@
+package testdata
+
+import (
+	"context"
+
+	"github.com/driftprogramming/pgxpoolmock"
+)
+
+type Order struct {
+	ID    int
+	Price float64
+}
+
+type OrderDAO struct {
+	Pool pgxpoolmock.PgxPool
+}
+
+func (dao *OrderDAO) GetOrderByID(id int) *Order {
+	rows, _ := dao.Pool.Query(context.Background(), "SELECT ID,Price FROM order WHERE ID =$1", id)
+	for rows.Next() {
+		order := &Order{}
+		rows.Scan(&order.ID, &order.Price)
+		return order
+	}
+
+	return nil
+}